《Symmetry》:Symmetry Diagnostics for Equivariant Molecular Force Fields: When a Bulk Metric Manufactures a Gauge Artifact
编辑推荐:
等变图神经网络编码了分子力场必须遵守的E(3)对称性,然而它们经常被训练用于优化几何目标,而这些几何目标自身的对称性却很少被检验。研究人员开发了一种对对称性敏感的诊断协议,结合了刚体变换不变性与等变度量、图级目标的宇称分析、辅助头的可预测性度量以及扭转与镜像扰
等变图神经网络编码了分子力场必须遵守的E(3)对称性,然而它们经常被训练用于优化几何目标,而这些几何目标自身的对称性却很少被检验。研究人员开发了一种对对称性敏感的诊断协议,结合了刚体变换不变性与等变度量、图级目标的宇称分析、辅助头的可预测性度量以及扭转与镜像扰动测试。该协议分离了全局误差度量所混淆的三种效应。将其应用于rMD17数据集上的精简GotenNet模型,得出了清晰的实证图景。该架构在数值精度上保持了旋转、平移和反射不变性(约10?6相对力误差),因此辅助头使等变性保持完整。然而,几何目标在宇称上存在差异:cos??和|cos??|是偶宇称,而sin??和标量三重积是奇宇称且均值接近零。cos??与|cos??|之间的差异反映的是扭转矩缩放,而非宇称。在三种种子的配对运行中,辅助头对力平均绝对误差的影响主要由种子噪声主导。当每个模型预测的能量被常数偏移时,在原始能量平均绝对误差中一种与种子一致且依赖于分子的效应消失了,尽管力和构象能量差异保持不变,这展示了全局度量如何制造规范假象。研究人员提供了保持对称性的辅助目标的设计原则,而非新的力场基准测试。
近年来,等变图神经网络已成为机器学习原子间势函数的标准设计。这些模型在架构上内置了分子势能遵守的E(3)对称性,使得结构对称性在设计上得到保证。然而,在训练这类模型时,研究人员逐渐引入几何辅助监督,即要求网络重现特定的几何量(如键角或扭转角统计)。这些辅助目标往往出于提升精度的动机,而其自身的变换行为(如宇称特性)却很少被系统性审查。全局平均绝对误差等误差度量在评估模型精度时,已知会抹平特定的对称性结构并混淆能量规范自由度,从而导致评估失真甚至产生虚假现象。由于架构层面的“保持对称性”并不能保证监督层面的“对称性匹配”,研究人员开展此项研究旨在提出一套完整的诊断协议,以分离并量化全局误差度量所混淆的效应。该研究得出结论指出,几何监督的影响主要源于目标设计而非等变性的破坏,并基于能量规范校正揭示了原始指标中的规范假象。此研究已在学术期刊《Symmetry》上发表,对规范未来力场辅助目标的对称性设计具有重要的理论和实践指导意义。
在关键技术方法上,研究人员主要采用了以下方法:首先构建了包含刚体变换不变性/等变度量、目标宇称分析、辅助头可预测性度量及扰动测试的对称敏感诊断框架;其次利用Haar均匀采样的旋转、平移和反射变换进行数值精度检验;随后采用镜像符号翻转残差分析图级目标的宇称特征;最后引入了基于常数偏移的能量规范校正与打乱目标的负控制实验。研究使用的样本队列来源于rMD17数据集的十种分子,并系统测试了精简的GotenNet骨干网络配置进行研究。
研究结果具体如下:
5.1. 结构等变性在数值精度上得到保持
通过将构象进行旋转、平移和镜像反射等刚体变换,研究人员检测了能量的不变性与力的等变误差。结果显示,在所有十种分子上,能量和力误差均达到浮点数精度级别,力等变误差在10
?6量级。由于物理辅助头仅在训练时激活,推理时使用纯净等变骨干网络,系统的结构对称性未被破坏,从而确认了任何辅助效应应完全归因于监督层面而非架构层面。
5.2. 扭矩设计与真实宇称的差异
镜像测试检验了常见目标的变换性质。研究发现,cos
??和|cos
??|在反射下绝对不变,而sin
??和标量三重积发生精确的符号翻转。研究证实了cos
??与|cos
??|的区别在于|cos
??|具有较大的稳定幅值且不在图层级平均中抵消,属于纯矩设计问题,而非宇称特性的反映。
5.3. 宇称-奇目标是近零均值且对称性不匹配
通过对帧级分布进行测算,发现|cos
??|在分子间保持较大且稳定的均值,作为监督信号缩放良好。相反,奇宇称目标sin
??和标量三重积在图层级池化后均值接近零,是物理能面在无法观测的任意能量基准点上产生的对称性错配现象。
5.4. 原始辅助效应:力的种子噪声与敏感于规范的能量符号
在三种种子的配对运行中,研究发现辅助头对力平均绝对误差(MAE)的影响被种子噪声主导,没有分子达到统计显著性。而在未校正的原始能量平均绝对误差评估中,观察到了依赖于特定分子的种子一致性符号,这在乙醇和水杨酸中尤为显著。结合辅助头的可预测性诊断发现,其预测相关性与实际产生的额外效应之间不具强预测关系。
5.5. 能量符号是一个能量-规范偏移
研究指出绝对能量的零点是规范自由度。偏移校正结果显示,去除每个模型中特定的任意常数后,乙醇和水杨酸上显著的困难构象能量偏移完全崩溃至0.001 kcal mol
?1以内,这种均一的稳定能量偏移完全掩盖了物理特性。该现象揭示了原始能量平均绝对误差将规范的常数偏移误读成了依赖于构象改变的组织物理效应。
5.6. 打乱-目标控制未重现原始偏移符号
通过将图级目标在批次内置换,使冗余额外头去耦于同事物几何分布,构造相斥对称控制实验。结果显示,这种打乱目标结构使构象破坏了能量原始符号的种子一致性,证明能量偏移现象被高度绑定于拥有正确对称结构的几何先验目标体系,而非简单的不利边界条件。
5.7. 扰动响应揭示了图级粗糙性
通过扫描阿司匹林和水杨酸的可旋转键,研究人员对比了流设模型与辅助约束模型的能量曲线扰动响应。尽管内部路径不同,两条全结构几何对称对应的能量响应曲线在每条正常几何结构两侧平滑且近似算符网络学习残差几乎一致。这表明将分布在图级的几何先验回避池化为标量后,并未为骨干网络额外提供有意义的构象空间预测信号。
讨论与结论
研究人员详细准则反映了三个关键层次:首先,必须独立区分骨架计算产生的结构等变性与监督前者执行网络架构本身的一致不一致原则;其次,将扭矩等几何识别池化为图层级标量会因函数系综对称抵消而破坏原始原子精度预警,且应当避免把奇宇称伪标量用于指导偶宇称能量力场计算;最后,研究发现全局能量度量横跨混杂产生的非相关物理自由度,诊断协议必须首先检验其对于自由度的规范敏感边界。研究人员总结得出,当前条件并存的图级全局均值误差一致表现被大尺度种子噪声主导,且唯独一致出现的结构依赖符号特征在规范分被证明将深度架构完全的无害演化验证曰路径,证实关注的几何与未表现途径。综上所述,研究人员提出,在未验证任何规范互联独立的偏置能量差或不可逆任务变化前,不应轻率将一个总额均值统合导致为物理演化,并建议开发非均构边缘级图案参差正则的动态长时聚焦检验标准,以延长至大型手性分子系统的残留游离对称结构模型。