arXiv AI
Oct 2

FedLore: Communication and Memory Efficient Federated Learning via Shared Gradient Low-Rank Projection

FedLore introduces a communication- and memory-efficient federated learning framework that shares a low-rank optimization basis across clients each round, mitigating subspace fragmentation and enabling exact low-rank aggregation. By refreshing this shared basis across rounds, FedLore allows model updates to exceed the per-round rank budget while maintaining a provable $O(T^{-1/2})$ stationarity bound under standard assumptions. Experiments on vision and language tasks, including federated pre‑training, demonstrate that FedLore outperforms low‑rank adapter baselines and matches or surpasses full‑parameter training while reducing communication and optimizer‑state memory.

By Junkang Liu
arXiv AI
Jul 21

Sparsity-Aware Low-Rank Representation for Efficient Fine-Tuning of Large Language Models

arXiv:2601. 16991v3 Announce Type: replace-cross Abstract: Adapting large pre-trained language models to downstream tasks often entails fine-tuning millions of parameters or deploying costly dense weight updates, which hinders their use in resource-constrained environments.

By Longteng Zhang, Sen Wu, Shuai Hou, Zhengyu Qing, Zhuo Zheng, Danning Ke, Qihong Lin, Qiang Wang, Shaohuai Shi, Xiaowen Chu
arXiv Machine Learning
Oct 2

Importance-Aware Feature Sparsification for Wireless Split Learning

The paper introduces Importance-Aware Class-Balanced Sparsification (ICS) for wireless split learning, a lightweight method that ranks feature channels on the server using Grad‑CAM scores from the true‑class logit. By aggregating per‑class scores into a class‑balanced importance vector, clients can retain the top‑N feature channels without extra forward or backward passes, reducing communication overhead while mitigating head‑class bias under label skew. The authors provide a non‑asymptotic convergence bound, analyze overhead against baselines, and extend the approach to parallel split learning and transformer models, demonstrating consistent performance gains, especially under severe non‑i.i.d. data partitions.

By Bumjun Kim, Yoon Huh, Wan Choi