arXiv:2607.28553cs.LGcs.AI2026-07

无需真实结构标签,用物理一致性自校正预测原子三维结构

APO: Unsupervised Atomic Policy Optimization for 3D Structure Prediction of Atomic Systems

论文配图:APO: Unsupervised Atomic Policy Optimization for 3D Structure Prediction of Atomic Systems
图 1 · 摘自论文原文
  • 设计双奖励机制,通过相似性分解和热力学稳定性引导无监督优化
  • 在晶体与抗体结构预测中超越有监督基线,匹配率与结构保真度达新高
  • 适合数据稀缺场景,如新晶相或全新蛋白的结构建模

预测原子系统的三维结构是推动材料科学和药物发现的基础。尽管流匹配模型(如FlowDPO)近期展现出潜力,但其性能高度依赖于通过有监督偏好学习对齐真实坐标。然而,获取新型晶相或从头蛋白的实验标签成本极高,在数据稀缺情况下形成瓶颈。本文提出APO(原子策略优化),一种完全无监督的对齐框架,无需真实参考结构。APO将群体相对策略优化适配至三维原子环境,采用新颖的双奖励机制:(i) 基于样本相似性矩阵的特征分解,强化模型主导的潜在结构模式;(ii) 引入热力学稳定性奖励。该框架使模型能从采样组中识别出物理上合理的构型并实现自我修正。在晶体与抗体结构预测上的广泛基准测试表明,APO持续优于全有监督基线,在匹配率与结构保真度上达到新纪录。此外,我们证明APO显著改善概率路径的平滑性,大幅提升推理效率。结果表明,内在物理一致性相比噪声较大的坐标对齐,可作为更优的对齐指引。

原文摘要 · Abstract (English)

Predicting the 3D structures of atomic systems is fundamental to advancing material science and drug discovery. While flow-matching models (, FlowDPO) have recently shown promise in this domain, their performance relies heavily on alignment with ground-truth coordinates via supervised preference learning. However, obtaining experimental labels for novel crystal phases or de novo proteins is prohibitively expensive, creating a bottleneck for structural modeling in data-scarce regimes. In this work, we propose (Atomic Policy Optimization), a fully unsupervised alignment framework that eliminates the need for ground-truth reference structures. APO adapts group-relative policy optimization to 3D atomic environments, utilizing a novel dual-reward mechanism: (i) a that reinforces the policy's dominant latent structural modes through eigen-decomposition of sample similarities, and (ii) a that enforces thermodynamic stability. Our framework enables the model to ``self-correct'' by identifying physically plausible configurations within sampled groups. Extensive benchmarks on crystal and antibody structure prediction demonstrate that APO consistently outperforms fully supervised baselines, achieving a new state-of-the-art in match rates and structural fidelity. Furthermore, we show that APO effectively straightens probability paths, significantly improving inference efficiency. Our results suggest that intrinsic physical consistency can serve as a superior guide for alignment compared to noisy, supervised coordinate matching.

结构预测无监督学习原子系统物理引导

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。