AutoScale: Linear Scalarization Guided by Multi-Task Optimization Metrics
Not on the map. Why?
Abstract (arXiv)
Recent multi-task learning studies suggest that linear scalarization, when using well-chosen fixed task weights, can achieve comparable to or even better performance than complex multi-task optimization (MTO) methods. It remains unclear why certain weights yield optimal performance and how to determine these weights without relying on exhaustive hyperparameter search. This paper establishes a direct connection between linear scalarization and MTO methods, revealing through extensive experiments that well-performing scalarization weights exhibit specific trends in key MTO metrics, such as high gradient magnitude similarity. Building on this insight, we introduce AutoScale, a simple yet effective two-phase framework that uses these MTO metrics to guide weight selection for linear scalarization, without expensive weight search. AutoScale consistently shows superior performance with high efficiency across diverse datasets including a new large-scale benchmark.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 38 of its 49 references that are in the map. First posted on arXiv 19 Aug 2025.
Builds on
38 of 49 references.
References · 38
- 2019nuScenes: A multimodal dataset for autonomous drivingcited by 4,523
- 2016The Cityscapes Dataset for Semantic Urban Scene Understandingcited by 4,444
- 2016Pyramid Scene Parsing Networkcited by 2,479
- 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
- 2018Taskonomy: Disentangling Task Transfer Learningcited by 634
- 2018End-to-End Multi-Task Learning with Attentioncited by 534
- 2016Cross-stitch Networks for Multi-task Learningcited by 467
- 2021Conflict-Averse Gradient Descent for Multi-task Learningcited by 459
- 2019Which Tasks Should Be Learned Together in Multi-task Learning?cited by 375
26 more
- 2022Multi-Task Learning as a Bargaining Gamecited by 215
- 2020Just Pick a Sign: Optimizing Deep Multitask Models with Gradient Sign Dropoutcited by 201
- 2015Instance-aware Semantic Segmentation via Multi-task Network Cascadescited by 121
- 2021Reasonable Effectiveness of Random Weighting: A Litmus Test for Multi-Task Learningcited by 101
- 2023FAMO: Fast Adaptive Multitask Optimizationcited by 78
- 2023UniTR: A Unified and Efficient Multi-Modal Transformer for Bird's-Eye-View Representationcited by 74
- 2022Auto-Lambda: Disentangling Dynamic Task Relationshipscited by 73
- 2023Independent Component Alignment for Multi-Task Learningcited by 70
- 2022In Defense of the Unitary Scalarization for Deep Multi-Task Learningcited by 67
- 2022Do Current Multi-Task Optimization Methods in Deep Learning Even Help?cited by 60
- 2023Revisiting Scalarization in Multi-Task Learning: A Theoretical Perspectivecited by 51
- 2019MultiNet++: Multi-Stream Feature Aggregation and Geometric Loss Strategy for Multi-Task Learningcited by 45
- 2021Multi-Objective Meta Learningcited by 39
- 2019Regularizing Deep Multi-Task Networks using Orthogonal Gradientscited by 36
- 2023Three-Way Trade-Off in Multi-Objective Learning: Optimization, Generalization and Conflict-Avoidancecited by 36
- 2024Fair Resource Allocation in Multi-Task Learningcited by 35
- 2023Direction-oriented Multi-objective Learning: Simple and Provable Stochastic Algorithmscited by 34
- 2023Dual-Balancing for Multi-Task Learningcited by 27
- 2024Robust Multi-Task Learning with Excess Riskscited by 24
- 2023Scalarization for Multi-Task and Multi-Domain Learning at Scalecited by 19
- 2024Jacobian Descent for Multi-Objective Optimizationcited by 18
- 2025Selective Task Group Updates for Multi-Task Optimizationcited by 17
- 2017Deep Asymmetric Multi-task Feature Learningcited by 15
- 2023FULLER: Unified Multi-modality Multi-task 3D Perception via Multi-level Gradient Calibrationcited by 13
- 2024Bayesian Uncertainty for Gradient Aggregation in Multi-Task Learningcited by 11
- 2024Adaptive Online Mirror Descent for Tchebycheff Scalarization in Multi-Objective Learningcited by 4
Built on by
Citing papers per year: 2025: 1; 2026: 1.
2 papers.