RobuMTL: Enhancing Multi-Task Learning Robustness Against Weather Conditions
About
Robust Multi-Task Learning (MTL) is crucial for autonomous systems operating in real-world environments, where adverse weather conditions can severely degrade model performance and reliability. In this paper, we introduce RobuMTL, a novel architecture designed to adaptively address visual degradation by dynamically selecting task-specific hierarchical Low-Rank Adaptation (LoRA) modules and a LoRA expert squad based on input perturbations in a mixture-of-experts fashion. Our framework enables adaptive specialization based on input characteristics, improving robustness across diverse real-world conditions. To validate our approach, we evaluated it on the PASCAL and NYUD-v2 datasets and compared it against single-task models, standard MTL baselines, and state-of-the-art methods. On the PASCAL benchmark, RobuMTL delivers a +2.8% average relative improvement under single perturbations and up to +44.4% under mixed weather conditions compared to the MTL baseline. On NYUD-v2, RobuMTL achieves a +9.7% average relative improvement across tasks. The code is available at GitHub.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Depth Estimation | NYU Depth V2 | RMSE0.67 | 177 | |
| Semantic segmentation | NYUD v2 | mIoU37.19 | 96 | |
| Surface Normal Estimation | NYU V2 | RMSE27.52 | 23 | |
| Human Parsing | PASCAL-Context clean and adverse conditions v1 (unseen) | mIoU55.01 | 19 | |
| Multi-task Learning | PASCAL-Context and NYUD-v2 clean and adverse conditions v1 (unseen) | Delta MADV (Adverse)0.028 | 19 | |
| Saliency Detection | PASCAL-Context clean and adverse conditions v1 (unseen) | mIoU62.75 | 19 | |
| Semantic segmentation | PASCAL-Context clean and adverse conditions v1 (unseen) | mIoU63.86 | 19 | |
| Surface Normal Estimation | NYUD clean and adverse conditions v2 v1 (unseen) | RMSE18.17 | 18 | |
| Edge Detection | NYUD v2 | -- | 16 |