A11 · 第 6 章 · 第三编 图任务与综合复习
几何深度学习综合复习:表达、瓶颈与评估
以分子构象的总能量与原子力预测贯穿图构造、消息传递、置换与旋转对称、谱—空间联系和图级读出,区分过平滑与过压缩,分析消息传递表达界限、图生成动力学、数据泄漏、结构拆分和规模外推。
报告页面错误本章目标
- 为带坐标图定义节点、边、单位、图级标量和节点向量的对称性合同。
- 手算消息传递与图级读出,并验证节点置换下的等变和不变关系。
- 把 Laplacian 多项式滤波与局部空间传播联系起来,说明跨图谱基迁移边界。
- 区分过平滑、过压缩和消息传递表达界限,并为各自设计诊断。
- 按实体、结构和轨迹拆分数据,排查近重复与目标泄漏并评价规模外推。
本页目录
贯穿任务:由分子构象预测总能量与原子力
每个样本是一种分子在某个构象下的图。节点代表原子,节点特征含元素类型和可核实属性;坐标 带长度单位。边可由化学键、距离截断或两者共同定义,边特征含键类型、距离和相对位移。目标是图级标量总能量 与每个原子的向量力 。
这个任务同时暴露几何学习的核心边界。原子编号改变不应改变物理对象;整体平移或旋转不应改变能量,力则应随旋转;相同分子的相邻构象高度相关,拆分不当会泄漏;更大分子有更长依赖路径,训练规模内的成功不保证外推。
图构造是模型的一部分。距离截断决定哪些交互可在一层发生,周期边界要使用最小镜像或明确晶胞,缺失长程边不能靠网络名称自动恢复。若真实作用依赖三体角度,只提供无方向距离仍可能丢失信息;需要角特征、高阶消息或等变张量表示。
先区分图信号和任务读出
原子类型是离散节点信号,电荷或质量是标量,位置和力是向量;键类型与相对位移属于边信号。把所有量拼成无单位数字会让尺度和变换规则混乱。坐标可中心化以去除平移,但中心定义、质量加权与孤立分量要固定。
图级能量是广延量:若两个无相互作用系统合并,总能量通常相加。因而把原子贡献求和是有物理动机的读出:
平均读出更适合每原子平均或其他强度量,却会删除节点数。最大读出只保留极端节点。读出选择必须与目标量语义和训练损失归约一致。
力可由等变节点头直接预测,也可定义为能量梯度
后一方式使力场由同一标量势导出,但只有在能量模型可微且结构正确时才有意义;它不保证学到真实势能,也不消除数值积分误差。
一个三原子构象输入顺序为 ,模型输出能量 和力矩阵 。交换两个同类氢的行并同步交换边后,能量应相同,力矩阵的两行也应交换。
再用旋转矩阵 旋转所有坐标。能量合同为 ,力合同为 。若只检查能量不变,无法发现力方向错误;若只旋转坐标而未同步处理外场或周期晶胞,测试对象也已改变。每种对称测试都要明确哪些输入共同变换。
反射是否是对称要由任务决定。普通无外场能量可对刚体反射不变,但手性识别可能必须区分镜像。将旋转和反射都硬编码为不变,会使模型无法完成依赖手性的任务。
消息传递连接局部关系与共享计算
典型层写成
共享 并用求和、均值或最大等交换不变聚合,可使节点表示对重编号等变。图级不变读出再消除行顺序。聚合是否求和影响度与规模:总作用常需求和,邻居平均可控制尺度却丢失邻居数,必要时把度作为额外信号。
路径 的标量初始特征为 。令消息 ,聚合为邻居求和,更新为 。于是
图级求和读出为 。交换节点一和三并同步交换边,输出行变为 ,求和仍为十六。若只交换特征而不交换邻接,就不是重编号测试,而是把不同原子放到不同结构位置。
一层只访问一跳邻域, 层通常最多组合 跳信息。残差、归一化和注意力会改变优化与权重,不改变没有全局通道时的基本路径限制。虚拟全局节点可缩短路径,却会形成新的信息瓶颈并改变对称和规模语义。
群等变要求特征类型一致
仅用距离构造标量消息天然对旋转不变,适合能量等标量;要预测力,模型需产生随旋转同步变化的向量。可让标量系数乘相对方向,或使用按旋转群表示组织的向量与高阶特征。普通多层感知机对坐标分量的任意混合没有结构保证。
等变层复合仍等变,前提是非线性、归一化、聚合和读出都尊重特征类型。对向量逐坐标使用不同阈值会破坏旋转;按向量范数门控标量系数可保持方向变换。数值测试应采样多组旋转、平移和置换,比较误差相对于输出尺度,而非只看一个示例。
对称性是归纳偏置,不是越多越好。外电场、墙面、重力或晶向会破坏完整旋转对称;可把这些外部方向作为共同旋转的输入,或只要求实际对称子群。
谱视角与空间视角在多项式处相遇
无向图 Laplacian 的谱分解 允许定义滤波 。若 用 次多项式近似,则
只沿至多 跳路径传播,因而无需显式特征分解,可实现为空间局部运算。这说明局部消息传递与谱滤波不是两个完全无关的目录,而是在线性共享算子下可以互相表达。
三节点路径的组合 Laplacian 为
对信号 ,有 。取一阶平滑滤波 ,得到 ,端点差异缩小。计算每个分量只需要本节点和一跳邻居,与一次线性消息聚合一致。
跨图直接学习每个特征值上的独立参数会绑定训练图的谱基和节点数。特征向量符号任意,重复特征值的特征子空间还可旋转;不同离散网格的谱不逐列对应。多项式、局部核或基于几何量的共享参数更易跨图,但仍受采样密度、边界和 Laplacian 选择影响。
过平滑与过压缩是不同故障
过平滑指多层传播后节点表示趋同,难以区分局部角色。可检查节点间余弦、类间距离、特征协方差秩和层间变化。它常与反复平均、图混合和深层优化有关;残差、初始特征跳接或限制平滑可缓解,但不保证解决长程任务。
过压缩指大量远端信息必须穿过有限维、有限边的瓶颈。即使节点表示尚未趋同,根节点也可能无法无损编码指数增长的远端组合。可按距离测梯度或敏感度、比较增加层数前后的远程任务,并检查图的割边与路径数。
两节点图初始特征为 。若每层把自己和邻居取均值,一层后两个节点都为零,身份差异立即消失,这是过平滑。
另一图是一棵深三层的二叉树,八个叶子的二值信号都要影响根节点分类。所有叶信息必须经根附近少数边压入固定维消息;即使采用不会简单平均成相同向量的更新,根仍需区分 种叶组合,这是过压缩压力。增加层数让信息到达根,却不增加割边容量。两个问题需要不同诊断,不能都叫“网络太深”。
重连、层级池化、全局注意力或虚拟节点可缩短路径;更高维消息可增加容量。它们也会增加成本、改变局部性并可能引入新泄漏。应在同等参数与计算预算下比较,并保留原图结构基线。
消息传递有可陈述的表达边界
使用共享更新和交换不变邻域聚合的常见消息传递网络,其区分能力通常受一维 Weisfeiler–Lehman 颜色细化类方法限制。若两个图在每轮都给每个节点相同的邻域多重集,任何这类网络都会保持相同节点表示,图级求和也无法分开。具体结论依赖聚合是否单射、初始特征和读出,不能扩展成所有图网络都完全等同。
图甲是六节点环,图乙是两个互不相连的三角形。所有节点初始特征相同;两个图中每个节点的度都为二。第一层时,每个节点都收到两个相同邻居消息,因此更新后仍相同。归纳地,任意层每个节点继续看到两个相同表示,两个图各有六个相同节点表示。
若图级读出只对节点表示求和,两个输出相同,尽管图甲连通、图乙有两个分量。加入连通分量、位置编码、高阶子图或更强测试可区分,但特征构造必须保持所需置换性质,并避免把样本身份写进唯一编号。
唯一节点 ID 能打破对称,却可能让模型记住训练图,且不同编号给不同结果。谱位置编码可增加全局信息,但有符号、重特征空间和跨图对齐问题。表达更强也可能更易过拟合;应围绕任务所需结构选择。
图生成和动力学扩展了评价对象
若任务从性质预测扩展到分子结构生成,需联合评价节点、边、有效性、唯一性和覆盖,先对同构图去重。结构合法不证明稳定或可合成。若进一步滚动分子动力学,需评价力、能量、动量、时间积分、长期分布和未见温度或规模。
在能量梯度模型中,力由标量势导出可提供保守结构;截断邻域、近似能量和积分器仍会产生误差。训练一步力误差低不能代替长时间轨迹验证。生成质量、性质预测和物理可信度必须分别给证据。
数据泄漏在图数据中常跨越边界
随机按构象拆分会让同一分子的近邻构象同时进入训练和测试,模型可能记住原子组成与局部几何。应按分子实体、骨架、实验批次或完整轨迹分组,再在组内划分构象用途。若目标是泛化到新分子,按单个构象随机拆分回答了错误问题。
节点或边任务也可能跨图泄漏:同一大网络随机拆边时,训练消息传递仍能看到测试边端点和邻域;这属于传导设定,不等于归纳到新图。需声明测试时是否可见完整图、测试节点特征和未标注边。
构图本身可泄漏。若边由目标相似度、未来接触或全数据标签统计产生,模型在输入中已看到答案线索。标准化均值、词表、谱基和近邻索引也只能用训练部分拟合。多构象、镜像和节点重编号副本应在拆分前聚类去重。
评价沿输出单位和偏移轴展开
能量误差要注明每图还是每原子及能量单位;力误差按原子向量和力单位报告,并按元素、距离、分子大小和构象能区间切片。总能量求和随节点数增长,平均误差与总误差回答不同问题。
除任务误差外,还要做置换、平移、旋转和必要的反射测试;报告等变残差而非只看数据增强结果。按未见骨架、节点数、度、几何范围和构图噪声评价分布偏移。对动态图再报告多时域滚动、守恒和稳定。
模型比较固定图构造、截断、输入特征、参数与计算预算、拆分和超参数选择数据。简单的按元素计数、局部描述符、Laplacian 平滑和无坐标消息传递基线能定位收益来自哪里。均值之外报告每实体结果和不确定性,避免大量相似构象制造虚假的窄区间。
贯穿诊断流程
若能量准确而旋转后的力错误,先查输出是否真正等变、坐标非线性是否破坏向量类型;若训练图好而大分子差,查读出广延性、度尺度、感受野和过压缩;若节点表示层层相同,查过平滑;若两个拓扑类别始终混淆,构造 WL 对照查表达界;若随机拆分远好于骨架拆分,优先查实体泄漏而非加深网络。
每个修复都要对应故障:增加位置编码解决某些全局区分,却不自动解决旋转;加残差缓解优化和过平滑,却不扩大割边容量;全局节点缩短路径,却可能造成新的过压缩。综合推理的目标是让结构选择与证据逐项对应。
练习
查看提示
查看解答
查看提示
查看解答
查看提示
查看解答
查看提示
查看解答
查看提示
查看解答
查看提示
查看解答
关系与资源
- 图与图信号 提供节点、边、Laplacian、任务粒度和单位合同。
- 消息传递神经网络 提供局部消息、聚合、更新和读出。
- 不变性与等变性 约束置换与空间群作用。
- 谱图方法与几何学习 连接 Laplacian 谱、局部多项式与离散几何。
- 图生成与图动力学 连接结构有效性、物理滚动和规模偏移。
Geometric Deep Learning: Grids, Groups, Graphs, Geodesics, and Gauges
Michael M. Bronstein, Joan Bruna, Taco Cohen, Petar Veličković
用于核对 A11 的不变性、等变性、群作用、谱方法和几何深度学习统一框架。
打开官方来源Neural Message Passing for Quantum Chemistry
Justin Gilmer, Samuel S. Schoenholz, Patrick F. Riley, Oriol Vinyals, George E. Dahl
用于核对消息、聚合、更新、读出和分子任务实验的原始定义。
打开官方来源