BLEND: Balancing Personalization vs. Generalization in Federated Vision-Language Models
Organizations: University at Buffalo–SUNY · Adobe · Zhejiang University
Abstract
Federated vision-language models (FedVLMs) represent one of the emerging frontiers of machine learning, aiming to integrate federated learning within the fine-tuning pipelines of vision-language models (VLMs). In this work, we introduce BLEND, a new framework designed to jointly enhance the personalization and generalization capabilities of FedVLMs. BLEND proposes a selective VLM parameter fine-tuning and aggregation strategy that consists of (i) global vision and text adapters shared across clients, and (ii) a personalized vision projection adapter tailored to each client. BLEND further introduces a novel loss function that blends the outputs of the global and personalized adapters to achieve personalization, while simultaneously promoting generalization through a principled anchor design implemented via the projection heads of the zero-shot VLM. Extensive experiments demonstrate that BLEND consistently outperforms state-of-the-art baselines, highlighting its ability to effectively balance personalization and generalization across diverse system settings. (Code is available at: https://github.com/payamsiabd/BLEND.git)
Figures & tables
| UCF101 | Oxford Pets | Stanford Cars | Flowers | Imagenet | FGVC Aircraft | |||||||||||||
| Per. | Gen. | H-Mean | Per. | Gen. | H-Mean | Per. | Gen. | H-Mean | Per. | Gen. | H-Mean | Per. | Gen. | H-Mean | Per. | Gen. | H-Mean | |
| Zero-shot CLIP | 68.10 0.00 | 75.18 0.00 | 71.46 0.00 | 89.47 0.00 | 96.81 0.00 | 93.00 0.00 | 63.77 0.00 | 74.94 0.00 | 68.91 0.00 | 69.42 0.00 | 76.74 0.00 | 72.90 0.00 | 72.40 0.00 | 68.08 0.00 | 70.17 0.00 | 27.49 0.00 | 33.05 0.00 | 30.02 0.00 |
| MMA+FedAvg | 85.40 0.93 | 75.95 1.20 | 80.40 1.08 | 95.12 0.07 | 97.65 0.08 | 96.37 0.05 | 76.23 1.63 | 69.60 0.40 | 72.75 0.82 | 96.93 0.22 | 70.87 1.19 | 81.87 0.87 | 75.52 2.03 | 69.18 0.98 | 72.21 0.32 | 38.92 2.03 | 33.77 0.98 | 36.10 0.32 |
| pFedMMA | 89.95 1.16 | 70.51 0.66 | 79.04 0.22 | 96.81 0.81 | 95.08 0.25 | 95.93 0.33 | 67.45 2.01 | 64.52 0.72 | 65.95 1.33 | 84.61 2.42 | 69.89 0.73 | 76.52 0.62 | 49.15 1.42 | 50.93 1.02 | 50.03 1.22 | 40.37 5.66 | 29.14 1.69 | 33.78 3.04 |
| pFedMoAP | 93.36 0.12 | 67.27 0.02 | 78.19 0.04 | 99.18 0.02 | 84.84 0.10 | 91.45 0.01 | 92.00 0.20 | 68.50 0.05 | 78.53 0.07 | 98.83 0.15 | 71.49 0.00 | 82.96 0.05 | 86.65 0.00 | 62.56 0.03 | 72.66 0.02 | 65.51 0.08 | 18.58 0.11 | 28.95 0.01 |
| FedPHA | 92.61 0.70 | 50.29 2.31 | 65.14 1.82 | 98.81 0.63 | 70.06 2.75 | 81.95 1.89 | 91.90 0.55 | 44.52 1.31 | 59.96 1.07 | 98.73 0.21 | 49.77 2.86 | 66.12 2.53 | 90.49 0.14 | 39.71 2.26 | 55.20 2.33 | 66.62 0.44 | 13.00 2.97 | 21.59 4.12 |
| Rank ( ) | Per. | Gen. | H-Mean |
|---|---|---|---|
| No adapter | 89.50 | 72.31 | 79.95 |
| 4 | 92.25 | 74.69 | 82.05 |
| 8 | 92.45 | 74.70 | 82.10 |
| 16 | 92.91 | 74.62 | 82.15 |
| 32 | 92.93 | 74.39 | 81.97 |
| 64 | 92.91 | 74.09 | 81.74 |
| Rank ( ) | Per. | Gen. | H-Mean |
|---|---|---|---|
| 4 | 89.99 | 75.00 | 81.44 |
| 8 | 90.45 | 74.83 | 81.73 |
| 16 | 91.00 | 74.86 | 81.73 |
| 32 | 91.59 | 74.91 | 81.96 |
| 64 | 92.12 | 74.64 | 81.92 |
| 128 | 92.57 | 74.71 | 82.12 |
| Agg. Strategy | Per. | Gen. | H-Mean |
|---|---|---|---|
| BLEND | 92.91 | 74.62 | 82.15 |
| Full | 83.35 | 75.41 | 79.01 |
| NoAgg | 89.12 | 69.71 | 77.71 |
| Only Vision | 74.64 | 70.02 | 71.82 |
| Only Text | 93.84 | 67.31 | 77.04 |
Appendix figures & tables85 assets
Supplementary material from the paper’s appendix.
Appendix
| Method | Total Number of Exchanged Params. | Communication Overhead (MB) | Per. | Gen. | H-Mean |
|---|---|---|---|---|---|
| MMA+FedAVG | 675K | 64.33 | 82.69 | 73.14 | 77.34 |
| BLEND | 507K | 48.38 | 92.92 | 74.61 | 82.15 |
| Method | Total Number of Exchanged Params. | Communication Overhead (MB) | Per. | Gen. | H-Mean |
|---|---|---|---|---|---|
| pFedMMA | 8.4K | 0.81 | 77.99 | 66.69 | 71.48 |
| pFedMOAP | 8.2K | 0.78 | 92.03 | 66.48 | 75.98 |
| FedPHA | 8.2K | 0.78 | 92.35 | 47.90 | 61.15 |
| FedOTP | 8.2K | 0.78 | 85.92 | 32.57 | 45.39 |
| PromptFL | 8.2K | 0.78 | 78.75 | 69.95 | 73.61 |
| PromptFolio | 8.2K | 0.78 | 85.75 | 50.84 | 61.85 |
| Method | Total Number of Trainable Params. | Per. | Gen. | H-Mean |
|---|---|---|---|---|
| MMA+FedAVG | 675K | 82.69 | 73.14 | 77.34 |
| pFedMMA | 675K | 77.99 | 66.69 | 71.48 |
| BLEND | 536K | 92.67 | 74.15 | 81.79 |
| Method | Total Number of Trainable Params. | Per. | Gen. | H-Mean |
|---|---|---|---|---|
| pFedMOAP | 74.2K | 92.03 | 66.48 | 75.98 |
| FedPHA | 16.4K | 92.35 | 47.90 | 61.15 |
| FedOTP | 16.4K | 85.92 | 32.57 | 45.39 |
| PromptFL | 8.2K | 78.75 | 69.95 | 73.61 |
| PromptFolio | 8.2K | 85.75 | 50.84 | 61.85 |
| FedPGP | 20.6K | 87.88 | 64.38 | 72.86 |
| Metric | BLEND | pFedMMA | MMA+FedAvg | FedPHA | FedOPT | FedPGP | PromptFL | pFedMOAP | PromptFolio |
|---|---|---|---|---|---|---|---|---|---|
| Time (s) | 11.18 | 11.53 | 11.59 | 13.35 | 9.70 | 15.19 | 10.30 | 10.01 | 10.11 |
| Mean GPU (GB) | 0.62 | 0.60 | 0.61 | 0.63 | 0.59 | 1.28 | 0.71 | 0.76 | 0.74 |
| UCF101 | Oxford Pets | Stanford Cars | Flowers | Imagenet | FGVC Aircraft | |||||||||||||
| Per. | Gen. | H-Mean | Per. | Gen. | H-Mean | Per. | Gen. | H-Mean | Per. | Gen. | H-Mean | Per. | Gen. | H-Mean | Per. | Gen. | H-Mean | |
| Zero-shot CLIP | 68.10 0.00 | 75.18 0.00 | 71.46 0.00 | 89.47 0.00 | 96.81 0.00 | 93.00 0.00 | 63.77 0.00 | 74.94 0.00 | 68.91 0.00 | 69.42 0.00 | 76.74 0.00 | 72.90 0.00 | 72.40 0.00 | 68.08 0.00 | 70.17 0.00 | 27.49 0.00 | 33.05 0.00 | 30.02 0.00 |
| MMA+FedAvg | 79.22 1.17 | 77.27 0.98 | 78.23 0.83 | 95.96 0.10 | 97.50 0.11 | 96.73 0.04 | 75.86 0.71 | 69.71 0.79 | 72.66 0.74 | 94.31 0.29 | 72.29 1.72 | 81.83 1.02 | 77.15 0.22 | 69.10 0.30 | 72.90 0.07 | 35.34 0.13 | 34.67 0.45 | 35.00 0.26 |
| pFedMMA | 92.85 0.51 | 69.35 0.70 | 79.40 0.53 | 97.19 0.96 | 94.97 0.24 | 96.06 0.37 | 77.73 2.77 | 62.93 0.54 | 69.52 0.92 | 86.38 1.12 | 72.15 0.29 | 78.62 0.46 | 70.39 2.26 | 58.47 0.85 | 63.87 1.44 | 40.40 1.40 | 28.13 0.45 | 33.16 0.60 |
| pFedMoAP | 93.94 0.35 | 67.22 0.01 | 78.36 0.12 | 99.63 0.09 | 84.78 0.01 | 91.61 0.04 | 92.49 0.54 | 68.45 0.01 | 78.67 0.20 | 96.73 0.63 | 71.44 0.01 | 82.18 0.22 | 90.21 0.00 | 62.49 0.00 | 73.84 0.00 | 73.23 2.16 | 18.53 0.00 | 29.58 0.18 |
| FedPHA | 93.37 0.56 | 51.90 3.30 | 66.64 2.58 | 99.82 0.06 | 56.42 3.71 | 72.02 3.02 | 93.80 0.70 | 41.06 0.10 | 57.12 0.03 | 95.07 0.46 | 43.03 1.25 | 59.23 1.28 | 91.60 0.01 | 34.09 0.50 | 49.69 0.53 | 71.86 0.00 | 8.03 0.00 | 14.44 0.00 |
| UCF101 | Oxford Pets | Stanford Cars | Flowers | Imagenet | FGVC Aircraft | |||||||||||||
| Per. | Gen. | H-Mean | Per. | Gen. | H-Mean | Per. | Gen. | H-Mean | Per. | Gen. | H-Mean | Per. | Gen. | H-Mean | Per. | Gen. | H-Mean | |
| Zero-shot CLIP | 68.10 0.00 | 75.18 0.00 | 71.46 0.00 | 89.47 0.00 | 96.81 0.00 | 93.00 0.00 | 63.77 0.00 | 74.94 0.00 | 68.91 0.00 | 69.42 0.00 | 76.74 0.00 | 72.90 0.00 | 72.40 0.00 | 68.08 0.00 | 70.17 0.00 | 27.49 0.00 | 33.05 0.00 | 30.02 0.00 |
| MMA+FedAvg | 77.84 0.74 | 77.54 1.06 | 77.68 0.55 | 94.78 0.23 | 97.50 0.05 | 96.12 0.10 | 72.45 1.07 | 71.57 0.40 | 72.00 0.49 | 92.89 0.49 | 73.19 1.61 | 81.86 0.82 | 77.57 0.25 | 69.20 0.30 | 73.15 0.28 | 32.25 0.96 | 35.15 1.04 | 33.64 0.98 |
| pFedMMA | 95.70 0.45 | 71.16 0.56 | 81.63 0.39 | 92.72 2.16 | 95.14 0.22 | 93.90 1.02 | 78.41 2.96 | 58.46 1.80 | 66.94 1.61 | 92.62 1.16 | 71.28 0.81 | 80.55 0.28 | 71.74 1.02 | 55.29 1.04 | 62.45 1.05 | 64.05 3.77 | 27.65 0.43 | 38.61 1.10 |
| pFedMoAP | 94.82 0.09 | 67.21 0.00 | 78.66 0.03 | 98.66 0.11 | 84.78 0.00 | 91.20 0.05 | 90.56 0.63 | 68.44 0.00 | 77.96 0.23 | 98.46 0.66 | 71.43 0.00 | 82.80 0.24 | 90.83 0.00 | 62.50 0.00 | 74.05 0.00 | 81.52 1.10 | 18.53 0.00 | 30.19 0.08 |
| FedPHA | 96.38 0.56 | 47.72 2.62 | 63.78 2.22 | 98.86 0.20 | 49.07 2.02 | 65.56 1.76 | 94.59 0.18 | 39.27 0.35 | 55.50 0.38 | 98.83 0.57 | 33.60 0.66 | 50.15 0.66 | 91.68 0.02 | 36.76 0.36 | 52.48 0.37 | 83.14 0.65 | 7.11 1.30 | 13.06 2.20 |
| Setting | Increasing Difficulty | H-Mean Gain vs. Best Baseline |
|---|---|---|
| Data heterogeneity (across 10 datasets in Sec. 4 ) | ||
| Number of clients (across 10 datasets in Sec. 4 ) | ||
| Real-World iNaturalist-User-120k Dataset (Appendix D ) | ||
| Local data scarcity (across 10 datasets in Sec. 4 ) | #Shots: |
| Method | Per. | Gen. | H-Mean |
|---|---|---|---|
| BLEND with Zero-Shot G-Head | 89.50 | 72.31 | 79.95 |
| BLEND with Zero-Shot Distilled G-Head | 92.83 | 72.35 | 80.54 |
| BLEND with Running Anchor Loss | 92.91 | 74.62 | 82.15 |
| Optimal | H-Mean | |
|---|---|---|
| 10 | 0.03 | 79.66 |
| 0.1 | 0.06 | 80.86 |
| 0.01 | 0.09 | 82.15 |
| Method | Per. | Gen. | H-Mean |
|---|---|---|---|
| BLEND (Sep-Per) | 93.56 | 70.96 | 79.78 |
| BLEND (Original) | 92.91 | 74.62 | 82.15 |
| 0.0 | 0.06 | 0.08 | 0.09 | 0.1 | 0.3 | 1 | |
|---|---|---|---|---|---|---|---|
| H-Mean | 78.86 | 82.25 | 82.38 | 82.39 | 82.36 | 79.47 | 73.47 |
| 0.0 | 0.1 | 0.19 | 0.2 | 0.21 | 0.5 | 1 | |
| H-Mean | 77.32 | 82.08 | 82.24 | 82.39 | 82.36 | 82.21 | 81.16 |
| Global adapter dim. | 0 | 4 | 8 | 16 | 32 | 64 | 128 | 256 | 512 |
|---|---|---|---|---|---|---|---|---|---|
| H-Mean | 80.84 | 82.11 | 82.18 | 82.39 | 82.06 | 81.97 | 81.75 | 81.31 | 81.32 |