Tram-FL: Routing-based Model Training for Decentralized Federated Learning
Maejima, Kota, Nishio, Takayuki, Yamazaki, Asato, Hara-Azumi, Yuko
–arXiv.org Artificial Intelligence
In decentralized federated learning (DFL), substantial traffic from frequent inter-node communication and non-independent and identically distributed (non-IID) data challenges high-accuracy model acquisition. We propose Tram-FL, a novel DFL method, which progressively refines a global model by transferring it sequentially amongst nodes, rather than by exchanging and aggregating local models. We also introduce a dynamic model routing algorithm for optimal route selection, aimed at enhancing model precision with minimal forwarding. Our experiments using MNIST, CIFAR-10, and IMDb datasets demonstrate that Tram-FL with the proposed routing delivers high model accuracy under non-IID conditions, outperforming baselines while reducing communication costs.
arXiv.org Artificial Intelligence
Aug-9-2023
- Country:
- South America > Chile
- North America
- United States > Oregon (0.04)
- Canada > Ontario
- Toronto (0.04)
- Asia
- Middle East > UAE
- Abu Dhabi Emirate > Abu Dhabi (0.14)
- Japan > Honshū
- Kantō > Tokyo Metropolis Prefecture > Tokyo (0.04)
- Middle East > UAE
- Genre:
- Research Report (0.83)
- Technology: