用多尺度胶囊网络提升野生水果识别准确率与可解释性
Multi-Scale Fruit Capsules: Dilated Convolutions and Dynamic Routing for In-the-Wild Explainable Fruit Recognition

- 用空洞卷积构建多尺度胶囊,动态路由捕捉整体结构关系
- 在3个公开数据集和新数据集上优于10个主流模型,最高超2.7%
- 可视化显示决策基于完整果实区域,非边缘,解释性强
同一种水果可能以成串、未采摘、去皮、塑料袋装或切片上桌等多种形态出现,因此野外水果自动分类需应对形状、大小、颜色和纹理的广泛类内变化与微小类间差异。传统卷积网络通过池化传递信息,会丢失感兴趣区域的方位和位置信息,导致跨呈现方式泛化能力差。我们提出FruitCapsNet,其水果胶囊取代标准卷积前端,采用空洞卷积:感受野呈指数增长且参数成本不变,使每个胶囊在动态路由前编码多尺度上下文。超参数(包括膨胀因子)通过贝叶斯优化选择,而非网格搜索。在三个公开数据集(SMP、FruitsGB、Fruits-360)及新构建的19类、10,639张图像的野外数据集(PD-19)上,FruitCapsNet在仅三分之一深度下超越十个微调的迁移学习骨干网络,尤其在最难数据集上领先第二名2.7%。从DigitCaps层传播的Grad-CAM显著性图表明,性能提升源于将决策归因于完整果实区域而非物体边缘,为效果非数据偏差提供了事后证据。
原文摘要 · Abstract (English)
The same fruit appears in a bunch, unpicked, peeled, bagged in plastic, or sliced on a dish, so automated fruit classification in the wild (AFCW) must absorb wide intra- class and narrow inter-class variability in shape, size, colour and texture. Convolutional networks route information through pooling, which discards the pose and location of the region of interest and therefore generalises poorly across these presentations. We propose FruitCapsNet, a capsule network whose Fruit Capsules replace the standard convolutional front end with dilated convolutions: the receptive field grows exponentially at constant parameter cost, so each capsule encodes multi-scale context before dynamic routing resolves part whole spatial agreement. Hyper-parameters, including the dilation factor, are selected by Bayesian optimisation rather than grid search. On three public datasets (SMP, FruitsGB, Fruits-360) and a new 19-class, 10,639-image in-the-wild dataset (PD-19), FruitCapsNet exceeds ten fine-tuned transfer-learning backbones at one-third the depth, with the largest margin (+2.7% over the nearest competitor) on the hardest set. Grad-CAM saliency propagated from the DigitCaps layer shows that the improvement comes from attributing decisions to whole-fruit regions rather than to object edges, giving post-hoc evidence that the gain is not a dataset artefact.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。