MDAgent2:用于分子动力学代码生成与知识问答的大语言模型 / MDAgent2: Large Language Model for Code Generation and Knowledge Q&A in Molecular Dynamics
1️⃣ 一句话总结
这篇论文提出了一个名为MDAgent2的端到端框架,它通过构建高质量数据集和创新的训练方法,训练出专门用于分子动力学领域的大语言模型,不仅能回答专业问题,还能自动生成可执行的模拟代码,从而显著降低了进行复杂科学仿真的技术门槛。
Molecular dynamics (MD) simulations are essential for understanding atomic-scale behaviors in materials science, yet writing LAMMPS scripts remains highly specialized and time-consuming tasks. Although LLMs show promise in code generation and domain-specific question answering, their performance in MD scenarios is limited by scarce domain data, the high deployment cost of state-of-the-art LLMs, and low code executability. Building upon our prior MDAgent, we present MDAgent2, the first end-to-end framework capable of performing both knowledge Q&A and code generation within the MD domain. We construct a domain-specific data-construction pipeline that yields three high-quality datasets spanning MD knowledge, question answering, and code generation. Based on these datasets, we adopt a three stage post-training strategy--continued pre-training (CPT), supervised fine-tuning (SFT), and reinforcement learning (RL)--to train two domain-adapted models, MD-Instruct and MD-Code. Furthermore, we introduce MD-GRPO, a closed-loop RL method that leverages simulation outcomes as reward signals and recycles low-reward trajectories for continual refinement. We further build MDAgent2-RUNTIME, a deployable multi-agent system that integrates code generation, execution, evaluation, and self-correction. Together with MD-EvalBench proposed in this work, the first benchmark for LAMMPS code generation and question answering, our models and system achieve performance surpassing several strong this http URL work systematically demonstrates the adaptability and generalization capability of large language models in industrial simulation tasks, laying a methodological foundation for automatic code generation in AI for Science and industrial-scale simulations. URL: this https URL
MDAgent2:用于分子动力学代码生成与知识问答的大语言模型 / MDAgent2: Large Language Model for Code Generation and Knowledge Q&A in Molecular Dynamics
这篇论文提出了一个名为MDAgent2的端到端框架,它通过构建高质量数据集和创新的训练方法,训练出专门用于分子动力学领域的大语言模型,不仅能回答专业问题,还能自动生成可执行的模拟代码,从而显著降低了进行复杂科学仿真的技术门槛。
源自 arXiv: 2601.02075