arXiv:2410.12718cs.CV2024-10被引 16

提出区域注意力网络,提升食物与农业病害识别精度

RAFA-Net: Region Attention Network For Food Items And Agricultural Stress Recognition

  • 通过区域间相关性建模长程依赖,增强上下文特征表达
  • 在3个食物数据集上最高达96.97%准确率,农业病害识别也领先
  • 无需额外参数,适合资源受限场景下的精准识别任务

深度卷积神经网络在识别各类食物和农业胁迫方面取得了显著进展。通过挖掘和分析基于区域的部分特征描述符,性能得到明显提升。以往工作还研究了使用多个CNN的计算开销较大的集成学习方法。本文提出一种区域注意力机制,通过建立输入图像内不同区域间的相关性来建模长程依赖。该注意力方法通过学习互补区域的上下文信息有用性来增强特征表示。空间金字塔池化与平均池化组合将局部描述符聚合为整体表征,二者均在不增加额外参数的前提下建立空间与通道关系。引入上下文门控机制以优化加权注意力特征的描述能力,有助于分类。所提出的用于食物与农业胁迫识别的区域注意力网络RAFA-Net,在三个公开食物数据集上均达到最先进性能,最高top-1准确率分别为91.69%(UECFood-100)、91.56%(UECFood-256)和96.97%(MAFood-121)。此外,在两个基准农业胁迫数据集上也取得更优结果:昆虫害虫(IP-102)最高准确率为92.36%,PlantDoc-27植物病害数据集为85.54%,表明其良好的泛化能力。

原文摘要 · Abstract (English)

Deep Convolutional Neural Networks (CNNs) have facilitated remarkable success in recognizing various food items and agricultural stress. A decent performance boost has been witnessed in solving the agro-food challenges by mining and analyzing of region-based partial feature descriptors. Also, computationally expensive ensemble learning schemes using multiple CNNs have been studied in earlier works. This work proposes a region attention scheme for modelling long-range dependencies by building a correlation among different regions within an input image. The attention method enhances feature representation by learning the usefulness of context information from complementary regions. Spatial pyramidal pooling and average pooling pair aggregate partial descriptors into a holistic representation. Both pooling methods establish spatial and channel-wise relationships without incurring extra parameters. A context gating scheme is applied to refine the descriptiveness of weighted attentional features, which is relevant for classification. The proposed Region Attention network for Food items and Agricultural stress recognition method, dubbed RAFA-Net, has been experimented on three public food datasets, and has achieved state-of-the-art performances with distinct margins. The highest top-1 accuracies of RAFA-Net are 91.69%, 91.56%, and 96.97% on the UECFood-100, UECFood-256, and MAFood-121 datasets, respectively. In addition, better accuracies have been achieved on two benchmark agricultural stress datasets. The best top-1 accuracies on the Insect Pest (IP-102) and PlantDoc-27 plant disease datasets are 92.36%, and 85.54%, respectively; implying RAFA-Net's generalization capability.

图像识别农业检测注意力机制

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。