arXiv AI

SplitLite: Low-Rank Residual Compression for Split Learning

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
arXiv AI
Jun 19

Techniques for Peak Memory Reduction for LoRA Fine-tuning of LLMs on Edge Devices

arXiv:2606. 19528v1 Announce Type: cross Abstract: Fine-tuning of Large Language Models (LLMs) using Low-Rank Adaptation (LoRA) on an end-user's data offers personalized experiences while keeping data private, but faces severe memory constraints on consumer hardware.

By Hassan Dbouk, Matthias Reisser, Prathamesh Mandke, Likhita Arun Navali, Christos Louizos
arXiv Machine Learning
Oct 8

LRCC: Generalizing Low-Rank Compression with Conditional Computation

The paper introduces Low‑Rank Conditional Computation (LRCC), a method that adds token‑dependent computation to pretrained language models by training lightweight routers to select among nested low‑rank paths while keeping low‑rank factors frozen. LRCC is evaluated on Llama and Qwen models, showing improved predictive performance over static low‑rank compression, including a 7.6‑percentage‑point gain in downstream accuracy on Llama‑2‑7B. The study also demonstrates that LRCC enhances perplexity and downstream accuracy on Llama‑3.2‑1B at matched decoding latency and analyzes the routers’ path choices to assess token‑wise path usefulness.

By Thomas Vaitses Fontanari, Maximo Eduardo Rulli, Federico Alvetreti, Donatella Genovese, Simone Scardapane