arXiv Machine Learning

D-FROST: Decentralized Federated pRompt-tuning via Optimal tranSporT for Non-IID and Imbalanced Data

The paper introduces D-FROST, an optimal‑transport‑based algorithm for decentralized federated prompt‑tuning, addressing challenges of heterogeneous local data and non‑IID distributions. It formulates prompt tuning as a Wasserstein optimization over prompt measures and merges neighborhood prompts into compact representative sets. The authors provide theoretical guarantees on consensus error and convergence, and demonstrate effectiveness through experiments on heterogeneous client data.

arXiv Machine Learning
Oct 3

SeedFlood: A Step Toward Scalable Decentralized Fine-Tuning of LLMs

SeedFlood is a novel decentralized fine‑tuning method for large language models that scales to billions of parameters and hundreds of clients. It leverages the seed‑reconstructible structure of zeroth‑order gradients to reduce message sizes to near‑zero, enabling efficient flooding across the network. Experiments show SeedFlood outperforms standard zeroth‑order baselines in communication efficiency and generalization, and rivals first‑order gossip methods while incurring far less communication cost.

By Jihun Kim, Dongyeop Lee, Namhoon Lee