DREAM: A Unified Framework for Drift-Corrected Federated Multi-Objective Learning
Yuan Zhou, Yidan Ou, Xinli Shi
摘要
Federated Multi-Objective Learning (FMOL) enables collaborative training of conflicting objectives but faces a compounded challenge: the recursive coupling between intra-task client drift and inter-task aggregation bias. We propose DREAM, a unified framework that jointly corrects these two coupled error sources through drift-aware control variates and momentum-smoothed local updates. On the server side, DREAM formulates multi-objective aggregation as a regularized quadratic program parameterized by a task correction matrix, which provides a generalized formulation that can flexibly adapt to scalarization, prioritization, and gradient manipulation strategies. Theoretically, we establish a linear speedup convergence rate of for non-convex objectives. We further provide theoretical guarantees for the conflict-avoidant direction distance. In the strongly convex setting, DREAM achieves convergence in weighted sub-optimality and admits a unified Lyapunov analysis showing linear convergence to a regularization-dependent neighborhood. Numerical experiments on representative benchmarks validate the effectiveness of DREAM in multi-objective optimization.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
它引用的顶会 Paper16
- SCAFFOLD: Stochastic Controlled Averaging for Federated LearningSai Praneeth Karimireddy, Satyen Kale, Mehryar Mohri, Sashank J. Reddi 等ICML 2020 · 被引用 3,875 次
- On the Convergence of FedAvg on Non-IID DataXiang Li, Kaixuan Huang, Wenhao Yang, Shusen Wang 等ICLR 2020 · 被引用 2,930 次
- Gradient Surgery for Multi-Task LearningTianhe Yu, Saurabh Kumar, Abhishek Gupta, Sergey Levine 等NeurIPS 2020 · 被引用 2,261 次
- Conflict-Averse Gradient Descent for Multi-task learningBo Liu, Xingchao Liu, Xiaojie Jin, Peter Stone 等NeurIPS 2021 · 被引用 686 次
- Which Tasks Should Be Learned Together in Multi-task Learning?Trevor Standley, Amir Zamir, Dawn Chen, Leonidas J. Guibas 等ICML 2020 · 被引用 651 次
相关 Paper
- Breaking the centralized barrier for cross-device federated learningSai Praneeth Karimireddy, Martin Jaggi, Satyen Kale, Mehryar Mohri 等NeurIPS 2021 · 被引用 113 次
- Federated Multi-Objective LearningHaibo Yang, Zhuqing Liu, Jia Liu, Chaosheng Dong 等NeurIPS 2023 · 被引用 28 次
- Momentum Benefits Non-iid Federated Learning Simply and ProvablyZiheng Cheng, Xinmeng Huang, Pengfei Wu, Kun YuanICLR 2024 · 被引用 40 次
- Direction-oriented Multi-objective Learning: Simple and Provable Stochastic AlgorithmsPeiyao Xiao, Hao Ban, Kaiyi JiNeurIPS 2023 · 被引用 46 次
- FedMoS: Taming Client Drift in Federated Learning with Double Momentum and Adaptive SelectionXiong Wang, Yuxin Chen, Yuqing Li, Xiaofei Liao 等INFOCOM 2023 · 被引用 13 次
