Personalized models and optimization in federated learning
2025 (English)Doctoral thesis, comprehensive summary (Other academic)Alternative title
Personanpassade modeller och optimering inom federerad inlärning (Swedish)
Abstract [en]
The rapid increase in data generation, combined with the impracticality of centralizing large-scale datasets and the growing complexity of machine learning tasks, has driven the development of distributed learning techniques. Among these, Federated Learning (FL) has gained significant attention due to its privacy-preserving approach, where multiple clients collaboratively train a global model without sharing their local data. However, FL faces several key challenges, including data heterogeneity, high computational costs, and communication inefficiencies. These issues become more pronounced in real-world scenarios where client data distributions are non-IID, computational resources are limited, and communication is constrained.
This thesis addresses these challenges through the development of efficient algorithms for Personalized Federated Learning (pFL) and Constrained Federated Learning. The proposed approaches are designed to handle heterogeneous data, minimize computational overhead, and reduce communication costs while maintaining strong theoretical guarantees.
Specifically, the thesis introduces three key contributions: (1) pFLMF, a novel pFL formulation based on low-rank matrix optimization, leveraging Burer-Monteiro factorization to enable personalization without relying on predefined distance metrics. (2) PerMFL, an algorithm for multi-tier pFL that introduces personalized decision variables for both teams and individual devices, enabling efficient optimization in scenarios with hierarchical client structures. (3) FedFW, a projection-free algorithm for constrained FL, which emphasizes low computational cost, privacy preservation, and communication efficiency through sparse signal exchanges.
By addressing critical issues in FL, such as data heterogeneity, computation costs, and communication bottlenecks, the proposed algorithms advance the field of Federated Learning, providing robust and scalable solutions for real-world applications.
Place, publisher, year, edition, pages
Umeå: Umeå University, 2025. , p. 30
Series
Research report in mathematical statistics, ISSN 1653-0829
Keywords [en]
Federated Learning, Machine Learning, Optimization
National Category
Computational Mathematics
Identifiers
URN: urn:nbn:se:umu:diva-234464ISBN: 9789180706001 (electronic)ISBN: 9789180705998 (print)OAI: oai:DiVA.org:umu-234464DiVA, id: diva2:1930455
Public defence
2025-02-12, UB.A.220, Samhällsvetarhuset, 09:00 (English)
Opponent
Supervisors
Funder
Wallenberg AI, Autonomous Systems and Software Program (WASP)2025-01-292025-01-232025-01-27Bibliographically approved
List of papers