Image classification, the process of categorizing images into predefined classes, plays a vital role in everyday life and various computer applications. For individuals, it enhances daily experiences through features like facial recognition in security systems and smartphone photo organization, where algorithms automatically categorize images based on content, making it easier to search and manage large libraries. Convolutional Neural Networks (CNNs) fall into the category of deep neural networks which are especially effective for medical imaging and image classification problems due to their architecture, which is particularly suited to processing pixel data. This paper explores the critical influence of CNN hyperparameters on training performance. By systematically adjusting key hyperparameters namely the count of convolutional layers, filter sizes, and learning rates, we demonstrate how these variables affect the model's ability to detect and classify pathological features in three general public datasets like MNIST, Fashion MNIST and CIFAR-10 and one medical imaging dataset named Chest X-Ray.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Enhanced Hyperparameter Optimization in CNN for Image Classification

  • Saptarsi Panda,
  • Nibedan Panda,
  • Umang Kumar Agrawal,
  • Debabrata Singh,
  • Debabrata Samanta

摘要

Image classification, the process of categorizing images into predefined classes, plays a vital role in everyday life and various computer applications. For individuals, it enhances daily experiences through features like facial recognition in security systems and smartphone photo organization, where algorithms automatically categorize images based on content, making it easier to search and manage large libraries. Convolutional Neural Networks (CNNs) fall into the category of deep neural networks which are especially effective for medical imaging and image classification problems due to their architecture, which is particularly suited to processing pixel data. This paper explores the critical influence of CNN hyperparameters on training performance. By systematically adjusting key hyperparameters namely the count of convolutional layers, filter sizes, and learning rates, we demonstrate how these variables affect the model's ability to detect and classify pathological features in three general public datasets like MNIST, Fashion MNIST and CIFAR-10 and one medical imaging dataset named Chest X-Ray.