CycleSL通过循环更新提升分片学习的性能与扩展性。
CycleSL: Server-Client Cyclical Update Driven Scalable Split Learning
- 采用服务器-客户端循环优化,避免模型聚合与资源冗余。
- 在5个非独立同分布数据集上,显著提升模型准确率与收敛速度。
- 适合资源受限场景下高效协作训练,可无缝集成现有框架。
分片学习作为一种有前景的分布式协同训练范式,通过在客户端与服务器间分割神经网络实现无原始数据交换的联合训练。然而,顺序式分片学习存在可扩展性差的问题,而并行变体如并行分片学习和分片联邦学习常因模型复制与聚合导致高服务器资源开销,且受客户端漂移与延迟影响,普遍出现性能下降与收敛缓慢。为此,我们提出一种新型无聚合分片学习框架CycleSL,显著提升可扩展性与性能,并可无缝集成至现有方法中。受交替块坐标下降启发,CycleSL将服务器端训练视为独立的高层次机器学习任务,通过对客户端提取的特征(被压缩的数据)进行重采样以缓解异质性与漂移问题。随后执行循环更新:先优化服务器模型,再利用更新后的服务器模型计算梯度以更新客户端。我们将CycleSL集成到已有算法中,在五个公开数据集上进行测试,数据分布为非独立同分布且存在部分客户端缺席。实验结果表明,CycleSL在提升模型性能方面效果显著。源代码已开源:https://gitlab.lrz.de/hctl/CycleSL。
原文摘要 · Abstract (English)
Split learning emerges as a promising paradigm for collaborative distributed model training, akin to federated learning, by partitioning neural networks between clients and a server without raw data exchange. However, sequential split learning suffers from poor scalability, while parallel variants like parallel split learning and split federated learning often incur high server resource overhead due to model duplication and aggregation, and generally exhibit reduced model performance and convergence owing to factors like client drift and lag. To address these limitations, we introduce CycleSL, a novel aggregation-free split learning framework that enhances scalability and performance and can be seamlessly integrated with existing methods. Inspired by alternating block coordinate descent, CycleSL treats server-side training as an independent higher-level machine learning task, resampling client-extracted features (smashed data) to mitigate heterogeneity and drift. It then performs cyclical updates, namely optimizing the server model first, followed by client updates using the updated server for gradient computation. We integrate CycleSL into previous algorithms and benchmark them on five publicly available datasets with non-iid data distribution and partial client attendance. Our empirical findings highlight the effectiveness of CycleSL in enhancing model performance. Our source code is available at https://gitlab.lrz.de/hctl/CycleSL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。