arrow
返回

Task Selection and Resource Optimization in Multi-Task Federated Learning With Model Decomposition

delete2025-01-01
delete0
PRE
AI
H
Haowen Sun
M
Ming Chen *
杨朝晖 封面图
杨朝晖 (Zhaohui Yang) *
潘
潘怡瑾 (Yijin Pan)
Y
Yihan Cang
张
张朝阳 (Zhaoyang Zhang)
DOI:10.1109/LCOMM.2024.3511663delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
In this letter, we investigate the training latency minimization problem for a multi-task federated learning (FL) framework with model decomposition over wireless communication networks. To handle the non-independent and non-identically distributed (non-IID) data, we first transform the multi-class classification task into multiple binary classification tasks. We then introduce sampling equalization to ensure the convergence of FL system. The optimization problem aims to minimize the training latency under energy and FL convergence constraints by optimizing task selection, number of learning iterations, and communication resource allocation. We decompose it into three sub-problems and propose alternating algorithm to address each sub-problem iteratively. Numerical results validate that the proposed algorithm significantly reduces time consumption compared to the conventional algorithms.
Keyword:
Convergence
Data models
Uplink
Vectors
Training
Federated learning
Bandwidth
Resource management
Radio frequency
Distributed databases
Multi-task federated learning
resource allocation
non-IID data

期刊

IEEE Communications Letters 封面图
IEEE Communications Letters
IF:
4.4
论文数:
1.3W
被引数:
2.2W

机构

S
southeast university - china
学者数:
5.3W
论文数: 4.9W
被引数: 57
引用论文

引用论文

err
IF0
err
err0
PREAI
err
err分享
err收藏
Low-Latency Federated Learning With DNN Partition in Distributed Industrial IoT Networks
err2023-03-01
err21
errOAAI
errDeng, Xiumei; Li, Jun; Ma, Chuan; Wei, Kang; Shi, Long; Ding, Ming; Chen, Wen
err分享
err收藏
Energy Efficient Federated Learning Over Wireless Communication Networks无线通信网络上的节能联合学习
err2021-03-01
err549
errOAAI
errYang, Zhaohui; Chen, Mingzhe; Saad, Walid; Hong, Choong Seon; Shikh-Bahaei, Mohammad
err分享
err收藏
Federated Multi-Task Learning with Non-Stationary and Heterogeneous Data in Wireless Networks
err2024-04-01
err1
PREAI
errZhang, Hongwei; Tao, Meixia; Shi, Yuanming; Bi, Xiaoyan; Letaief, Khaled B.
err分享
err收藏
err分享
err收藏
没有更多内容