用强化学习动态选模型分层点,让不同设备高效协作训练。
QSplitFL: Capability Aware Deep Q-Learning for Optimal Split Point Selection in Split Federated Learning

- 根据设备性能实时选最优分层位置,避免资源过载。
- 在多个数据集上收敛更快,准确率最高提升2.3%。
- 适合边缘计算、手机等异构设备的联邦学习场景。
联邦学习(FL)与分割学习(SL)结合可实现资源受限设备上的隐私保护深度神经网络训练,并降低整体训练成本。然而,在客户端硬件能力异构的情况下,确定最优模型分层点(即模型分割位置)仍是关键挑战。固定分层点可能导致弱设备过载,增加通信和服务器负载,从而减慢收敛速度并降低稳定性。本文提出QSplitFL,一种基于深度强化学习的新型能力感知框架,用于分割联邦学习(SFL)环境下的最优分层点选择。不同于依赖高维模型权重表示的方法,QSplitFL采用直接来自客户端硬件指标(如CPU利用率、内存、电池电量、网络延迟)的轻量级状态表示。该框架引入衰减损失下降奖励函数以优先考虑早期收敛,并采用基于委员会的DQN架构与多数投票机制来缓解奖励欺骗问题。在MNIST、Fashion-MNIST、CIFAR-10和CIFAR-100数据集上,使用CNN、ResNet50、MobileNetV4和ConvNeXt架构的大量实验表明,该方法在收敛速度和准确率方面优于现有方法,同时能有效适应异构设备资源。代码已开源:https://github.com/AIPO-Lab/QSplitFL。
原文摘要 · Abstract (English)
Federated Learning (FL) combined with Split Learning (SL) is a privacy preserving paradigm that enables training deep neural networks (DNNs) on resource constrained devices while reducing overall training cost. However, determining the optimal split point, meaning the layer where the model is divided still remains a critical challenge, especially when clients have heterogeneous hardware capabilities. Fixed split points can overload weak devices and increase the communication and server load, which slows convergence and reduces stability. This paper introduces QSplitFL, a novel capability-aware Deep Q-Network (DQN) framework for optimal split point selection in Split learning based Federated Learning (SFL) environments. Unlike existing approaches that rely on high-dimensional model weight representations, QSplitFL employs a lightweight state representation derived directly from client hardware metrics, including CPU utilization, memory, battery level, and network latency. The proposed framework incorporates a decayed loss-drop reward function that prioritizes early convergence, and a committee-based DQN architecture with majority voting to mitigate reward hacking. Extensive experiments on MNIST, Fashion-MNIST, CIFAR-10, and CIFAR-100 datasets using CNN, ResNet50, MobileNetV4, and ConvNeXt architectures demonstrate that our approach achieves better convergence and higher accuracy compared to existing methods, while effectively adapting to heterogeneous device resources. The source code is publicly available at https://github.com/AIPO-Lab/QSplitFL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。