Wireless Federated Multi-Task LLM Fine-Tuning via Sparse-and-Orthogonal LoRA
Organizations: Beijing Laboratory of Advanced Information Network, and the Beijing Key Laboratory of Network System Architecture and Convergence, Beijing University of Posts and Telecommunications, Beijing 100876, China · Department of Electrical and Computer Engineering and Institute for Data Science and Computing, University of Miami, Coral Gables, FL, 33146 USA · Information Systems Technology and Design Pillar, Singapore University of Technology and Design, 487372, Singapore
Abstract
Decentralized federated learning (DFL) based on low-rank adaptation (LoRA) enables mobile devices with multi-task datasets to collaboratively fine-tune a large language model (LLM) by exchanging locally updated parameters with a subset of neighboring devices via wireless connections for knowledge integration.However, directly aggregating parameters fine-tuned on heterogeneous datasets induces three primary issues across the DFL life-cycle: (i) catastrophic knowledge forgetting during fine-tuning process, arising from conflicting update directions caused by data heterogeneity; (ii) inefficient communication and convergence during model aggregation process, due to bandwidth-intensive redundant model transmissions; and (iii) multi-task knowledge interference during inference process, resulting from incompatible knowledge representations coexistence during inference. To address these issues in a fully decentralized scenario, we first propose a sparse-and-orthogonal LoRA that ensures orthogonality between model updates to eliminate direction conflicts during fine-tuning.Then, we analyze how device connection topology affects multi-task performance, prompting a cluster-based topology design during aggregation.Finally, we propose an implicit mixture of experts (MoE) mechanism to avoid the coexistence of incompatible knowledge during inference. Simulation results demonstrate that the proposed approach effectively reduces communication resource consumption by up to and enhances average performance by compared with the traditional LoRA method.
Figures & tables
| Method | # Params (%) | # Trans. (%) | BoolQ | Piqa | SocI.qa | GSM8K | ARC-E. | ARC-C. | Hella.S. | HumanE. | Avg. |
| Qwen2.5-1.5B-Instruct | |||||||||||
| FPFT | 3.2G (100%) | 3.2G (100%) | 79.81 | 80.72 | 79.16 | 77.83 | 84.49 | 64.2 | 87.35 | 62.05 | 76.95 |
| LoRA | 65.5M (2.01%) | 65.5M (2.01%) | 76.94 | 78.59 | 79.20 | 71.90 | 62.30 | 52.00 | 68.30 | 47.90 | 68.03 |
| LoRI | 32M (1.05%) | 32M (1.05%) | 79.29 | 81.48 | 78.64 | 74.12 | 82.78 | 62.11 | 84.4 | 58.58 | 75.17 |
| Hard-routing MoE | 3.2G (100%) | 3.2G (100%) | 83.23 | 84.87 | 83.04 | 82.44 | 87.56 | 69.12 | 90.91 | 66.29 | 80.93 |
| Proposed (Single) | 16M (0.54%) | 8M (0.27%) | 80.21 | 81.69 | 78.38 | 78.21 | 82.95 | 64.25 | 84.74 | 61.09 | 76.44 |
| Method | # Params (%) | BoolQ | Piqa | SocI.qa | GSM8K | ARC-E. | ARC-C. | Hella.S. | HumanE. | Avg. |
| Qwen2.5-1.5B-Instruct | ||||||||||
| FPFT | 3.2G (100%) | 83.23 | 84.87 | 83.04 | 82.44 | 87.56 | 69.12 | 90.91 | 66.29 | 80.93 |
| LoRA | 65.5M (2.01%) | 80.09 | 83.65 | 79.33 | 75.78 | 84.41 | 63.38 | 85.15 | 60.5 | 76.53 |
| LoRI | 32M (1.05%) | 81.72 | 84.19 | 80.82 | 77.07 | 85.14 | 64.93 | 86.47 | 61.17 | 77.69 |
| Proposed Method | 16M (0.54%) | 82.88 | 84.54 | 81.25 | 81.36 | 86.30 | 67.02 | 87.63 | 63.94 | 79.37 |
| Qwen2.5-7B-Instruct | ||||||||||
Federated Fine-Tuning
Federated fine-tuning adapts pretrained models across decentralized data holders without pooling their private data.
See all 115 papers