Journal of Inorganic Materials ›› 2021, Vol. 36 ›› Issue (11): 1178-1184.DOI: 10.15541/jim20200748

• RESEARCH ARTICLE • Previous Articles     Next Articles

Training Model for Predicting Adsorption Energy of Metal Ions Based on Machine Learning

ZHANG Ruihong1(), WEI Xin2, LU Zhanhui1, AI Yuejie3()   

  1. 1. College of Mathematics and Physics, North China Electric Power University, Beijing 102206, China
    2. College of Control and Computer Engineering, North China Electric Power University, Beijing 102206, China
    3. MOE Key Laboratory of Resources and Environmental System Optimization, College of Environmental Science and Engineering, North China Electric Power University, Beijing 102206, China
  • Received:2020-12-31 Revised:2021-04-15 Published:2021-11-20 Online:2021-06-01
  • Contact: AI Yuejie, associate professor. E-mail:
  • About author:ZHANG Ruihong(1996-), femal, Master candidate. E-mail:
  • Supported by:
    National Natural Science Foundation of China(22076044);National Key Research and Development Program of China(2017YFA0207002);Fundamental Research Funds for the Central Universities(2017YQ001)


The adsorption behavior of graphene oxide and metal ions was simulated theoretically by density functional theory. In the process of training the prediction model based on the machine learning method, the missing values were processed by matrix completion method, which was widely used in the recommendation systems, and gradient boosting machine (GBM) was trained to explain the importance of factors that affect the adsorption energy. The result showed that nine properties of the adsorption, namely ionic radius, zero-point vibration energy, Mulliken charge, boiling point, dipole moment, atomic weight, molar heat capacity at constant volume (CV), spin multiplicity and bond length, were found to provide 90% importance of the cumulative adsorption energy. Then six regression methods, including support vector regression, ridge regression, random forest, extremely randomized trees, extreme gradient boosting, and light gradient boosting machine, were used to quantitatively evaluate the prediction accuracy. The results showed that machine learning could provide sufficient accuracy to predict adsorption energy. Among them, extremely randomized trees displayed the best prediction performance, with a mean square error only 0.075. Furthermore, the trained model was tested in a system of vanillin adsorbing metal ions, verifying the feasibility of training the prediction model of adsorption energy based on machine learning. But it is still necessary to be further improved. In general, this research takes the advantage of machine learning on the basis of saving experimental time to provide an instructive reference for theoretical research on metal ion removal.

Key words: machine learning, density functional theory, adsorption energy, metal ions, extremely randomized trees

CLC Number: