Leveraging multi-task learning regressor chains for small and sparse tabular data in materials design

Publikation: Beitrag in FachzeitschriftForschungsartikelBeigetragenBegutachtung

Abstract

Machine learning has become increasingly important in materials design, yet traditional single-task learning (STL) models fail to fully exploit the potential of available data in scenarios involving multiple targets and incomplete datasets. While STL models overlook the inherent correlations between target properties, this study showcases how multi-task learning (MTL) effectively leverages these correlations. Therefore, the performance of MTL methods compared to STL is evaluated across five datasets, covering twelve prediction tasks and incorporating different types and levels of data sparsity. Our findings reveal that MTL significantly outperforms STL, particularly in sparse data scenarios, with up to 15% prediction improvements across all tasks. Moreover, MTL methods utilizing regressor chains with automated machine learning tools achieve superior performance compared to those based on neural networks, with minimal training effort required. This work advances data efficiency in data-driven materials design, establishing MTL as a potent tool for simultaneous learning and predicting multiple material properties.

Details

OriginalspracheEnglisch
Aufsatznummer015045
Seitenumfang18
FachzeitschriftMachine learning: science and technology
Jahrgang6 (2025)
Ausgabenummer1
Frühes Online-Datum20 Feb. 2025
PublikationsstatusVeröffentlicht - März 2025
Peer-Review-StatusJa

Externe IDs

ORCID /0000-0001-7540-4235/work/178928749
unpaywall 10.1088/2632-2153/adae53
Mendeley 306bb017-a7c0-3478-b941-1b252b1e1bce
Scopus 85218640429

Schlagworte

Forschungsprofillinien der TU Dresden

DFG-Fachsystematik nach Fachkollegium

Fächergruppen, Lehr- und Forschungsbereiche, Fachgebiete nach Destatis

Schlagwörter

  • Usable AI, Materials Engineering, Machine learning for materials, AutoML, Multi-task learning