用思维推理和强化学习生成符合物理规律的晶体结构。
CrystalReasoner: Reasoning and RL for Property-Conditioned Crystal Structure Generation

- 引入物理先验作为思考令牌,连接自然语言与三维结构。
- 通过多目标强化学习提升结构有效性、化学一致性和热力学稳定性。
- 支持属性条件生成,适配材料设计与逆向工程研究者。
生成建模已成为晶体结构发现的有前途方法。然而,现有基于大模型的生成方法在原子级精度上表现不足,而基于扩散的方法难以融合高层次科学知识,导致生成结构常无效、不稳定或缺乏理想属性。为此,我们提出CrystalReasoner(CrysReas),一种端到端的大模型框架,通过推理与对齐从自然语言指令生成晶体结构。CrysReas引入晶体学对称性、局部配位环境及预测物理性质等物理先验作为思考令牌,在生成原子坐标前进行推理,弥合自然语言与3D结构之间的差距。随后,采用多目标密集奖励函数的强化学习,使生成结果满足物理有效性、化学一致性与热力学稳定性。针对属性条件生成任务,设计特定奖励函数,并训练专用模型以处理离散约束(如空间群)与连续属性(如弹性、热膨胀系数)。实验表明,相比先前工作及无思考轨迹或强化学习的基线,CrysReas在多种指标上表现更优,三倍提升S.U.N.比率,且在属性条件生成中表现更佳。CrysReas还表现出自适应推理能力,随着原子数增加自动延长推理长度。本工作展示了利用思考轨迹与强化学习生成有效、稳定且属性可控晶体结构的潜力。
原文摘要 · Abstract (English)
Generative modeling has emerged as a promising approach for crystal structure discovery. However, existing LLM-based generative models struggle with low-level atomic precision, while diffusion-based methods fall short in integrating high-level scientific knowledge. As a result, generated structures are often invalid, unstable, or do not possess desirable properties. To address this gap, we propose CrystalReasoner (CrysReas), an end-to-end LLM framework that generates crystal structures from natural language instructions through reasoning and alignment. CrysReas introduces physical priors as thinking tokens, which include crystallographic symmetry, local coordination environments and predicted physical properties before generating atomic coordinates. This bridges the gap between natural language and 3D structures. CrysReas then employs reinforcement learning (RL) with a multi-objective, dense reward function to align generation with physical validity, chemical consistency, and thermodynamic stability. For property-conditioned tasks, we design task-specific reward functions and train specialized models for discrete constraints (e.g., space group) and continuous properties (e.g., elasticity, thermal expansion). Empirical results demonstrate that compared to prior works and baselines without thinking traces or RL, CrysReas obtains better performance on diverse metrics, triples S.U.N. ratio, and achieves better performance for property conditioned generation. CrysReas also exhibits adaptive reasoning, increasing reasoning lengths as the number of atoms increases. Our work demonstrates the potential of leveraging thinking traces and RL for generating valid, stable, and property-conditioned crystal structures.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。