LLM2D
MatterTune: 一个集成的、用户友好的平台,用于微调原子基础模型以加速材料模拟与发现
MatterTune: An Integrated, User-Friendly Platform for Fine-Tuning Atomistic Foundation Models to Accelerate Materials Simulation and Discovery
作者: Lingyu Kong, Nima Shoghi, Guoxiang Hu, Pan Li, Victor Fung
发布日期: 4/16/2025
arXiv ID: oai:arXiv.org:2504.10655v1

摘要

arXiv:2504.10655v1 双语标题: cross 摘要:几何机器学习模型,如图神经网络,在化学和材料科学领域,例如高通量虚拟筛选和原子尺度模拟中,近年来取得了显著的成功。这些模型的成功可以归因于它们能够直接从训练数据中有效地学习原子结构的潜在表示。相反,这也导致了这些模型对大量数据的高需求,限制了它们在数据稀疏问题上的应用,而在这些领域中,数据稀疏问题是常见的。为了解决这一限制,预训练机器学习模型的发展正在增长,这些模型学会了原子数据中的通用、基本的几何关系,并可以进行微调以适应更小的应用特定数据集。特别是,在多样性和大规模原子数据集上进行预训练的模型显示出了出色的泛化能力和对下游应用的灵活性,并越来越多地被称为原子基础模型。为了充分利用这些基础模型的潜力,我们引入了MatterTune,这是模块化和可扩展的框架,提供了高级微调能力,并无缝地将原子基础模型集成到下游材料信息学和模拟工作流中,从而降低了采用门槛,并促进了材料科学中多样化的应用。目前,MatterTune 支持包括 ORB、MatterSim、JMP 和 EquformerV2 在内的多项先进基础模型,并包括模块化和灵活的设计、分布式和可定制的微调、广泛的下游信息任务支持等。