Robust FCC solute diffusion predictions from ab-initio machine learning methods

Abstract

We evaluate the performance of four machine learning methods for modeling and predicting FCC solute diffusion barriers. More than 200 FCC solute diffusion barriers from previous density functional theory (DFT) calculations served as our dataset to train four machine learning methods: linear regression (LR), decision tree (DT), Gaussian kernel ridge regression (GKRR), and artificial neural network (ANN). We separately optimize key physical descriptors favored by each method to model diffusion barriers. We also assess the ability of each method to extrapolate when faced with new hosts with limited known data. GKRR and ANN were found to perform the best, showing 0.15 eV cross-validation errors and predicting impurity diffusion in new hosts to within 0.2 eV when given only 5 data points from the host. We demonstrate the success of a combined DFT + data mining approach towards solving materials science challenges and predict the diffusion barrier of all available impurities across all FCC hosts.

13 pages, 4 figures

References in corpus (4)

• Scikit-learn: Machine Learning in Python
• A General-Purpose Machine Learning Framework for Predicting Properties of Inorganic Materials
• Machine learning with systematic density-functional theory calculations: Application to melting temperatures of single and binary component solids
• High-Throughput ab-initio Dilute Solute Diffusion Database

Cited by in corpus (10)

• Opportunities and Challenges for Machine Learning in Materials Science
• Developing an improved Crystal Graph Convolutional Neural Network framework for accelerated materials discovery
• A Deep Dive into Machine Learning Density Functional Theory for Materials Science and Chemistry
• The Materials Simulation Toolkit for Machine Learning (MAST-ML): an automated open source toolkit to accelerate data-driven materials research
• KineCluE: a Kinetic Cluster Expansion code to compute transport coefficients beyond the dilute limit
• Exploring effective charge in electromigration using machine learning
• Explainable Machine Learning for Hydrogen Diffusion in Metals and Random Binary Alloys
• Predicting activation energies for vacancy-mediated diffusion in alloys using a transition-state cluster expansion
• Accelerating Ensemble Error Bar Prediction with Single Models Fits
• Explainable Machine Learning for Oxygen Diffusion in Perovskites and Pyrochlores