Robust-multi-task gradient boosting
• Proposes a Robust-Multi-Task Gradient Boosting model (R-MTGB). • Detects and adapts to outlier tasks automatically during training. • Learns outlier and non-outlier tasks separately to enhance robustness. • Provides task-level outlier scores for interpretability and diagnostics. • Improves predictive accuracy across heterogeneous multi-task datasets. The objective of this study is to develop a robust boosting framework capable of handling heterogeneous and outlier tasks in Multi-Task Learning (MTL). Conventional MTL methods assume strong relatedness among tasks, which often fails in real-world scenarios involving adversarial or unaligned tasks that degrade performance. To address this limitation, we propose Robust Multi-Task Gradient Boosting (R-MTGB), a novel ensemble framework that explicitly models task heterogeneity within the gradient boosting paradigm. The methodology structures learning into three sequential stages: (1) shared representation learning to extract common patterns across tasks, (2) outlier-aware partitioning using a learnable task-specific parameter to separate and reweight outlier and non-outlier tasks, and (3) task-specific fine-tuning to refine individual predictors. Extensive experiments on both synthetic and real-world datasets demonstrate that R-MTGB consistently improves predictive accuracy, effectively identifies outlier tasks, and enhances generalization compared to state-of-the-art methods. The achieved results confirm that R-MTGB not only ensures robust performance and interpretability through task-level outlier scores but also provides a scalable and principled framework for reliable multi-task learning in heterogeneous environments.
Authors
- Daniel Hernández-Lobato (ORCID: https://orcid.org/0000-0001-5845-437X)
- Seyedsaman Emami (ORCID: https://orcid.org/0000-0002-6306-1180)
- Gonzalo Martínez-Mu Noz (ORCID: https://orcid.org/0000-0002-6125-6056)
Publication Details
- Journal
- Expert Systems with Applications
- Published
- 2025-12-07
- DOI
- https://doi.org/10.1016/j.eswa.2025.130696
- Citations
- 1
- Primary Topic
- Explainable Artificial Intelligence (XAI)
- Type
- article
- Field-Weighted Citation Impact
- 2.48