How do Co-Folding Models Organize Structural Information?

技术摘要：共折叠模型如何组织结构信息？

问题陈述
尽管共折叠模型（如 AlphaFold3、Boltz-1）在从序列预测生物分子复合物结构方面取得了极高的准确度，但其组织和处理结构信息的内部机制仍不明确。具体而言，目前尚不清楚这些模型如何区分链内折叠（intra-chain folding）与链间对接（inter-chain docking），如何跨网络层构建成对距离信息，以及“躯干”（即表示生成器）是否在将信息传递给扩散模块之前，已充分解析了由配体结合诱导的构象变化。与单链折叠不同，共折叠引入了在单体系统中没有直接类比的链间对块（inter-chain pair blocks），这使得现有的可解释性方法难以适用。

研究方法
作者通过将对表示进行链感知分解（将其分解为链内 $Z_{intra}$ 和链间 $Z_{inter}$ 块）的方法，分析了“Boltz-1”的躯干架构。本研究采用了四种分析视角：

1. 消融实验（Ablation）： 将单表示（$S$）、$Z_{intra}$ 和 $Z_{inter}$ 中的激活值置零、取平均值或替换为高斯噪声。通过由冻结的扩散模块进行解码，以衡量这些修改对链局部几何、链间排列以及蛋白质-配体接触的影响。
2. 表示几何（Representation Geometry）： 计算每一层和每一次循环迭代中激活矩阵的奇异值统计量（矩阵熵、稳定秩、前 1 个奇异值质量），以表征信息流（混合—压缩—精炼轨迹）。
3. 线性探测（Linear Probing）： 对隐藏激活值进行低容量线性解码器拟合，以量化不同深度下 $Z_{intra}$ 和 $Z_{inter}$ 中成对 $C\beta$–$C\beta$ 距离的线性可及性。
4. 激活修补（Activation Patching）： 将“apo”（配体未结合）和“holo”（配体结合）推理运行中的激活值进行交换，以测试躯干是否完全编码了配体诱导的骨架变化，或者扩散模块是否补偿了这些不一致性。

关键结果

• 表示的功能特化：
  • 单表示 ($S$)： 包含标记内部的原子细节。消融 $S$（同时保持统计特性）会维持链级几何结构，但会导致化学性质无效的局部结构（例如，错误的脯氨酸环几何形状和侧链距离）。
  • 链内对 ($Z_{intra}$)： 编码链级几何结构。消融会导致链级 LDDT 发生灾难性下降（约 70–78%），表明该流是链内折叠信息的主要载体。
  • 链间对 ($Z_{inter}$)： 控制相对分子排列。消融会保留单个链的折叠，但会破坏界面 LDDT 和蛋白质-配体接触质量，证实了其在对接中的作用。
• 成对信息的构建：
  • 非对称预处理（Asymmetric Preconditioning）： 链内距离内容主要由 MSA 模块在过程早期提供（在进入 Pairformer 前 $R^2 \approx 0.3$，在循环后达到 $\approx 0.8$）。相比之下，链间距离内容是在每一次循环迭代中通过 Pairformer 层逐步构建的，初始相关性接近于零。
  • 混合—压缩—精炼 (MCR) 轨迹： Pairformer 躯干遵循类似于大型语言模型的 MCR 轨迹：MSA 模块混合信息（熵下降），中间层压缩表示，随后的层向扩散交付阶段重新扩张。至关重要的是，即使在压缩阶段，距离可及性也能得到保持。
• 配体诱导的变化：
  • 修补实验表明，躯干在扩散阶段之前并未完全解析配体诱导的骨架变化。当将 holo 状态的 $Z_{intra}$ 修补到 apo 表示中时，扩散模块必须协调部分不一致的链内与链间约束，以生成正确的结构。

核心贡献

1. 信息流的分解： 本文证明了共折叠模型将不同方面的特征编码在三个独立的流中：$S$ 中的局部原子细节、$Z_{intra}$ 中的链级几何以及 $Z_{inter}$ 中的相对取向。
2. 对表示的深度剖析： 作者绘制了成对距离信息的演化图谱，显示出链内信号是经过 MSA 饱和并精炼的，而链间信号则是跨越整个躯干深度被积极构建的。
3. 躯干与扩散的交互： 研究提供了证据，表明躯干的表示在处理大规模构象变化时可能是不完整的，扩散模块在协调躯干尚未完全解决的约束方面发挥着关键作用。

意义
这项工作为共折叠躯干如何组织结构信息提供了机械论视角，超越了黑盒性能指标。通过识别出链间信息是逐步构建的，且躯干可能无法完全解析配体诱导的变化，研究结果表明，未来的生物分子结构预测模型可以受益于“分区感知”（partition-aware）的设计。此类设计可能会显式地分离处理链内共进化信号与链间相互作用构建的过程，从而潜在地提高效率和可解释性。