arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2602.13136cs.LG
收藏

retrosynthesis 中的顺序至关重要:通过反应中心引导的离散流匹配实现结构感知生成

Order Matters in Retrosynthesis: Structure-aware Generation via Reaction-Center-Guided Discrete Flow Matching

School of Data Science, The Chinese University of Hong Kong, Shenzhen(数据科学学院,香港中文大学(深圳));Shanghai Artificial Intelligence Laboratory(上海人工智能实验室)

Chenguang Wang, Zihan Zhou, Lei Bai, Tianshu Yu

AI总结:

本研究提出了一种结构感知的模板自由框架,通过反应中心引导的离散流匹配实现高效的 retrosynthesis 生成。

摘要:

模板自由的 retrosynthesis 方法将任务视为黑盒序列生成,限制了学习效率,而半模板方法依赖于刚性的反应库,限制了泛化能力。我们通过一个关键洞察来解决这一差距:神经表示中的原子顺序很重要。基于这一洞察,我们提出了一种结构感知的模板自由框架,将化学反应的两阶段性质编码为位置诱导偏差。通过将反应中心原子置于序列头部,我们的方法将隐含的化学知识转化为显式的位置模式,模型可以轻易捕捉。所提出的 RetroDiT 主干,一种带有旋转位置嵌入的图Transformer,利用这种顺序优先考虑化学关键区域。结合离散流匹配,我们的方法解耦了训练和采样,并使生成在20-50步内完成,而先前的扩散方法需要500步。我们的方法在USPTO-50k(61.2% top-1)和大规模USPTO-Full(51.3% top-1)上均取得了最先进的性能,使用预测的反应中心。在使用oracle中心时,性能分别达到71.1%和63.4%,超越了在100亿次反应上训练的基础模型,同时使用了数量级更少的数据。消融研究进一步表明,结构先验优于暴力扩展:一个280K参数模型在适当顺序下与一个65M参数模型无顺序时表现相当。

英文摘要:

Template-free retrosynthesis methods treat the task as black-box sequence generation, limiting learning efficiency, while semi-template approaches rely on rigid reaction libraries that constrain generalization. We address this gap with a key insight: atom ordering in neural representations matters. Building on this insight, we propose a structure-aware template-free framework that encodes the two-stage nature of chemical reactions as a positional inductive bias. By placing reaction center atoms at the sequence head, our method transforms implicit chemical knowledge into explicit positional patterns that the model can readily capture. The proposed RetroDiT backbone, a graph transformer with rotary position embeddings, exploits this ordering to prioritize chemically critical regions. Combined with discrete flow matching, our approach decouples training from sampling and enables generation in 20--50 steps versus 500 for prior diffusion methods. Our method achieves state-of-the-art performance on both USPTO-50k (61.2% top-1) and the large-scale USPTO-Full (51.3% top-1) with predicted reaction centers. With oracle centers, performance reaches 71.1% and 63.4% respectively, surpassing foundation models trained on 10 billion reactions while using orders of magnitude less data. Ablation studies further reveal that structural priors outperform brute-force scaling: a 280K-parameter model with proper ordering matches a 65M-parameter model without it.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑