arXiv:2502.18691cs.CV2025-02被引 18

11种数据增强方法提升小样本图像分类,新方法显著降低过拟合。

Enhancing Image Classification with Augmentation: Data Augmentation Techniques for Improved Image Classification

  • 提出三种新型增强:通道值迁移、对象遮挡、多类型掩码
  • 在Caltech-101上,集成增强使EfficientNet-B0准确率提升明显
  • 适合小数据集图像分类任务,对避免过拟合有实际帮助

卷积神经网络(CNN)是深度学习的核心,在依赖图像的众多领域广泛应用。然而,当训练数据量较小时,CNN容易过拟合。通过数据增强可引入更多学习样本,显著提升性能。本研究评估了11种数据增强技术的效果,包括三种新提出的方案:第一种为通道值迁移,将随机图像的红、绿、蓝、色相、饱和度值转移到所有图像;第二种为新型遮挡法,用数据集中随机选择的物体遮挡原图目标;第三种为新型掩码法,使用竖向、横向、圆形和棋盘格掩码遮挡图像部分区域。此外,还考察了旋转、水平/垂直翻转、缩放、平移、模糊、色彩抖动、随机擦除等现有方法的影响。针对每种增强策略微调基础EfficientNet-B0模型,并在Caltech-101数据集上进行对比分析。结果表明,所提出的增强组合在该数据集上表现最优,证明多样化的数据增强是提升图像分类性能的有效手段。

原文摘要 · Abstract (English)

Convolutional Neural Networks (CNNs) serve as the workhorse of deep learning, finding applications in various fields that rely on images. Given sufficient data, they exhibit the capacity to learn a wide range of concepts across diverse settings. However, a notable limitation of CNNs is their susceptibility to overfitting when trained on small datasets. The augmentation of such datasets can significantly enhance CNN performance by introducing additional data points for learning. In this study, we explore the effectiveness of 11 different sets of data augmentation techniques, which include three novel sets proposed in this work. The first set of data augmentation employs pairwise channel transfer, transferring Red, Green, Blue, Hue, and Saturation values from randomly selected images in the database to all images in the dataset. The second set introduces a novel occlusion approach, where objects in the images are occluded by randomly selected objects from the dataset. The third set involves a novel masking approach, using vertical, horizontal, circular, and checkered masks to occlude portions of the images. In addition to these novel techniques, we investigate other existing augmentation methods, including rotation, horizontal and vertical flips, resizing, translation, blur, color jitter, and random erasing, and their effects on accuracy and overfitting. We fine-tune a base EfficientNet-B0 model for each augmentation method and conduct a comparative analysis to showcase their efficacy. For the evaluation and comparison of these augmentation techniques, we utilize the Caltech-101 dataset. The ensemble of image augmentation techniques proposed emerges as the most effective on the Caltech-101 dataset. The results demonstrate that diverse data augmentation techniques present a viable means of enhancing datasets for improved image classification.

图像分类数据增强小样本学习EfficientNet

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