Blitz：加速 AlphaFold3 和 Boltz-2 的扩散模块

arXiv:2610.23703v1 公告类型：新 摘要：基于扩散的蛋白质结构预测模型通常需要约 200 个去噪步骤来生成全原子结构，而这一成本推动了多种少步采样器的开发。我们提出 Blitz，这是一种少步蒸馏框架，采用几何噪声调度，使各步骤在不同噪声尺度间分布得更加均匀，随后通过多步一致性训练学习这些更新。在推理过程中，物理引导和无需目标的候选筛选有助于减少残留的几何缺陷。我们将 Blitz 应用于 AlphaFold 3 和 Boltz-2，将去噪步骤从 200 步减少到 8 步。由于已发表的少步结果采用了不同的基准和评分流程，我们在统一的评估协议下，将这两个蒸馏模型与其教师模型及其他少步方法进行比较，评估内容涵盖结构准确性、界面质量、配体定位和立体化学有效性。Blitz 保持了教师模型级别的结构准确性，其中 AF3-Blitz 的表现优于所有参与评估的少步基线方法，并在多个准确性指标上超过其教师模型。