Perplexity Predicts Protection: Choosing Pretrained Backbones for Worst-Client Fairness in Federated Parameter-Efficient Fine-Tuning
Not on the map. Why?
Abstract (arXiv)
Federated learning lets multiple parties train a shared model without pooling their data, but a client with far less data than the others can end up poorly served even when the group's average accuracy looks fine. We ask whether the choice of pretrained backbone affects this under LoRA fine-tuning, and whether per-word perplexity on the target text predicts which backbone helps the worst-off client before federated training starts. We ran 313 experiments across three text-classification datasets and three similarly sized backbones (RoBERTa, BERTweet, PubMedBERT), each compared against a task-specific baseline on identical data splits. Lower-perplexity backbones consistently produced larger gains for the worst-performing client, with a rank correlation of -0.87 across nine dataset-backbone pairs; a backbone held out of the analysis confirmed the pattern. Personalization with Ditto recovered only 4-12% of the gap between training alone and full federation, and removing aggregation entirely erased the benefit. A client's update also showed no sign of conflicting with the group's update; the two are close to orthogonal, ruling out one proposed explanation for this failure. Practically: measure perplexity on a sample of task text before choosing a backbone, and do not rely on personalization to protect a data-poor client. We release our code, predictions, and full results for others to test.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 26 of its 37 references that are in the map. First posted on arXiv 20 Sep 2026.
Builds on
26 of 37 references.
References · 26
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,372
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,669
- 2015Character-level Convolutional Networks for Text Classificationcited by 2,330
- 2014Convolutional Neural Networks for Sentence Classificationcited by 1,127
- 2020Domain-Specific Language Model Pretraining for Biomedical Natural Language Processingcited by 1,009
- 2019Measuring the Effects of Non-Identical Data Distribution for Federated Visual Classificationcited by 851
- 2020Ditto: Fair and Robust Federated Learning Through Personalizationcited by 672
- 2020Personalized Federated Learning with Moreau Envelopescited by 618
- 2019Agnostic Federated Learningcited by 456
- 2019Fair Resource Allocation in Federated Learningcited by 394
- 2020BERTweet: A pre-trained language model for English Tweetscited by 346
- 2020How Good is Your Tokenizer? On the Monolingual Performance of Multilingual Language Modelscited by 318
14 more
- 2023Towards Building the Federated GPT: Federated Instruction Tuningcited by 185
- 2024Improving LoRA in Privacy-preserving Federated Learningcited by 161
- 2024FLoRA: Federated Fine-Tuning Large Language Models with Heterogeneous Low-Rank Adaptationscited by 146
- 2017PubMed 200k RCT: a Dataset for Sequential Sentence Classification in Medical Abstractscited by 124
- 2024Heterogeneous LoRA for Federated Fine-tuning of On-Device Foundation Modelscited by 116
- 2023Stochastic Controlled Averaging for Federated Learning with Communication Compressioncited by 101
- 2024Federated Fine-tuning of Large Language Models under Heterogeneous Tasks and Client Resourcescited by 93
- 2024Selective Aggregation for Low-Rank Adaptation in Federated Learningcited by 80
- 2022Where to Begin? On the Impact of Pre-Training and Initialization in Federated Learningcited by 71
- 2023pFedLoRA: Model-Heterogeneous Personalized Federated Learning with LoRA Tuningcited by 40
- 2024Non-IID data in Federated Learning: A Survey with Taxonomy, Metrics, Methods, Frameworks and Future Directionscited by 13
- 2025Federated Low-Rank Adaptation for Foundation Models: A Surveycited by 9
- 2024A Survey on Group Fairness in Federated Learning: Challenges, Taxonomy of Solutions and Directions for Future Researchcited by 2
- 2026When More Parameters Hurt: Foundation Model Priors Amplify Worst-Client Disparity Under Extreme Federated Heterogeneitycited by 1
Built on by
Not cited yet.