FedFit: Federated Fine-Tuning of LLMs via Vector-Bank Parameterization and Quantization
cs.LG
Submitted: 2026-10-01
Updated: 2026-10-01
License: http://creativecommons.org/licenses/by-sa/4.0/
The gist: Federated Learning (FL) enables privacy-preserving fine-tuning of Large Language Models (LLMs), yet the massive communication overhead remains a critical bottleneck.
Terminology
Abstract
Federated Learning (FL) enables privacy-preserving fine-tuning of Large Language Models (LLMs), yet the massive communication overhead remains a critical bottleneck. Furthermore, applying Low-Rank Adaptation (LoRA) in FL faces a fundamental "aggregation dilemma" between the accurate Sum-of-Products (SoP) and the communication-efficient Product-of-Sums (PoS) implementations. To tackle these challenges, we propose FedFit. First, to significantly reduce communication overhead, we introduce a disjoint shared vector-bank parameterization that reconstructs high-dimensional adapter matrices from two compact and disjoint global vector banks. Second, to address the aggregation dilemma, we devise an alternating optimization schedule. By cycling between decoupled single-bank updates (which allow for accurate aggregation) and joint updates corrected by a Residual Spectral Aggregation mechanism, we resolve the conflict between SoP and PoS. Additionally, we integrate blockwise quantization with client-side error feedback to further compress the transmitted vectors. Furthermore, we establish theoretical convergence guarantees for the proposed algorithm. Extensive experiments on Qwen2.5 models demonstrate that FedFit achieves perplexity performance comparable to standard federated LoRA methods, while providing compression ratios up to 100x higher.
Sources
- Federated Large Language Models: Current Progress and Future Directions
- HAFLQ: Heterogeneous Adaptive Federated LoRA Fine-tuned LLM with Quantization
- A Survey on Federated Fine-tuning of Large Language Models
- FwdLLM: Efficient FedLLM using Forward Gradient
- Federated Full-Parameter Tuning of Billion-Sized Language Models with Communication Cost under 18 Kilobytes
- EcoLoRA: Communication-Efficient Federated Fine-Tuning of Large Language Models
- OLoRA: Orthonormal Low-Rank Adaptation of Large Language Models
- Uni-LoRA: One Vector is All You Need
- Improving LoRA in Privacy-preserving Federated Learning
- Convergence Analysis of Aggregation-Broadcast in LoRA-enabled Distributed Fine-Tuning
- Robust Federated Finetuning of Foundation Models via Alternating Minimization of LoRA
- signSGD with Majority Vote is Communication Efficient And Fault Tolerant
- ADF-LoRA: Alternating Low-Rank Aggregation for Decentralized Federated Fine-Tuning
- Qwen2.5 Technical Report
- Training Verifiers to Solve Math Word Problems
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks