通过随机空洞卷积与多分支结构提升图像分类精度,有效抑制噪声并捕捉细粒度特征。
Image Classification via Random Dilated Convolution with Multi-Branch Feature Extraction and Context Excitation
- 采用多分支随机空洞卷积,动态捕获多尺度特征并增强抗噪能力。
- 在五个数据集上准确率超越现有方法,最高提升达4.73%。
- 适合需要细粒度识别和背景抑制的图像分类任务。
图像分类是计算机视觉中的基础且具挑战性的任务,尤其在同时要求细粒度特征提取与背景噪声抑制时。传统卷积神经网络虽在层次化特征学习上表现优异,但往往难以捕捉多尺度上下文信息,且在面对含噪或无关区域时易过拟合。本文提出RDCNet(基于随机空洞卷积的图像分类网络),在ResNet-34基础上融合三项协同创新:(1) 多分支随机空洞卷积(MRDC)模块,通过不同膨胀率的并行分支结合随机掩码机制,实现多尺度细粒度特征捕捉,并增强对噪声和过拟合的鲁棒性;(2) 细粒度特征增强(FGFE)模块,在MRDC中引入自适应池化与双线性插值,连接全局上下文与局部特征表示,提升对细微视觉模式的敏感性;(3) 上下文激励(CE)模块,利用softmax空间注意力与通道重校准,动态强调任务相关特征并抑制背景干扰。在CIFAR-10、CIFAR-100、SVHN、Imagenette和Imagewoof五个基准数据集上的大量实验表明,RDCNet持续达到最优分类准确率,分别优于次优方法0.02%、1.12%、0.18%、4.73%和3.56%,验证了该方法在多样化视觉识别场景下的有效性与泛化能力。
原文摘要 · Abstract (English)
Image classification remains a fundamental yet challenging task in computer vision, particularly when fine-grained feature extraction and background noise suppression are required simultaneously. Conventional convolutional neural networks, despite their remarkable success in hierarchical feature learning, often struggle with capturing multi-scale contextual information and are susceptible to overfitting when confronted with noisy or irrelevant image regions. In this paper, we propose RDCNet (Image Classification Network with Random Dilated Convolution), a novel architecture built upon ResNet-34 that integrates three synergistic innovations to address these limitations: (1) a Multi-Branch Random Dilated Convolution (MRDC) module that employs parallel branches with varying dilation rates combined with a stochastic masking mechanism to capture fine-grained features across multiple scales while enhancing robustness against noise and overfitting; (2) a Fine-Grained Feature Enhancement (FGFE) module embedded within MRDC that bridges global contextual information with local feature representations through adaptive pooling and bilinear interpolation, thereby amplifying sensitivity to subtle visual patterns; and (3) a Context Excitation (CE) module that leverages softmax-based spatial attention and channel recalibration to dynamically emphasize task-relevant features while suppressing background interference. Extensive experiments conducted on five benchmark datasets -- CIFAR-10, CIFAR-100, SVHN, Imagenette, and Imagewoof -- demonstrate that RDCNet consistently achieves state-of-the-art classification accuracy, outperforming the second-best competing methods by margins of 0.02\%, 1.12\%, 0.18\%, 4.73\%, and 3.56\%, respectively, thereby validating the effectiveness and generalizability of the proposed approach across diverse visual recognition scenarios.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。