arXiv:2603.05900cs.LGcs.AI2026-03被引 4

用参考分子引导大模型生成新分子,提升优化效果与稳定性。

Reference-guided Policy Optimization for Molecular Optimization via LLM Reasoning

  • 基于参考分子设计新策略,无需轨迹数据即可训练。
  • 在多个基准上成功率与相似性综合得分显著优于基线方法。
  • 适合需要高效探索新分子结构的研究者或药物设计人员。

大语言模型在推理任务中受益于监督微调(SFT)和可验证奖励的强化学习(RLVR),但在指令驱动的分子优化任务中表现不佳,因每个样本仅提供单一优化后的参考分子,缺乏逐步优化轨迹。本文指出,仅使用参考分子进行答案式SFT会破坏推理过程,而RLVR在相似性约束下反馈稀疏,因模型探索能力弱,导致学习缓慢且优化受限。为此,提出参考引导策略优化(RePO),在不依赖轨迹数据的前提下,从模型中采样候选分子及其中间推理路径,并以满足性质且符合相似性约束的可验证奖励进行强化学习训练。同时,通过保留中间推理作为上下文,仅对最终答案进行监督训练,实现参考引导。该方法在多个分子优化基准上持续优于SFT与RLVR基线(如GRPO),在优化指标(成功率×相似性)上取得提升,更好平衡多目标冲突,并对未见指令风格具有更强泛化能力。代码已开源。

原文摘要 · Abstract (English)

Large language models (LLMs) benefit substantially from supervised fine-tuning (SFT) and reinforcement learning with verifiable rewards (RLVR) in reasoning tasks. However, these recipes perform poorly in instruction-based molecular optimization, where each data point typically provides only a single optimized reference molecule and no step-by-step optimization trajectory. We reveal that answer-only SFT on the reference molecules collapses reasoning, and RLVR provides sparse feedback under similarity constraints due to the model's lack of effective exploration, which slows learning and limits optimization. To encourage the exploration of new molecules while balancing the exploitation of the reference molecules, we introduce Reference-guided Policy Optimization (RePO), an optimization approach that learns from reference molecules without requiring trajectory data. At each update, RePO samples candidate molecules with their intermediate reasoning trajectories from the model and trains the model using verifiable rewards that measure property satisfaction under similarity constraints in an RL manner. Meanwhile, it applies reference guidance by keeping the policy's intermediate reasoning trajectory as context and training only the answer in a supervised manner. Together, the RL term promotes exploration, while the guidance term mitigates reward sparsity and stabilizes training by grounding outputs to references when many valid molecular edits exist. Across molecular optimization benchmarks, RePO consistently outperforms SFT and RLVR baselines (e.g., GRPO), achieving improvements on the optimization metric (Success Rate $\times$ Similarity), improving balance across competing objectives, and generalizing better to unseen instruction styles. Our code is publicly available at https://github.com/tmlr-group/RePO.

分子优化大模型强化学习生成

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