用强化学习动态调整农业虫害识别模型的分割点,提升边缘设备效率。
ReinDSplit: Reinforced Dynamic Split Learning for Pest Recognition in Precision Agriculture
- 基于强化学习动态选择每台设备的神经网络分割位置。
- 在三个数据集上使用MobileNetV2达到94.31%准确率。
- 适合计算能力差异大的农业物联网场景,兼顾隐私与性能。
为推动分布式机器学习在精准农业中的应用,分层学习(SL)将深度神经网络(DNN)在边缘设备与服务器间划分,以减轻计算负担并保护数据隐私。然而,传统SL框架采用统一分割策略,在农业环境中因边缘昆虫监测设备存在显著的计算能力、能耗和连接性差异,导致执行延迟、资源利用低效及模型性能下降。为此,本文提出ReinDSplit,一种基于强化学习(RL)的动态分层学习框架,可为每台设备自适应调整DNN分割点,优化效率而不损失精度。具体而言,一个Q-learning代理作为动态调度器,在设备间平衡负载与延迟阈值,避免计算饥饿或过载。通过将分割层选择建模为有限状态马尔可夫决策过程,确保资源受限设备能持续参与训练。在三个昆虫分类数据集上,使用ResNet18、GoogleNet和MobileNetV2进行评估,ReinDSplit在MobileNetV2上实现94.31%的准确率。该方法不仅适用于农业场景,更在异构环境下开创了融合强化学习实现资源高效、隐私保护与可扩展性的新范式。
原文摘要 · Abstract (English)
To empower precision agriculture through distributed machine learning (DML), split learning (SL) has emerged as a promising paradigm, partitioning deep neural networks (DNNs) between edge devices and servers to reduce computational burdens and preserve data privacy. However, conventional SL frameworks' one-split-fits-all strategy is a critical limitation in agricultural ecosystems where edge insect monitoring devices exhibit vast heterogeneity in computational power, energy constraints, and connectivity. This leads to straggler bottlenecks, inefficient resource utilization, and compromised model performance. Bridging this gap, we introduce ReinDSplit, a novel reinforcement learning (RL)-driven framework that dynamically tailors DNN split points for each device, optimizing efficiency without sacrificing accuracy. Specifically, a Q-learning agent acts as an adaptive orchestrator, balancing workloads and latency thresholds across devices to mitigate computational starvation or overload. By framing split layer selection as a finite-state Markov decision process, ReinDSplit convergence ensures that highly constrained devices contribute meaningfully to model training over time. Evaluated on three insect classification datasets using ResNet18, GoogleNet, and MobileNetV2, ReinDSplit achieves 94.31% accuracy with MobileNetV2. Beyond agriculture, ReinDSplit pioneers a paradigm shift in SL by harmonizing RL for resource efficiency, privacy, and scalability in heterogeneous environments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。