Closer to Reality: Practical Semi-Supervised Federated Learning for Foundation Model Adaptation
By: Guangyu Sun , Jingtao Li , Weiming Zhuang and more
Potential Business Impact:
Teaches AI to learn from private data safely.
Foundation models (FMs) exhibit remarkable generalization but require adaptation to downstream tasks, particularly in privacy-sensitive applications. Due to data privacy regulations, cloud-based FMs cannot directly access private edge data, limiting their adaptation. Federated learning (FL) provides a privacy-aware alternative, but existing FL approaches overlook the constraints imposed by edge devices -- namely, limited computational resources and the scarcity of labeled data. To address these challenges, we introduce Practical Semi-Supervised Federated Learning (PSSFL), where edge devices hold only unlabeled, low-resolution data, while the server has limited labeled, high-resolution data. In this setting, we propose the Federated Mixture of Experts (FedMox), a novel framework that enhances FM adaptation in FL. FedMox tackles computational and resolution mismatch challenges via a sparse Mixture-of-Experts architecture, employing a spatial router to align features across resolutions and a Soft-Mixture strategy to stabilize semi-supervised learning. We take object detection as a case study, and experiments on real-world autonomous driving datasets demonstrate that FedMox effectively adapts FMs under PSSFL, significantly improving performance with constrained memory costs on edge devices. Our work paves the way for scalable and privacy-preserving FM adaptation in federated scenarios.
Similar Papers
Flexible Personalized Split Federated Learning for On-Device Fine-Tuning of Foundation Models
Distributed, Parallel, and Cluster Computing
Helps AI learn better from small, different data.
Hierarchical Federated Learning for Social Network with Mobility
Machine Learning (CS)
Learns from phones without seeing your private stuff.
Federated Foundation Models in Harsh Wireless Environments: Prospects, Challenges, and Future Directions
Networking and Internet Architecture
Makes smart computers work anywhere, even with bad internet.