arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2601.04727cs.CVcs.NE

在五个异质图像数据集上训练定制的CNN

Training a Custom CNN on Five Heterogeneous Image Datasets

Anika Tabassum, Tasnuva Mahazabin Tuba, Nafisa Naznin

AI总结:

本研究提出了一种高效的定制CNN,在多个异质图像数据集上实现高性能,并通过比较分析揭示了迁移学习在数据受限场景中的优势。

摘要:

深度学习已改变了视觉数据分析,卷积神经网络(CNNs)在从图像中学习有意义的特征表示方面变得非常有效。与传统的手动特征工程方法不同,CNNs自动提取层次化的视觉模式,使在各种现实世界场景中取得优异表现成为可能。本研究探讨了基于CNN的架构在五个异质数据集上的有效性,这些数据集涵盖了农业和城市领域:芒果品种分类、水稻品种识别、道路表面状况评估、两轮汽车检测和小径侵占监控。这些数据集引入了不同的挑战,包括光照差异、分辨率、环境复杂性和类别不平衡,需要适应性强且稳健的学习模型。我们评估了一个轻量级、任务特定的定制CNN以及已建立的深度架构,包括ResNet-18和VGG-16,这些模型既从头开始训练,也使用迁移学习。通过系统化的预处理、增强和受控实验,我们分析了架构复杂性、模型深度和预训练如何影响收敛性、泛化能力和在不同规模和难度的数据集上的性能。本研究的主要贡献是:(1)开发了一个高效的定制CNN,能够在多个应用领域中取得竞争性的性能;(2)全面的比较分析,突显了在数据受限环境中迁移学习和深度架构提供的显著优势。这些发现为在资源有限但影响重大的现实世界视觉分类任务中部署深度学习模型提供了实用见解。

英文摘要:

Deep learning has transformed visual data analysis, with Convolutional Neural Networks (CNNs) becoming highly effective in learning meaningful feature representations directly from images. Unlike traditional manual feature engineering methods, CNNs automatically extract hierarchical visual patterns, enabling strong performance across diverse real-world contexts. This study investigates the effectiveness of CNN-based architectures across five heterogeneous datasets spanning agricultural and urban domains: mango variety classification, paddy variety identification, road surface condition assessment, auto-rickshaw detection, and footpath encroachment monitoring. These datasets introduce varying challenges, including differences in illumination, resolution, environmental complexity, and class imbalance, necessitating adaptable and robust learning models. We evaluate a lightweight, task-specific custom CNN alongside established deep architectures, including ResNet-18 and VGG-16, trained both from scratch and using transfer learning. Through systematic preprocessing, augmentation, and controlled experimentation, we analyze how architectural complexity, model depth, and pre-training influence convergence, generalization, and performance across datasets of differing scale and difficulty. The key contributions of this work are: (1) the development of an efficient custom CNN that achieves competitive performance across multiple application domains, and (2) a comprehensive comparative analysis highlighting when transfer learning and deep architectures provide substantial advantages, particularly in data-constrained environments. These findings offer practical insights for deploying deep learning models in resource-limited yet high-impact real-world visual classification tasks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑