用智能体决策模拟原子扩散,兼顾物理准确、可解释与大规模计算。
SwarmThinkers: Learning Physically Consistent Atomic KMC Transitions at Scale
- 将原子视为决策智能体,通过共享策略网络学习转移路径。
- 单张A100 GPU实现全尺度模拟,速度提升3185倍,内存降低485倍。
- 适合需要高保真、可解释的材料模拟研究者使用。
科学仿真系统能否同时具备物理一致性、可解释性与跨尺度可扩展性?尽管数十年进展,这一目标仍难以实现。经典方法如动力学蒙特卡洛(KMC)保证热力学准确性但扩展性差;基于学习的方法虽高效,却常牺牲物理一致性和可解释性。本文提出SwarmThinkers,一种强化学习框架,将原子尺度模拟重构为基于物理的群体智能系统。每个扩散粒子被建模为局部决策智能体,通过共享策略网络在热力学约束下选择转移路径。重加权机制融合学习偏好与转移速率,保持统计保真度的同时支持可解释的逐步决策。训练采用集中式训练、分布式执行范式,使策略无需重新训练即可泛化至不同体系尺寸、浓度与温度。在辐射诱导铁铜合金析出的基准测试中,SwarmThinkers首次在单张A100 GPU上实现全尺度物理一致模拟,此前仅能通过OpenKMC在超级计算机上完成。其计算速度最高提升4963倍(平均3185倍),内存占用降低485倍。通过将粒子视为决策主体而非被动采样者,SwarmThinkers标志着科学模拟范式的根本转变——以智能体驱动的智能,统一物理一致性、可解释性与可扩展性。
原文摘要 · Abstract (English)
Can a scientific simulation system be physically consistent, interpretable by design, and scalable across regimes--all at once? Despite decades of progress, this trifecta remains elusive. Classical methods like Kinetic Monte Carlo ensure thermodynamic accuracy but scale poorly; learning-based methods offer efficiency but often sacrifice physical consistency and interpretability. We present SwarmThinkers, a reinforcement learning framework that recasts atomic-scale simulation as a physically grounded swarm intelligence system. Each diffusing particle is modeled as a local decision-making agent that selects transitions via a shared policy network trained under thermodynamic constraints. A reweighting mechanism fuses learned preferences with transition rates, preserving statistical fidelity while enabling interpretable, step-wise decision making. Training follows a centralized-training, decentralized-execution paradigm, allowing the policy to generalize across system sizes, concentrations, and temperatures without retraining. On a benchmark simulating radiation-induced Fe-Cu alloy precipitation, SwarmThinkers is the first system to achieve full-scale, physically consistent simulation on a single A100 GPU, previously attainable only via OpenKMC on a supercomputer. It delivers up to 4963x (3185x on average) faster computation with 485x lower memory usage. By treating particles as decision-makers, not passive samplers, SwarmThinkers marks a paradigm shift in scientific simulation--one that unifies physical consistency, interpretability, and scalability through agent-driven intelligence.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。