Federated Language Models Under Bandwidth Budgets: Distillation Rates and Conformal Coverage
Not on the map. Why?
Abstract (arXiv)
Hospitals, companies, and research consortia want language models that learn from, and answer questions over, data that must stay where it is held, over links with limited bandwidth, such as the satellite links of remote clinics and ships. Averaging the sites' model parameters makes each message grow with the model and requires one shared architecture; methods that exchange predictions avoid both limits, but, to our knowledge, none bounds what compressing those predictions costs. We introduce two methods that compress every message at a small and predictable cost. Federated Probe-Logit Distillation (FPLD) trains a shared language model: each site sends compressed logits on shared unlabeled probes, and a hub distills one model from their average. Federated Conformal Retrieval-Augmented Generation (FC-RAG) answers questions: each site scores candidate answers using its own documents and sends compressed scores, and the hub returns a conformal prediction set that contains the correct answer at a guaranteed rate. Both use one codec, which centers each message and rounds it after adding a random offset known to the receiver. For logits within a fixed clipping range, we bound the expected log-loss added by compression independently of vocabulary size; the bound depends only on the bits, the range, and the number of sites, falls as more sites take part, and lets the bit budget be set before deployment. We also prove that recalibrating after compression keeps the coverage guarantee, and that FPLD's training error bounds how far coverage can slip when a threshold is reused after a model update. In experiments with GPT-2 and Qwen2.5, 8 bits per number cause no detectable loss in the trained model or in coverage. FPLD also trains a shared model across different architectures, where parameter averaging does not apply; with one shared architecture, parameter averaging remains more accurate.
Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Builds on 11 of its 22 references that are in the map. First posted on arXiv 11 May 2026.
Builds on
11 of 22 references.
References · 11
- 2020Ensemble Distillation for Robust Model Fusion in Federated Learningcited by 682
- 2019Conformalized Quantile Regressioncited by 629
- 2019FedMD: Heterogenous Federated Learning via Model Distillationcited by 459
- 2021FedKD: Communication Efficient Federated Learning via Knowledge Distillationcited by 187
- 2023TRAQ: Trustworthy Retrieval Augmented Question Answering via Conformal Predictioncited by 25
- 2015A Distributed One-Step Estimatorcited by 11
- 2025Response Quality Assessment for Retrieval-Augmented Generation via Conditional Conformal Factualitycited by 8
- 2025Principled Context Engineering for RAG: Statistical Guarantees via Conformal Predictioncited by 6
- 2025Feature Distillation is the Better Choice for Model-Heterogeneous Federated Learningcited by 5
- 2026Efficient Federated Conformal Prediction with Group-Conditional Guaranteecited by 4
- 2025Federated Conditional Conformal Prediction via Generative Modelscited by 2
Built on by
Citing papers per year: 2026: 3.
3 papers.
- 2026Certified Adaptive Refresh: Anytime-Valid Monitoring for Federated Conformal RAGcited by 2
- 2026Matching Rates and Optimal Allocation for Federated Probe-Logit Distillation under Heterogeneous Bandwidth Budgetscited by 0
- 2026Byzantine-Robust Federated RAG via Aligned Calibration and Fixed-Membership Conformal Predictioncited by 0
- 2026Byzantine-Robust Federated RAG via Aligned Calibration and Fixed-Membership Conformal Prediction27 Sep 2026
- 2026Matching Rates and Optimal Allocation for Federated Probe-Logit Distillation under Heterogeneous Bandwidth Budgets28 May 2026
- 2026Certified Adaptive Refresh: Anytime-Valid Monitoring for Federated Conformal RAG27 May 2026
Similar, not cited
Close in wording, no citation either way
- 2026FedSDR: Federated Self-Distillation with Rectification
- 2025Communication-Aware Knowledge Distillation for Federated LLM Fine-Tuning over Wireless Networks
- 2026FLoKD: Adaptive Knowledge Distillation for Federated Low-Rank LLM over Wireless Networks
- 2024Thinking Forward: Memory-Efficient Federated Finetuning of Language Models
- 2022Scaling Language Model Size in Cross-Device Federated Learning
- 2025FedCoT: Communication-Efficient Federated Reasoning Enhancement for Large Language Models
- 2026Federation of Experts: Communication Efficient Distributed Inference for Large Language Models
- 2025FedNano: Toward Lightweight Federated Tuning for Pretrained Multimodal Large Language Models