Communication Optimization for Decentralized Learning atop Bandwidth-limited Edge Networks

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Sun, Tingyang, Nguyen, Tuan, He, Ting
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916698125238272
author Sun, Tingyang
Nguyen, Tuan
He, Ting
author_facet Sun, Tingyang
Nguyen, Tuan
He, Ting
contents Decentralized federated learning (DFL) is a promising machine learning paradigm for bringing artificial intelligence (AI) capabilities to the network edge. Running DFL on top of edge networks, however, faces severe performance challenges due to the extensive parameter exchanges between agents. Most existing solutions for these challenges were based on simplistic communication models, which cannot capture the case of learning over a multi-hop bandwidth-limited network. In this work, we address this problem by jointly designing the communication scheme for the overlay network formed by the agents and the mixing matrix that controls the communication demands between the agents. By carefully analyzing the properties of our problem, we cast each design problem into a tractable optimization and develop an efficient algorithm with guaranteed performance. Our evaluations based on real topology and data show that the proposed algorithm can reduce the total training time by over $80\%$ compared to the baseline without sacrificing accuracy, while significantly improving the computational efficiency over the state of the art.
format Preprint
id arxiv_https___arxiv_org_abs_2504_12210
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Communication Optimization for Decentralized Learning atop Bandwidth-limited Edge Networks
Sun, Tingyang
Nguyen, Tuan
He, Ting
Networking and Internet Architecture
Artificial Intelligence
Distributed, Parallel, and Cluster Computing
Machine Learning
Decentralized federated learning (DFL) is a promising machine learning paradigm for bringing artificial intelligence (AI) capabilities to the network edge. Running DFL on top of edge networks, however, faces severe performance challenges due to the extensive parameter exchanges between agents. Most existing solutions for these challenges were based on simplistic communication models, which cannot capture the case of learning over a multi-hop bandwidth-limited network. In this work, we address this problem by jointly designing the communication scheme for the overlay network formed by the agents and the mixing matrix that controls the communication demands between the agents. By carefully analyzing the properties of our problem, we cast each design problem into a tractable optimization and develop an efficient algorithm with guaranteed performance. Our evaluations based on real topology and data show that the proposed algorithm can reduce the total training time by over $80\%$ compared to the baseline without sacrificing accuracy, while significantly improving the computational efficiency over the state of the art.
title Communication Optimization for Decentralized Learning atop Bandwidth-limited Edge Networks
topic Networking and Internet Architecture
Artificial Intelligence
Distributed, Parallel, and Cluster Computing
Machine Learning
url https://arxiv.org/abs/2504.12210