arXiv · 2509.12697
Bi-level Personalization for Federated Foundation Models: A Task-vector Aggregation Approach
Abstract
Federated foundation models represent a new paradigm to jointly fine-tune pre-trained foundation models across clients. It is still a challenge to fine-tune foundation models for a small group of new users or specialized scenarios, which typically involve limited data compared to the large-scale data used in pre-training. In this context, the trade-off between personalization and federation becomes more sensitive. To tackle these, we proposed a bi-level personalization framework for federated fine-tuning on foundation models. Specifically, we conduct personalized fine-tuning on the client-level using its private data, and then conduct a personalized aggregation on the server-level using similar users measured by client-specific task vectors. Given the personalization information gained from client-level fine-tuning, the server-level personalized aggregation can gain group-wise personalization information while mitigating the disturbance of irrelevant or interest-conflict clients with non-IID data. The effectiveness of the proposed algorithm has been demonstrated by extensive experimental analysis in benchmark datasets.
Explore related subjects
Keep this discovery
Yiyuan Yang, Guodong Long, Qinghua Lu, Liming Zhu, Jing Jiang. 2025-09-16. Bi-level Personalization for Federated Foundation Models: A Task-vector Aggregation Approach. https://arxiv.org/abs/2509.12697
Cite the original work for its findings. Save a collection to share your selection of sources.