Adaptive edge-aware graph convolutional with multi-task learning for simultaneous prediction of material properties
The targeted design of functional materials often requires the concurrent optimization of multiple interdependent properties. For boron-doped graphene (BDG), both the band gap and work function critically influence performance in electronic and catalytic applications, yet existing machine learning (ML) approaches typically focus on single-property prediction and rely on hand-crafted features, limiting their generality. Here we present an adaptive edge-aware graph convolutional neural network with multi-task learning (AEGCNN-MTL) for simultaneous prediction of multiple material properties. On a DFT-computed BDG dataset of 2613 structures, AEGCNN-MTL achieved high accuracy (R² = 0.9905 for band gap and 0.9778 for work function), and under identical training budgets, outperformed representative single-task GNN baselines. When transferred to the QM9 benchmark, the framework delivered competitive performance across 12 diverse quantum chemical properties, demonstrating strong generalization capability. These results highlight the potential of AEGCNN-MTL as a scalable and accurate tool for high-throughput, multi-property screening and the data-driven discovery of multifunctional materials.
Authors
- Yunhua Lu
- Chao Zhang (ORCID: https://orcid.org/0000-0002-4588-4036)
- Qiuyan Bi (ORCID: https://orcid.org/0000-0002-7117-8290)
- Shiai XU
- Qingwei Zhang
- Junan Zhang
- Mingyue Chen
Institutions
- Qinghai University (CN)
- East China University of Science and Technology (CN)
- Chongqing University of Technology (CN)
Publication Details
- Journal
- npj Computational Materials
- Published
- 2025-12-24
- DOI
- https://doi.org/10.1038/s41524-025-01917-7
- Primary Topic
- Machine Learning in Materials Science
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Natural Science Foundation of China
- Ministry of Industry and Information Technology of the People's Republic of China