On Exploiting Network Topology for Hierarchical Coded Multi-Task Learning

doi:10.1109/TCOMM.2024.3381671

On Exploiting Network Topology for Hierarchical Coded Multi-Task Learning

2024-08-01

1

PRE

AI

DOI：10.1109/TCOMM.2024.3381671

Original

Original request for help

Save

Abstract

En 中文

Distributed multi-task learning (MTL) is a learning paradigm where distributed users simultaneously learn multiple tasks by leveraging the correlations among tasks. However, distributed MTL suffers from a more severe communication bottleneck than single-task learning as more than one models need to be transmitted in the communication phase. To address this issue, we investigate the hierarchical MTL system where distributed users wish to jointly learn different learning models orchestrated by a central server with the help of multiple relays. We propose a coded distributed computing scheme for hierarchical MTL systems that jointly exploits the network topology and relays' computing capability to create coded multicast opportunities to improve communication efficiency. We theoretically prove that the proposed scheme can significantly reduce the communication loads both in the uplink and downlink transmissions between relays and the server. To further illustrate the optimality of the proposed scheme, we derive information-theoretic lower bounds on the minimum uplink and downlink communication loads and prove that the gaps between achievable upper bounds and lower bounds are within the minimum number of connected users among all relays. In particular, when the network topology can be delicately designed, the proposed scheme can achieve the information-theoretic optimal communication loads. Experiments on real-world datasets show that our proposed scheme can greatly reduce the overall training time compared to the conventional hierarchical MTL scheme.

Keywords:

Relays

Servers

Task analysis

Distance learning

Computer aided instruction

Downlink

Uplink

Multi-task learning

coding techniques

distributed learning

hierarchical systems

communication load

On Exploiting Network Topology for Hierarchical Coded Multi-Task Learning

Abstract

Journal

Organization