轻量级联邦学习模型,高效识别番茄病害,兼顾隐私与性能。
U-FedTomAtt: Ultra-lightweight Federated Learning with Attention for Tomato Disease Recognition
- 采用稀疏瓶颈结构与线性注意力,极小化计算开销。
- 在多个数据集上实现超99%的准确率与F1分数。
- 适合资源受限的农田边缘设备部署,保护数据隐私。
联邦学习为部署智能农业解决方案提供了隐私保护且高效的途径。在地理分散的农场中实现精准的边缘诊断,对可持续农业中的番茄病害识别至关重要。传统集中式训练需将原始数据汇聚至中心服务器,带来通信开销、隐私风险和延迟问题。同时,边缘设备需轻量化网络以在有限资源下运行。本文提出U-FedTomAtt,一种面向资源受限与分布式环境的番茄病害识别超轻量级联邦学习框架。该模型仅含245.34K参数和71.41 MFLOPS。首先,设计一种结合空洞瓶颈(DBNeck)模块与线性变压器的超轻量神经网络,以降低计算与内存开销;为缓解精度损失,引入新颖的局部-全局残差注意力(LoGRA)模块。其次,提出联邦双自适应权重聚合(FedDAWA)算法,提升全局模型精度。第三,框架在三个番茄病害基准数据集上通过模拟联邦设置验证。实验结果表明,该方法在SLIF-Tomato和PlantVillage番茄数据集上分别达到0.9910%和0.9915%的Top-1准确率,以及0.9923%和0.9897%的F1分数。
原文摘要 · Abstract (English)
Federated learning has emerged as a privacy-preserving and efficient approach for deploying intelligent agricultural solutions. Accurate edge-based diagnosis across geographically dispersed farms is crucial for recognising tomato diseases in sustainable farming. Traditional centralised training aggregates raw data on a central server, leading to communication overhead, privacy risks and latency. Meanwhile, edge devices require lightweight networks to operate effectively within limited resources. In this paper, we propose U-FedTomAtt, an ultra-lightweight federated learning framework with attention for tomato disease recognition in resource-constrained and distributed environments. The model comprises only 245.34K parameters and 71.41 MFLOPS. First, we propose an ultra-lightweight neural network with dilated bottleneck (DBNeck) modules and a linear transformer to minimise computational and memory overhead. To mitigate potential accuracy loss, a novel local-global residual attention (LoGRA) module is incorporated. Second, we propose the federated dual adaptive weight aggregation (FedDAWA) algorithm that enhances global model accuracy. Third, our framework is validated using three benchmark datasets for tomato diseases under simulated federated settings. Experimental results show that the proposed method achieves 0.9910% and 0.9915% Top-1 accuracy and 0.9923% and 0.9897% F1-scores on SLIF-Tomato and PlantVillage tomato datasets, respectively.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。