Benchmark analysis of various pre-trained deep learning models on ASSIRA cats and dogs dataset

被引:0
|
作者
Galib Muhammad Shahriar Himel [1 ]
Md. Masudul Islam [2 ]
机构
[1] Universiti Sains Malaysia,School of Computer Sciences
[2] 11800 USM,Department of Computer Science and Engineering
[3] Jahangirnagar University,undefined
关键词
Convolutional neural network; Machine learning; Artificial intelligence; Image classification; Data augmentation; Cats vs dogs;
D O I
10.1007/s43995-024-00094-w
中图分类号
学科分类号
摘要
Image classification using deep learning has gained significant attention, with various datasets available for benchmarking algorithms and pre-trained models. This study focuses on the Microsoft ASIRRA dataset, renowned for its quality and benchmark standards, to compare different pre-trained models. Through experimentation with optimizers, loss functions, and hyperparameters, this research aimed to enhance model performance. Notably, this study achieved significant accuracy improvements with minimal modifications to the training process. Experiments were conducted across three computer architectures, yielding superior accuracy results compared to previous studies on this dataset. The NASNet Large model emerged with the highest accuracy at 99.65%. The findings of this research demonstrate the effectiveness of hyperparameter tuning for renowned pre-trained models, suggesting optimal settings for improved classification accuracy. This study underscores the potential of deep learning approaches in achieving superior performance by hyperparameter tuning for image classification tasks.
引用
收藏
页码:134 / 149
页数:15
相关论文
共 50 条
  • [1] Exploratory Architectures Analysis of Various Pre-trained Image Classification Models for Deep Learning
    Deepa, S.
    Zeema, J. Loveline
    Gokila, S.
    JOURNAL OF ADVANCES IN INFORMATION TECHNOLOGY, 2024, 15 (01) : 66 - 78
  • [2] Unregistered Multiview Mammogram Analysis with Pre-trained Deep Learning Models
    Carneiro, Gustavo
    Nascimento, Jacinto
    Bradley, Andrew P.
    MEDICAL IMAGE COMPUTING AND COMPUTER-ASSISTED INTERVENTION, PT III, 2015, 9351 : 652 - 660
  • [3] Classification and Analysis of Pistachio Species with Pre-Trained Deep Learning Models
    Singh, Dilbag
    Taspinar, Yavuz Selim
    Kursun, Ramazan
    Cinar, Ilkay
    Koklu, Murat
    Ozkan, Ilker Ali
    Lee, Heung-No
    ELECTRONICS, 2022, 11 (07)
  • [4] EVALUATION OF DIFFERENT PARAMETERS FOR PLANT CLASSIFICATION BY PRE-TRAINED DEEP LEARNING MODELS WITH BIGEARTHNET DATASET
    Naali, F.
    Alipour-Fard, T.
    Arefi, H.
    ISPRS GEOSPATIAL CONFERENCE 2022, JOINT 6TH SENSORS AND MODELS IN PHOTOGRAMMETRY AND REMOTE SENSING, SMPR/4TH GEOSPATIAL INFORMATION RESEARCH, GIRESEARCH CONFERENCES, VOL. 10-4, 2023, : 569 - 574
  • [5] Analysis of Layer Efficiency and Layer Reduction on Pre-trained Deep Learning Models
    Nugraha, Brilian Tafjira
    Su, Shun-Feng
    2018 INTERNATIONAL CONFERENCE ON SYSTEM SCIENCE AND ENGINEERING (ICSSE), 2018,
  • [6] Adversarial Attacks on Pre-trained Deep Learning Models for Encrypted Traffic Analysis
    Seok, Byoungjin
    Sohn, Kiwook
    JOURNAL OF WEB ENGINEERING, 2024, 23 (06): : 749 - 768
  • [7] Classification and Analysis of Agaricus bisporus Diseases with Pre-Trained Deep Learning Models
    Albayrak, Umit
    Golcuk, Adem
    Aktas, Sinan
    Coruh, Ugur
    Tasdemir, Sakir
    Baykan, Omer Kaan
    AGRONOMY-BASEL, 2025, 15 (01):
  • [8] Microstructure segmentation with deep learning encoders pre-trained on a large microscopy dataset
    Joshua Stuckner
    Bryan Harder
    Timothy M. Smith
    npj Computational Materials, 8
  • [9] Microstructure segmentation with deep learning encoders pre-trained on a large microscopy dataset
    Stuckner, Joshua
    Harder, Bryan
    Smith, Timothy M.
    NPJ COMPUTATIONAL MATERIALS, 2022, 8 (01)
  • [10] Class-Incremental Learning Based on Big Dataset Pre-Trained Models
    Wen, Bin
    Zhu, Qiuyu
    IEEE ACCESS, 2023, 11 : 62028 - 62038