End-to-end Material Thermal Conductivity Prediction through Machine Learning

Yagyank Srivastava,Ankit Jain
2023-11-06
Abstract:We investigated the accelerated prediction of the thermal conductivity of materials through end- to-end structure-based approaches employing machine learning methods. Due to the non-availability of high-quality thermal conductivity data, we first performed high-throughput calculations based on first principles and the Boltzmann transport equation for 225 materials, effectively more than doubling the size of the existing dataset. We assessed the performance of state-of-the-art machine learning models for thermal conductivity prediction on this expanded dataset and observed that all these models suffered from overfitting. To address this issue, we introduced a novel graph-based neural network model, which demonstrated more consistent and regularized performance across all evaluated datasets. Nevertheless, the best mean absolute percentage error achieved on the test dataset remained in the range of 50-60%. This suggests that while these models are valuable for expediting material screening, their current accuracy is still limited.
Materials Science,Machine Learning
What problem does this paper attempt to address?
The paper attempts to address several key challenges in predicting the thermal conductivity of materials: 1. **Data Scarcity**: High-quality thermal conductivity data is scarce, which limits the performance of machine learning models in predicting the thermal conductivity of materials. 2. **Overfitting**: Existing machine learning models are prone to overfitting on small datasets, resulting in poor generalization ability. 3. **Prediction Accuracy**: Although machine learning methods can accelerate the material screening process, the current prediction accuracy is still limited, especially on high-precision datasets. To address these issues, the authors have undertaken the following work: - First, they expanded the existing high-quality thermal conductivity dataset by more than doubling its size through large-scale high-throughput calculations using first-principles calculations and the Boltzmann transport equation. - Next, they developed a new model based on graph neural networks to reduce the overfitting problem in predicting the thermal conductivity of materials and improved the model's generalization performance across different datasets. Overall, this study aims to enhance the accuracy and reliability of thermal conductivity predictions of materials by augmenting the dataset and improving machine learning models.