Bounded Precision-Geometry Scaling for Robust Multi-Task Learning under Loss Scale Mismatch
Not on the map. Why?
Abstract (arXiv)
Multi-task learning often combines losses that span several orders of magnitude, causing homoscedastic uncertainty weighting to degrade severely. We propose Bounded Precision-Geometry Scaling (BPGS), a method that maps each task's log-variance through a bounded sigmoid parameterisation anchored to detached batch loss statistics, and decouples network optimisation from uncertainty optimisation. Its normalised task weights are provably invariant to uniform rescaling under non-degenerate loss scales. We evaluate BPGS on synthetic stress tests and three real-world benchmarks: NYUv2 dense prediction, Yeast multi-label classification, and RF1 multi-target regression. Under pure loss rescaling from × 1 to × 1000, its macro score changes from 0.777 to 0.778, whereas Kendall weighting drops from 0.780 to 0.637; ℓ1-normalising Kendall's weights does not close the gap. On NYUv2, BPGS records the lowest depth absolute relative error (0.223), depth RMSE (0.790), and total loss (1.891) among all compared methods, including Nash-MTL. Sensitivity studies on batch size and calibration show small variation across the tested ranges, and runtime overhead relative to Kendall is under 1%. BPGS posts the highest Yeast micro-F1 (0.616) and is competitive on RF1, though PCGrad leads RMSE and MAE there. These findings establish BPGS as a scale-robust alternative to homoscedastic uncertainty weighting, notably effective when loss-scale disparities dominate multi-task optimisation.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 9 of its 17 references that are in the map. First posted on arXiv 21 Aug 2026.
Builds on
9 of 17 references.
References · 9
- 2017Multi-Task Learning Using Uncertainty to Weigh Losses for Scene Geometry and Semanticscited by 1,358
- 2020Gradient Surgery for Multi-Task Learningcited by 1,256
- 2017GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networkscited by 834
- 2018Multi-Task Learning as Multi-Objective Optimizationcited by 821
- 2018End-to-End Multi-Task Learning with Attentioncited by 534
- 2021Conflict-Averse Gradient Descent for Multi-task Learningcited by 459
- 2022Multi-Task Learning as a Bargaining Gamecited by 215
- 2023FAMO: Fast Adaptive Multitask Optimizationcited by 78
- 2022Auto-Lambda: Disentangling Dynamic Task Relationshipscited by 73
Built on by
Not cited yet.