Multi-Task Regression Model for Predicting Photocatalytic Performance of Inorganic Materials

As renewable energy technologies advance, identifying efficient photocatalytic materials for water splitting to produce hydrogen has become an important research focus in materials science. This study presents a multi-task regression model (MTRM) designed to predict the conduction band minimum (CBM), valence band maximum (VBM), and solar-to-hydrogen efficiency (STH) of inorganic materials. Utilizing crystallographic and band gap data from over 15,000 materials in the SNUMAT database, machine-learning methods are applied to predict CBM and VBM, which are subsequently used as additional features to estimate STH. A deep neural network framework with a multi-branch, multi-task regression structure is employed to address the issue of error propagation in traditional cascading models by enabling feature sharing and joint optimization of the tasks. The calculated results show that, while traditional tree-based models perform well in single-task predictions, MTRM achieves superior performance in the multi-task setting, particularly for STH prediction, with an MSE of 0.0001 and an R2 of 0.8265, significantly outperforming cascading approaches. This research provides a new approach to predicting photocatalytic material performance and demonstrates the potential of multi-task learning in materials science.