Suiren-1.0 技术报告:分子基础模型家族
化学物理
2026-04-29 v4 人工智能
摘要
我们介绍 Suiren-1.0,这是一套用于准确建模多样有机系统的分子基础模型。Suiren-1.0 由三个专门变体组成(Suiren-Base、Suiren-Dimer 和 Suiren-ConfAvg),并集成在一种算法框架中,将 3D 构象几何与 2D 统计集合空间之间的鸿沟连接起来。我们首先在 7000 万样本的密度泛函理论数据集上对 Suiren-Base(18 亿参数)进行预训练,采用空间自监督和 SE(3) 不变性架构,实现了对量子性质预测的稳健性能。Suiren-Dimer 通过在 1350 万分子相互作用样本上继续预训练来扩展了这一能力。为实现高效的下游应用,我们提出了构象压缩蒸馏(CCD),这是一种基于扩散的框架,将复杂的 3D 结构表示蒸馏到 2D 构象平均表示中。这产生了轻量级的 Suiren-ConfAvg 版本,可以从 SMILES 或分子图生成高保真表示。我们的广泛评估表明,Suiren-1.0 在一系列任务上建立了最先进的结果。所有模型和基准均已开源。
引用
@article{arxiv.2603.21942,
title = {Suiren-1.0 Technical Report: A Family of Molecular Foundation Models},
author = {Junyi An and Xinyu Lu and Yun-Fei Shi and Li-Cheng Xu and Nannan Zhang and Chao Qu and Yuan Qi and Fenglei Cao},
journal= {arXiv preprint arXiv:2603.21942},
year = {2026}
}
备注
24 pages,5 figures