Multimodal machine learning for materials science: composition-structure bimodal learning for experimentally measured properties

Sheng Gong,Shuo Wang,Taishan Zhu,Yang Shao-Horn,Jeffrey C. Grossman
2023-08-04
Abstract:The widespread application of multimodal machine learning models like GPT-4 has revolutionized various research fields including computer vision and natural language processing. However, its implementation in materials informatics remains underexplored, despite the presence of materials data across diverse modalities, such as composition and structure. The effectiveness of machine learning models trained on large calculated datasets depends on the accuracy of calculations, while experimental datasets often have limited data availability and incomplete information. This paper introduces a novel approach to multimodal machine learning in materials science via composition-structure bimodal learning. The proposed COmposition-Structure Bimodal Network (COSNet) is designed to enhance learning and predictions of experimentally measured materials properties that have incomplete structure information. Bimodal learning significantly reduces prediction errors across distinct materials properties including Li conductivity in solid electrolyte, band gap, refractive index, dielectric constant, energy, and magnetic moment, surpassing composition-only learning methods. Furthermore, we identified that data augmentation based on modal availability plays a pivotal role in the success of bimodal learning.
Materials Science,Artificial Intelligence,Machine Learning
What problem does this paper attempt to address?
### Problems Addressed by the Paper The paper primarily addresses the issue in materials science where experimental datasets often have complete compositional information but incomplete structural information. Specifically, the paper proposes a novel approach to improve the prediction accuracy of experimentally measured material properties by combining compositional and structural bimodal learning. #### Main Contributions 1. **Proposed a Combined Composition-Structure Bimodal Learning Framework**: To improve the learning performance of datasets with incomplete structural information, the researchers proposed the COmposition-Structure Bimodal Network (COSNet) framework. 2. **Demonstrated the Importance of Data Augmentation for Bimodal Learning Improvement**: The study found that data augmentation based on modality availability is crucial for improving bimodal learning, as it ensures that the combined network can fully utilize all available data for training. #### Experimental Results - For four experimentally measured material properties (lithium conductivity, band gap, refractive index, dielectric constant), COSNet reduced the prediction error by approximately 7% to 10% on all complete datasets compared to models using only compositional information, and these improvements were statistically significant. - This improvement was observed not only in data points with structural information but also in those without structural information, indicating that the structural information of some data points can enhance the prediction of other data points lacking structural information. #### Method Overview - The study used four experimental datasets (lithium conductivity, band gap, refractive index, dielectric constant) and two theoretical datasets (total magnetic moment per chemical formula and energy per atom) to test the effectiveness of bimodal learning and COSNet. - In terms of models, ROOST and de-CGCNN were chosen as the compositional and structural networks in COSNet, based on their strong performance on small datasets. - Data augmentation strategies ensured that the combined network could be fully trained, enhancing its predictive performance even for data points without structural information. In summary, the paper aims to improve the prediction of material properties by combining compositional and structural information and demonstrates the effectiveness of this approach on experimental datasets, particularly excelling in cases with incomplete structural information.