Leveraging transfer learning for accurate estimation of ionic migration barriers in solids

Abstract

Ionic mobility determines the rate performance of several applications, such as batteries, fuel cells, and electrochemical sensors and is exponentially dependent on the migration barrier ($E_m$), a difficult to measure/calculate quantity. Previous approaches to identify materials with high ionic mobility have relied on imprecise descriptors given the lack of generalizable models to predict $E_m$. Here, we present a graph neural network based architecture that leverages principles of transfer learning to efficiently and accurately predict $E_m$ across a diverse set of materials. We use a model pre-trained simultaneously on seven distinct bulk properties (labeled MPT), modify the MPT model to classify different migration pathways in a structure, and fine-tune (FT) on a manually-curated literature-derived dataset of 619 $E_m$ data points calculated with density functional theory. Importantly, our best-performing FT model (labeled MODEL-3) demonstrates substantial improvements in prediction accuracy compared to classical machine learning methods, graph models trained from scratch, and a universal machine learned interatomic potential, with a R${}^2$ score of 0.703 and a mean absolute error of 0.261 eV on the test set. Notably, MODEL-3 is able to distinguish different migration pathways within a structure and also demonstrates excellent ability to generalize across intercalant compositions and chemistries. As a classifier, MODEL-3 exhibits 80\% accuracy and 82.8\% precision in identifying materials that are `good' ionic conductors (i.e., structures with $E_m<$0.65~eV). Thus, our work demonstrates the effective use of FT strategies and architectural modifications necessary for making swift and accurate $E_m$ predictions, which will be useful for materials discovery in batteries and for predicting other data-scarce material properties.

References in corpus (16)

• Crystal Graph Convolutional Neural Networks for an Accurate and Interpretable Prediction of Material Properties
• A Universal Graph Deep Learning Interatomic Potential for the Periodic Table
• Atomistic Line Graph Neural Network for Improved Materials Property Predictions
• Improved initial guess for minimum energy path calculations
• A foundation model for atomistic materials chemistry
• Scalable Parallel Algorithm for Graph Neural Network Interatomic Potentials in Molecular Dynamics Simulations
• Crystallography companion agent for high-throughput materials discovery
• Influence of inversion on Mg mobility and electrochemistry in spinels
• Machine-learning-accelerated simulations to enable automatic surface reconstruction
• A Hitchhiker's Guide to Geometric GNNs for 3D Atomic Systems
• Searching Ternary Oxides and Chalcogenides as Positive Electrodes for Calcium Batteries
• Automatic diffusion path exploration for multivalent battery cathodes using geometrical descriptors
• Optimal pre-train/fine-tune strategies for accurate material property predictions
• Predicting ionic conductivity in solids from the machine-learned potential energy landscape
• Predicting the Volumes of Crystals
• A literature-derived dataset of migration barriers for quantifying ionic transport in battery materials