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

Hetero-SplitEE:带有早期退出的神经网络分裂学习用于异构物联网设备

Hetero-SplitEE: Split Learning of Neural Networks with Early Exits for Heterogeneous IoT Devices

The University of Tokyo(东京大学)

Yuki Oda, Yuta Ono, Hiroshi Nakamura, Hideki Takase

AI总结:

Hetero-SplitEE通过引入异构早期退出和协作训练策略,实现异构物联网设备的高效协同神经网络训练。

摘要:

深度神经网络的持续扩展已从根本上改变了机器学习,更大的模型在多样化的任务中表现出改进的性能。这种模型规模的增长极大地增加了训练过程所需的计算资源。因此,分布式方法,如联邦学习和分裂学习,已成为可扩展部署的重要范式。然而,现有的分裂学习方法假设客户端同质性和所有参与者中的均匀分裂点。这极大地限制了它们在现实世界物联网系统中的应用,其中设备在计算资源上表现出异质性。为了解决这一限制,本文提出了Hetero-SplitEE,一种新的方法,使异构物联网设备能够并行协作地训练共享的深度神经网络。通过在分层训练中整合异构早期退出,我们的方法允许每个客户端选择不同的分裂点(切层),以适应其计算能力。此外,我们提出了两种协作训练策略,顺序策略和平均策略,以促进具有不同分裂点的客户端之间的协作。顺序策略通过共享服务器模型依次训练客户端以减少计算开销。平均策略允许并行客户端训练,并在周期性地进行跨层聚合。在使用ResNet-18对CIFAR-10、CIFAR-100和STL-10数据集进行的大量实验中,我们的方法在保持竞争性准确度的同时,高效地支持多样化的计算约束,使协作深度学习在异构物联网生态系统中的实际部署成为可能。

英文摘要:

The continuous scaling of deep neural networks has fundamentally transformed machine learning, with larger models demonstrating improved performance across diverse tasks. This growth in model size has dramatically increased the computational resources required for the training process. Consequently, distributed approaches, such as Federated Learning and Split Learning, have become essential paradigms for scalable deployment. However, existing Split Learning approaches assume client homogeneity and uniform split points across all participants. This critically limits their applicability to real-world IoT systems where devices exhibit heterogeneity in computational resources. To address this limitation, this paper proposes Hetero-SplitEE, a novel method that enables heterogeneous IoT devices to train a shared deep neural network in parallel collaboratively. By integrating heterogeneous early exits into hierarchical training, our approach allows each client to select distinct split points (cut layers) tailored to its computational capacity. In addition, we propose two cooperative training strategies, the Sequential strategy and the Averaging strategy, to facilitate this collaboration among clients with different split points. The Sequential strategy trains clients sequentially with a shared server model to reduce computational overhead. The Averaging strategy enables parallel client training with periodic cross-layer aggregation. Extensive experiments on CIFAR-10, CIFAR-100, and STL-10 datasets using ResNet-18 demonstrate that our method maintains competitive accuracy while efficiently supporting diverse computational constraints, enabling practical deployment of collaborative deep learning in heterogeneous IoT ecosystems.

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

↑