Research map / Architectures and efficiency / Pruning and efficient tuning
No More Tuning: Prioritized Multi-Task Learning with Lagrangian Differential Multiplier Methods
Read the paperCheck your idea against 108,967 papers
AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.
Proposes No More Tuning (NMT), a step-wise constrained-optimization MTL framework that uses Lagrange multipliers to prioritize a primary task without loss-weight hyperparameter tuning, evaluated on TikTok, QK-Video and Taobao search A/B tests.
Problem
Multi-task learning with tasks of different priorities: manual loss-weight tuning grows exponentially with the number of tasks, and joint optimization can degrade high-priority task performance.
Approach
Cast prioritized MTL as sequential constrained optimization where each lower-priority task is minimized subject to not exceeding the optimal values of higher-priority tasks, solved via gradient descent on parameters and gradient ascent on Lagrange multipliers, with no extra hyperparameters.
Key results
- On TikTok, NMT raises Like AUC by +0.38% to +0.49% across Shared-Bottom, OMoE, MMoE and PLE, with Finish AUC also slightly improving.
- On QK-Video, Like AUC improves by +0.03% to +0.10% with NMT, while Click AUC shifts by at most about -0.07%.
- In Taobao A/B tests, pay + relevance with NMT gives +0.26% order volume and +0.72% relevance, whereas manual weight adjustment drops order volume by -0.35%.
Measured effects reported in the paper
| What was changed | Where | Measured | Effect |
|---|---|---|---|
| Add NMT (Lagrangian constrained optimization for Like priority) to Shared-Bottom | Shared-Bottom · TikTok | Like AUC: +0.38% | Helped |
| Add NMT to OMoE | OMoE · TikTok | Like AUC: +0.38% | Helped |
| Add NMT to MMoE | MMoE · TikTok | Like AUC: +0.41% | Helped |
| Add NMT to PLE | PLE · TikTok | Like AUC: +0.49% | Helped |
| Add NMT to Shared-Bottom, OMoE, MMoE, PLE | Shared-Bottom, OMoE, MMoE, PLE · TikTok | Finish AUC: +0.03% to +0.06% | Helped |
| Add NMT to Shared-Bottom, OMoE, MMoE, PLE | Shared-Bottom, OMoE, MMoE, PLE · QK-Video | Like AUC: +0.03% to +0.1% | Helped |
| Add NMT to models on QK-Video | Shared-Bottom, OMoE, MMoE, PLE · QK-Video | Click AUC: -0.01% to -0.07% | Hurt |
| Add NMT to models on QK-Video | Shared-Bottom, OMoE, MMoE, PLE · QK-Video | Average AUC: -0.01% to +0.03% | No clear effect |
| Add NMT to FAMO gradient-based MTL (ablation) | FAMO · TikTok | Like AUC: ShareBottom +0.21% to +0.41% range | Helped |
| NMT with weight-adjustment alternative: manual loss weighting pay+relevance in online search (ablation) | Multi-task ranking (pay + relevance) · Taobao search A/B test | Order Volume: -0.35% vs +0.26% with NMT | Hurt |
| Add NMT for pay+relevance instead of manual weight adjustment (ablation) | Multi-task ranking (pay + relevance) · Taobao search A/B test | Relevance: +0.72% vs +0.42% for weight adjustment | Helped |
| Add NMT for pay+amount (GMV) task | Multi-task ranking (pay + amount) · Taobao search A/B test | GMV: +0.57% | Helped |
Surprising findings
- Weight adjustment in the Taobao pay + relevance setup reduced order volume by -0.35%, while NMT did not reduce it.
- On QK-Video, NMT slightly lowered Click AUC for some models while raising Like AUC.
Limitations
- Strong-duality guarantee relies on assumptions (no overfitting, small parameter change) and is only approximate for non-convex models.
- Theoretical argument is partly informal; the convexity proof is approximate (O(epsilon) terms).
- Tasks are prioritized by a fixed order; the method was not compared against tuned weight grids on public datasets beyond a TikTok sweep.
Experimental setup
Datasets: TikTok QK-Video Taobao search logs
Benchmarks: TikTok QK-Video Taobao search A/B test
Metrics: AUC order volume GMV relevance
Baselines: Shared-Bottom OMoE MMoE PLE FAMO weight adjustment
Builds on: Lagrangian methods MMoE PLE FAMO
Closest papers by meaning
- Towards Consistent Multi-Task Learning: Unlocking the Potential of Task-Specific Parameters (CVPR 2025 · similarity 0.87)
- Quantifying Task Priority for Multi-Task Optimization (CVPR 2024 · similarity 0.87)
- Delve into the Applicability of Advanced Optimizers for Multi-Task Learning (NeurIPS 2026 · similarity 0.86)
- Revisiting Fairness in Multitask Learning: A Performance-Driven Approach for Variance Reduction (CVPR 2025 · similarity 0.86)
- Bayesian Uncertainty for Gradient Aggregation in Multi-Task Learning (ICML 2024 · similarity 0.85)
- Selective Task Group Updates for Multi-Task Optimization (ICLR 2025 · similarity 0.85)
Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.
