无需调参的多任务学习框架,自动提升高优先级任务性能
No More Tuning: Prioritized Multi-Task Learning with Lagrangian Differential Multiplier Methods
- 用拉格朗日微分乘子法实现分步优化,避免任务间干扰
- 在淘宝搜索系统中显著提升多个业务指标,无需人工调参
- 理论保证优化可靠性,适合工业级多任务场景
多任务学习在实际系统中广泛应用,但各任务常有不同优先级。例如在网页搜索中,相关性通常比点击率或用户参与度更重要。现有方法通过调整任务损失权重来区分优先级,但随着任务数量增加,超参数调优复杂度呈指数增长,且低优先级任务可能干扰高优先级任务的性能。本文提出一种基于拉格朗日微分乘子法的新型多任务学习框架,实现分步优化,可有效提升高优先级任务性能而不受其他任务干扰。其核心优势在于无需为不同任务设置平衡超参数,完全自动化优化过程。我们提供了理论分析,证明该方法具有优化保证。实验在多个公开数据集及淘宝搜索系统上验证了其有效性,显著提升了各类业务指标。
原文摘要 · Abstract (English)
Given the ubiquity of multi-task in practical systems, Multi-Task Learning (MTL) has found widespread application across diverse domains. In real-world scenarios, these tasks often have different priorities. For instance, In web search, relevance is often prioritized over other metrics, such as click-through rates or user engagement. Existing frameworks pay insufficient attention to the prioritization among different tasks, which typically adjust task-specific loss function weights to differentiate task priorities. However, this approach encounters challenges as the number of tasks grows, leading to exponential increases in hyper-parameter tuning complexity. Furthermore, the simultaneous optimization of multiple objectives can negatively impact the performance of high-priority tasks due to interference from lower-priority tasks. In this paper, we introduce a novel multi-task learning framework employing Lagrangian Differential Multiplier Methods for step-wise multi-task optimization. It is designed to boost the performance of high-priority tasks without interference from other tasks. Its primary advantage lies in its ability to automatically optimize multiple objectives without requiring balancing hyper-parameters for different tasks, thereby eliminating the need for manual tuning. Additionally, we provide theoretical analysis demonstrating that our method ensures optimization guarantees, enhancing the reliability of the process. We demonstrate its effectiveness through experiments on multiple public datasets and its application in Taobao search, a large-scale industrial search ranking system, resulting in significant improvements across various business metrics.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。