← 返回 2026-06-30

超越药物发现:纳米技术分子优化基准 Beyond Drug Discovery: The Nanotechnology Molecular Optimization (NMO) Benchmark

Matthias Blaschke, Daniel Kienzle, Zsuzsanna Koczor-Benda, Julian Lorenz, Rainer Lienhart, Fabian Pauly 📅 2026-06-29 👍 7 2026-07-13 08:37
分子优化 分子生成 基准测试 纳米技术 量子模拟

提出面向纳米技术应用的分子优化基准,用量子模拟替代代理指标

前置知识

分子生成

分子生成是指使用机器学习方法自动设计和生成具有特定属性的新分子结构的技术。它通常通过学习分子表示(如SMILES、SELFIES、图等)与目标属性之间的关系,来探索化学空间并寻找最优候选。在药物发现、材料科学等领域有广泛应用。传统方法主要基于大规模数据集预训练,然后通过强化学习或搜索策略进行优化。

本文的核心主题就是分子生成,但与传统的药物发现应用不同,它面向的是纳米技术领域的量子物理任务。理解分子生成的基本概念和方法有助于理解NMO基准的创新之处和挑战所在。

量子模拟

量子模拟是使用计算方法模拟量子系统行为的技术,常用于计算分子的电子结构、振动特性、输运性质等。它基于量子力学原理(如薛定谔方程),通过求解分子的波函数来预测其物理化学性质。计算精度从半经验方法(如xtb)到密度泛函理论(DFT)再到高精度波函数方法不等,精度越高计算成本越大。

NMO基准的关键创新就是用真实的量子模拟替代了传统的简单代理oracle。理解量子模拟的特点(计算成本高、物理意义明确、难以被简单模型利用)有助于理解为什么NMO构成了一个真正的挑战性基准。

分子表示

分子表示是将分子结构编码为机器学习模型可以处理的格式。常见表示包括SMILES(字符串)、SELFIES(保证有效性的字符串)、图表示(节点为原子,边为键)等。不同的表示有不同的优缺点:SMILES简单但不保证有效性;SELFIES保证有效性但可能产生截断;图表示直观但需要复杂的图神经网络。表示方法的选择直接影响模型的表达能力和优化效率。

本文提出的Graph Group SELFIES(GGS)是一个关键创新,它专门针对NMO任务的需求(需要建模电极结合位点)设计了新的表示方法。理解分子表示的原理有助于理解GGS的设计动机和技术优势。

Genetic Flow Networks (GFNs)

生成流网络是一种用于生成建模的机器学习方法,它通过学习一个策略来采样与奖励函数成正比的样本。与传统的强化学习不同,GFN不依赖于显式的策略梯度,而是通过匹配流动方程来学习。GFN结合了遗传算法的搜索特性和深度学习的表达能力,在分子优化等离散结构生成任务中表现良好。

本文的基线方法基于Genetic GFN框架,并对其进行了多项扩展以适应NMO任务的挑战(如崎岖的能量景观、结构约束)。理解GFN的基本原理有助于理解本文方法的技术基础和改进动机。

研究动机

现有的生成分子设计方法存在严重的领域局限性问题。虽然这些方法在药物发现基准上取得了强劲的性能,但它们主要集中在药物应用中,并且只使用简单的代理oracle进行评估,这些oracle不能反映真实世界目标的复杂性。具体来说,许多看似'通用'的模型实际上是隐式专业化的,它们在像PMO这样的简单基准上过度拟合,而不是学习可泛化的原理。例如,GenMol、InVirtuoGen和f-RAG等先进方法在它们被优化的代理任务上表现出色,但并不是为在全新的空间中发现结构而设计的。这导致了一个关键问题:这些方法在物理科学等结构上与药物发现完全不同的领域中的泛化能力非常有限。

本文的目标是本文的目标是建立一个连接机器学习和量子材料科学的桥梁,具体来说就是引入Nanotechnology Molecular Optimization(NMO)Benchmark Suite。这个基准套件有三个核心目标:第一,为ML社区提供一个严格的测试平台,用于评估分子设计在复杂量子物理应用中的性能;第二,为纳米技术研究提供一个发现引擎,帮助识别具有实际应用价值的候选分子;第三,建立一个可复制的接口,让ML研究者无需物理背景就能解决材料科学和纳米技术中的真实世界问题。NMO目前覆盖三个不同的纳米技术领域:Phonon Transport(声子传输)、Thermoelectrics(热电)和Molecular Optomechanics(分子光机学)。

与已有工作不同的是,本文的独特切入角度在于它不是简单地将现有的药物分子设计方法应用到纳米技术领域,而是从根本上重新设计了评估协议和任务设置。与之前使用简单代理oracle的基准不同,NMO用真实的量子模拟替代了这些代理,并且引入了严格的评估协议来防止针对特定任务的过拟合。更重要的是,NMO的任务需要建模分子与金表面的结合,这完全改变了分子的性质并引入了硬性的结构约束。这种设置创造了一个非平凡的机器学习挑战:极其崎岖的适应度景观(由量子模拟产生)、由于分子结合而产生的硬性约束,以及严格设计的基准协议。本文不仅提出了挑战,还识别了现有方法在NMO上的关键失败模式,并提出了相应的解决方案。

核心方法

本文的方法可以从两个层面来理解:基准层面和算法层面。在基准层面,NMO提供了一个标准化的评估环境,它通过自动化的后端处理复杂的量子模拟,使得ML研究者只需要输出一个一维的分子字符串表示(如SMILES或GGS),就能得到标量适应度分数和丰富的元数据。每个任务都有一个明确的适应度函数,结合了感兴趣的物理属性和惩罚项,并在有限的预算内进行优化。在算法层面,本文提供了一个扩展的Genetic GFN框架作为强基线方法。这个方法包括几个关键组件:使用GGS编码替代SMILES以原生建模电极结合位点、在合成数据集上进行预训练以避免药物数据集偏差、使用Transformer架构替代循环网络、引入自适应稳定性控制机制(动态冷却和动态探索)来防止在崎岖景观上的训练崩溃,以及通过描述符注入提供化学直觉。

核心创新点有三个方面:第一,Graph Group SELFIES(GGS)表示方法,它通过片段库和DAG结构构建分子,能够原生建模电极结合位点,保证化学有效性,并且支持从合成数据集进行无偏见预训练。GGS的关键优势在于它通过构造保证有效性,而GS需要后处理解析器,这可能导致截断。第二,领域无关的预训练策略,通过在30万个随机图的合成数据集上预训练,模型学习基本的化学组装原理,而不继承历史数据集的不可控偏差。这对于NMO任务特别关键,因为没有可用的已知分子属性数据集进行预训练。第三,严格的基准协议,禁止任务特定的数据集和超参数调整,要求单一配置在三个物理上不同的任务上都有效,这激励通用优化方法而非针对特定任务的过拟合专家。

方法步骤详情

NMO基准的工作流程如下:生成代理输出一个分子字符串表示(可以是SMILES或GGS)。对于GGS表示,基准后端解析字符串,构建功能纳米结构(单分子结或SAM),执行3D几何优化,然后运行复杂的量子模拟。对于每个任务,适应度函数定义为 $f_{task}(x) = c_{task} \cdot O_{task}(x) \cdot PSA_{task}(x) \cdot Prot(x) \cdot \Theta(x)$,其中 $O_{task}(x)$ 是感兴趣的物理属性,$PSA_{task}(x)$ 是基于合成可及性评分的惩罚,$Prot(x)$ 是旋转键数量的惩罚,$\Theta(x)$ 是硬性物理约束(如果分子不符合标准则适应度为零)。基线方法的训练过程包括:首先在30万个随机GGS图的合成数据集上预训练模型,建立基础的有效性先验;然后在优化阶段,模型通过Genetic GFN框架探索化学空间,使用遗传操作(交叉、变异)生成新的候选分子,并通过oracle评估获得适应度分数;模型根据这些反馈更新策略,同时在辅助任务上预测分子描述符以防止潜在空间坍缩。

技术新颖性

技术新颖性体现在多个方面:GGS表示是第一个原生支持电极结合位点的分子表示方法,它通过DAG结构跟踪每个节点的价态和可用耦合位点,只允许化学上可能的连接,确保每个生成的图都描述一个有效的分子。合成预训练策略避免了药物数据集的分布偏差,这是一个通用原则,可以应用到其他领域特定数据稀缺的场景。严格基准协议的设计,特别是禁止任务特定数据集和强制单一配置的要求,在分子设计基准中是不常见的,这使得NMO成为一个真正的通用能力测试。此外,本文将三个具有科学实用性的纳米技术任务标准化为机器学习基准,这本身就是一项创新,因为它需要深入理解量子物理和机器学习两个领域。

Fragment library and GGS representation
Figure 2: Fragment library and GGS representation

实验结果

实验结果揭示了一些令人意外的发现。最先进的方法在NMO任务上表现糟糕:f-RAG在所有三个任务上的AUC接近零(0.00-0.10),GenMol同样失败(AUC 0.00-0.02),尽管它们在PMO基准上有SOTA性能。相比之下,简单的训练无关遗传算法molGA在MO任务上具有竞争力(AUC 0.57),并且在TE和PH上也有非零性能。这表明最近在药物基准上的进展主要来自任务特定先验而非更好的优化。通过消融研究,论文识别出关键的成功因素:GGS编码对于双锚定拓扑的TE和PH任务至关重要,molGA从SMILES切换到GGS后TE任务的AUC从0.13大幅提升到0.68;合成预训练也有帮助但不是充分的。基线方法在所有任务上都找到了超越文献结果的候选分子:TE任务找到ZT=8.5的分子(文献阈值ZT>3),PH任务找到κph=0.1 pW/K的分子,MO任务找到P=9.9的分子(之前最佳是7.88)。此外,论文发现GGS编码在所有任务上都显著降低了SA分数,使其更容易合成,这对实际应用很重要。

Evaluation of Literature Methods and Ablation Study on the NMO benchmark
Table 1: Evaluation of Literature Methods and Ablation Study on the NMO benchmark
查看结构化数据
任务指标本文基线提升
Thermoelectric (TE) Top-10 AUC 0.78 ± 0.23 molGA (SMILES): 0.13 ± 0.10 ~5倍提升
Thermoelectric (TE) Mean Top-10 Fitness 1.19 ± 0.42 molGA (SMILES): 0.26 ± 0.13 ~4.6倍提升
Phonon (PH) Top-10 AUC 0.33 ± 0.16 molGA (SMILES): 0.07 ± 0.01 ~4.7倍提升
Phonon (PH) Mean Top-10 Fitness 0.75 ± 0.48 molGA (SMILES): 0.10 ± 0.02 ~7.5倍提升
Molecular Optomechanics (MO) Top-10 AUC 0.42 ± 0.08 molGA (SMILES): 0.57 ± 0.14 molGA略优,但本文SA更易合成

局限与改进

作者承认了几个局限性。半经验xtb计算涉及精度和可计算性之间的权衡:MO任务可能在高适应度尾部(P>15)高估上转换强度,候选分子在这个范围内应谨慎解释,尽管显式检查仍显示强性能。最近发布的xtb方法可能会在未来解决这一问题。安全措施不能捕获所有不理想的模式:例如,PH任务有一个退化区域,任意长的分子实现人为的低热导,这在基准协议级别通过允许长度边界来解决,并且通过计算成本进一步自限制。片段库引入了一定程度的偏差,但这可以显式控制。高计算成本将评估限制在五个种子,降低了统计效力,但这匹配甚至超过了标准实践。我观察到另一个潜在局限性:虽然GGS原生支持电极结合,但它依赖于片段库的设计,如果库中的片段不适合某些特定的分子结构,可能会限制搜索空间。

独立分析的弱点

尽管基线方法取得了良好的结果,但仍有一些可以改进的方向。首先是计算效率:目前每次oracle调用需要1-5秒(取决于任务和分子大小),10000次调用的预算意味着完整优化需要数小时。虽然这已经相对高效,但更快的近似方法或并行化可以加速探索。其次是表示的灵活性:GGS虽然保证有效性,但片段库的设计隐式地限制了搜索空间。对于某些任务,更灵活的表示(如SMILES加上后处理验证)可能探索到更好的区域,特别是在MO任务上观察到GGS降低了AUC。第三是预训练数据的质量:虽然合成预训练避免了药物偏差,但完全随机的图可能不是最优的先验。可以考虑基于化学原理或领域知识设计更智能的合成分布。第四是稳定性控制机制:动态冷却和动态探索虽然有效,但它们的超参数可能需要调整以适应不同任务。更自适应的机制可能会更好。最后是评估指标:虽然Top-10 AUC和RI都很好,但它们关注的是最终结果,缺乏对优化过程的理解。可以引入中间指标来分析方法的探索-利用平衡。

未来方向

作者提出的未来方向包括:在纳米技术社区内整合NMO,收集实验验证结果以增强基准的科学可信度。扩展任务到其他纳米技术领域,如金属有机框架或更广泛的功能材料。探索更高精度的量子化学方法(如DFT)作为验证或替代oracle。基于成果可延伸的方向包括:开发更先进的锚点建模方法,如通过辅助任务或潜变量来学习最优结合位置。研究在NMO类任务上表现更好的新架构,特别是能够处理崎岖能量景观和硬约束的方法。探索多任务学习,同时在三个NMO任务上训练以获得更好的泛化能力。将NMO的严格协议应用到其他领域,如催化剂设计或有机电子学。最后,随着更多方法在NMO上测试,可以建立一个排行榜来跟踪进展,类似于PMO或其他基准。

复现评估

论文的复现性总体上是良好的。作者提到NMO基准套件是完全可访问的,后端自动处理复杂的量子模拟,使得ML研究者无需物理背景就能使用。对于基线方法,论文提供了详细的方法描述和消融研究,包括架构选择、超参数设置和训练细节。虽然没有明确提到代码是否开源,但论文描述的方法足够详细,经验丰富的研究者应该能够复现。计算资源方面,每次oracle调用需要1-5秒,10000次调用的预算意味着每个种子需要数小时,完整评估(5个种子)需要数天到一周的计算时间,这对大多数研究机构来说是可行的。数据方面,论文提供了片段库和合成预训练数据集的SMILES翻译版本,确保所有方法都能访问相同的基础知识。然而,一个潜在的问题是半经验xtb包的实现细节可能影响结果的可比性,如果不同版本或实现产生略有不同的物理属性。总体而言,这篇论文的复现性处于机器学习论文的中上水平,特别是考虑到它涉及跨领域的技术。