arXiv:2409.19552cond-mat.mtrl-scics.AI2024-09

用迁移学习加速材料X射线吸收谱预测,提升精度与效率

OmniXAS: A Universal Deep-Learning Framework for Materials X-ray Absorption Spectra

  • 基于M3GNet提取局部化学环境特征,性能优于传统方法
  • 分层迁移学习使跨元素预测准确率提升69%
  • 跨精度迁移可提高11%准确率,适合高通量材料研究

X射线吸收谱(XAS)是探测吸收原子局域化学环境的强大表征技术。然而,分析XAS数据面临巨大挑战,常需大量计算密集型模拟及深厚领域知识,限制了快速、鲁棒的高通量分析流程发展。为此,我们提出OmniXAS框架,包含多种迁移学习方法,显著提升预测准确率与效率。该框架基于三项策略:首先,利用M3GNet获取吸收位点的局域化学环境潜在表示,作为XAS预测输入,相较传统特征化方法性能提升达数量级;其次,采用分层迁移学习,先在多种元素上训练通用多任务模型,再进行元素特异性微调,微调后模型性能相比仅针对单元素训练的模型最高提升69%;第三,实现跨精度迁移学习,将通用模型适配至更高计算成本的仿真生成谱图,预测准确率较仅在目标精度数据上训练的模型提升11%。本方法使XAS建模吞吐量较第一性原理模拟提升数量级,且可扩展至更多元素。该迁移学习框架具有普适性,适用于材料研究中其他性质的深度学习模型增强。

原文摘要 · Abstract (English)

X-ray absorption spectroscopy (XAS) is a powerful characterization technique for probing the local chemical environment of absorbing atoms. However, analyzing XAS data presents significant challenges, often requiring extensive, computationally intensive simulations, as well as significant domain expertise. These limitations hinder the development of fast, robust XAS analysis pipelines that are essential in high-throughput studies and for autonomous experimentation. We address these challenges with OmniXAS, a framework that contains a suite of transfer learning approaches for XAS prediction, each contributing to improved accuracy and efficiency, as demonstrated on K-edge spectra database covering eight 3d transition metals (Ti-Cu). The OmniXAS framework is built upon three distinct strategies. First, we use M3GNet to derive latent representations of the local chemical environment of absorption sites as input for XAS prediction, achieving up to order-of-magnitude improvements over conventional featurization techniques. Second, we employ a hierarchical transfer learning strategy, training a universal multi-task model across elements before fine-tuning for element-specific predictions. Models based on this cascaded approach after element-wise fine-tuning outperform element-specific models by up to 69%. Third, we implement cross-fidelity transfer learning, adapting a universal model to predict spectra generated by simulation of a different fidelity with a higher computational cost. This approach improves prediction accuracy by up to 11% over models trained on the target fidelity alone. Our approach boosts the throughput of XAS modeling by orders of magnitude versus first-principles simulations and is extendable to XAS prediction for a broader range of elements. This transfer learning framework is generalizable to enhance deep-learning models that target other properties in materials research.

XAS预测迁移学习材料建模

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