Evaluating explorative prediction power of machine learning algorithms for materials discovery using k-fold forward cross-validation

By Zheng Xiong, Yuxin Cui, Zhonghao Liu, Yong Zhao, Ming Hu, Jianjun Hu. Computational materials science, 2020

300 citations recorded by the source corpus.

Abstract The materials discovery problem usually aims to identify novel “outlier” materials with extremely low or high property values outside of the scope of all known materials. It can be mapped as an explorative prediction problem. However, currently the performance of machine learning algorithms for materials property prediction is usually evaluated via k-fold cross-validation (CV) or holdout-test, which tend to over-estimate their explorative prediction performance in discovering novel materials. We propose k -fold- m -step forward cross-validation ( km FCV) as a new way for evaluating exploration performance in materials property prediction and conducted a comprehensive benchmark evaluation on the exploration performance of a variety of prediction models on materials property (including formation energy, band gap, and superconducting critical temperature) prediction with different materials representation and machine learning algorithms. Our results show that even though current machine learning models can achieve good results when evaluated with traditional CV, their explorative power is actually very low as shown by our proposed km FCV evaluation method and the proposed exploration accuracy. More advanced explorative machine learning algorithms are strongly needed for new materials discovery.

Subfields: Computational Theory and Mathematics, Materials Chemistry

View Paper: https://www.semanticscholar.org/paper/a8a18263630c853684a5c99818ac9ff52c30c39c