用推理时强化学习让材料生成模型自动优化目标性能。
Open Materials Generation with Inference-Time Reinforcement Learning
- 直接操作生成模型的速率场,无需计算梯度得分
- 实现能量目标优化,采样效率提升十倍以上
- 适合需要高效生成稳定晶体结构的研究者
针对晶态材料的连续时间生成模型在逆向材料设计中可预测稳定晶体结构,但如何将明确目标属性融入生成过程仍具挑战。策略梯度强化学习(RL)为对齐生成模型与下游目标提供了理论框架,但通常需依赖得分信息,这限制了其在仅学习速率场的流模型中的应用。本文提出推理时强化学习的开放材料生成方法(OMatG-IRL),直接作用于学习到的速率场,无需显式计算得分。该方法通过随机扰动生成动态,在不降低预训练模型基线性能的前提下,实现在推理阶段的探索与策略梯度估计。首次将强化学习应用于晶体结构预测(CSP)。所提方法有效强化能量目标,同时通过组分条件保持多样性,性能媲美基于得分的强化学习方法。此外,证明可学习随时间变化的速率退火调度,使采样效率提升一个数量级,生成时间显著减少。相关代码已集成至开源材料生成框架 OMatG v2.0,项目地址:https://github.com/FERMat-ML/OMatG。
原文摘要 · Abstract (English)
Continuous-time generative models for crystalline materials enable inverse materials design by learning to predict stable crystal structures, but incorporating explicit target properties into the generative process remains challenging. Policy-gradient reinforcement learning (RL) provides a principled mechanism for aligning generative models with downstream objectives but typically requires access to the score, which has prevented its application to flow-based models that learn only velocity fields. We introduce Open Materials Generation with Inference-time Reinforcement Learning (OMatG-IRL), a policy-gradient RL framework that operates directly on the learned velocity fields and eliminates the need for the explicit computation of the score. OMatG-IRL leverages stochastic perturbations of the underlying generation dynamics preserving the baseline performance of the pretrained generative model while enabling exploration and policy-gradient estimation at inference time. Using OMatG-IRL, we present the first application of RL to crystal structure prediction (CSP). Our method enables effective reinforcement of an energy-based objective while preserving diversity through composition conditioning, and it achieves performance competitive with score-based RL approaches. Finally, we show that OMatG-IRL can learn time-dependent velocity-annealing schedules, enabling accurate CSP with order-of-magnitude improvements in sampling efficiency and, correspondingly, reduction in generation time. The OMatG-IRL code is included in a new release of the Open Materials Generation (OMatG) framework available at https://github.com/FERMat-ML/OMatG.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。