An Interpretable Hybrid Machine Learning Prediction of Dielectric Constant of Alkali Halide Crystals

Jiandong Deng,Guozhu Jia
DOI: https://doi.org/10.1016/j.chemphys.2022.111457
IF: 2.552
2022-01-01
Chemical Physics
Abstract:Exploring the data-driven prediction strategy of physical and chemical properties is attractive for the rational design of crystal dielectrics with target characteristics, especially for the optoelectronics and electronic devices. To accelerate the design and discovery of novel crystal based dielectric devices, this work proposed an interpretable hybrid machine learning (ML) model to predict the dielectric constant of halide alkali crystals. The approach uses Particle Swarm Optimization-Random Forest (PSO-RF) model and polarization ion parameters of molecular dynamics simulation accurately predict the dielectric constant of 19 alkali halides on a small data set. The predicted performance of the PSO-RF model reaches R-2 of 0.987, which exceeds other seven popular machine learning models. The PSO-RF model is interpreted by feature importance and partial dependence plots. This work allows for data expansion using experimental data and molecular structure information, provides ideas for predicting the physical and chemical properties of materials on small data sets.
What problem does this paper attempt to address?