arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2603.09792cs.LGcs.AI
收藏

利用自适应通道剪枝实现高效的分裂学习通信

Exploiting Adaptive Channel Pruning for Communication-Efficient Split Learning

School of Physics and Information Engineering, Fuzhou University(福州市物理与信息工程学院);Department of Electrical and Electronic Engineering, The University of Hong Kong(香港大学电子与电气工程系);School of Electrical Engineering and Telecommunications, University of New South Wales Sydney(新南威尔士大学悉尼分校电子与电信工程学院);Hong Kong JC STEM Lab of Smart City and the Department of Computer Science, City University of Hong Kong(香港智慧城市JC STEM实验室及城市大学计算机科学系);Data61, CSIRO, Marsfield, NSW 2122, Australia, and the School of Computing Science and Engineering, and the University of New South Wales, Kennington, NSW 2052, Australia(Data61, CSIRO, 澳大利亚马尔斯菲尔德, 新南威尔士州2122, 和计算科学与工程学院, 新南威尔士大学肯宁顿, 新南威尔士州2052, 澳大利亚)

Jialei Tan, Zheng Lin, Xiangming Cai, Ruoxi Zhu, Zihan Fang, Pingping Chen, Wei Ni

AI总结:

本文提出自适应通道剪枝辅助分裂学习方案,通过设计标签感知通道重要性评分模块和自适应通道剪枝模块,压缩中间特征表示以减少通信开销,实验表明其在测试准确率和训练轮次上均优于基准方案。

摘要:

分裂学习(SL)将大部分训练工作转移到服务器,从而减轻客户端设备的计算负担。然而,中间特征表示的传输(称为压碎数据)带来了显著的通信开销,尤其是在涉及大量客户端设备时。为了解决这一挑战,我们提出了一种自适应通道剪枝辅助的分裂学习(ACP-SL)方案。在ACP-SL中,设计了一个标签感知通道重要性评分(LCIS)模块,用于生成通道重要性评分,区分重要通道和不重要通道。基于这些评分,开发了一个自适应通道剪枝(ACP)模块,用于剪枝不重要通道,从而压缩相应的压碎数据并减少通信开销。实验结果表明,ACP-SL在测试准确率上始终优于基准方案。此外,它在更少的训练轮次内达到目标测试准确率,从而减少通信开销。

英文摘要:

Split learning (SL) transfers most of the training workload to the server, which alleviates computational burden on client devices. However, the transmission of intermediate feature representations, referred to as smashed data, incurs significant communication overhead, particularly when a large number of client devices are involved. To address this challenge, we propose an adaptive channel pruning-aided SL (ACP-SL) scheme. In ACP-SL, a label-aware channel importance scoring (LCIS) module is designed to generate channel importance scores, distinguishing important channels from less important ones. Based on these scores, an adaptive channel pruning (ACP) module is developed to prune less important channels, thereby compressing the corresponding smashed data and reducing the communication overhead. Experimental results show that ACP-SL consistently outperforms benchmark schemes in test accuracy. Furthermore, it reaches a target test accuracy in fewer training rounds, thereby reducing communication overhead.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑