arXiv:2601.04352cs.CVcs.LG2026-01

对比自建CNN与预训练模型在孟加拉五数据集上的表现,发现微调效果更优。

Comparative Analysis of Custom CNN Architectures versus Pre-trained Models and Transfer Learning: A Study on Five Bangladesh Datasets

  • 用迁移学习微调预训练模型,性能优于从零构建的CNN
  • 微调使准确率提升3%至76%,部分数据集达100%准确率
  • 适合资源有限但需高精度图像分类的实践者参考

本研究系统比较了自建卷积神经网络(CNN)与主流预训练模型(ResNet-18和VGG-16)在特征提取和迁移学习两种方式下的表现。评估基于孟加拉国五个不同图像分类数据集:Footpath Vision、Auto Rickshaw Detection、Mango Image Classification、Paddy Variety Recognition 和 Road Damage Detection。实验结果表明,采用微调的迁移学习方法始终优于从零开始构建的CNN及特征提取法,在各数据集上准确率提升达3%至76%。特别地,使用微调的ResNet-18在Road Damage BD数据集上达到100%准确率。尽管自建CNN参数量更小(340万),训练效率更高,但在复杂任务和数据有限时,预训练模型结合迁移学习仍具明显优势。该研究为根据数据特性、算力与性能需求选择合适深度学习方案提供了实用指导。

原文摘要 · Abstract (English)

This study presents a comprehensive comparative analysis of custom-built Convolutional Neural Networks (CNNs) against popular pre-trained architectures (ResNet-18 and VGG-16) using both feature extraction and transfer learning approaches. We evaluated these models across five diverse image classification datasets from Bangladesh: Footpath Vision, Auto Rickshaw Detection, Mango Image Classification, Paddy Variety Recognition, and Road Damage Detection. Our experimental results demonstrate that transfer learning with fine-tuning consistently outperforms both custom CNNs built from scratch and feature extraction methods, achieving accuracy improvements ranging from 3% to 76% across different datasets. Notably, ResNet-18 with fine-tuning achieved perfect 100% accuracy on the Road Damage BD dataset. While custom CNNs offer advantages in model size (3.4M parameters vs. 11-134M for pre-trained models) and training efficiency on simpler tasks, pre-trained models with transfer learning provide superior performance, particularly on complex classification tasks with limited training data. This research provides practical insights for practitioners in selecting appropriate deep learning approaches based on dataset characteristics, computational resources, and performance requirements.

图像分类迁移学习CNN孟加拉数据

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