Research Article Open Access

Performance Analysis of Garbage Classification Using Balanced and Unbalanced Dataset with EfficientNetV2B1 Architecture

Retinderdeep Singh1, Chander Prabha1, Ayman Noor2, Mohammad Zubair Khan3, Prakash Srivastava4 and Gurjot Kaur1
  • 1 Department of Computer Science and Engineering, Chitkara University Institute of Engineering and Technology, Chitkara University, Punjab, India
  • 2 Department of Computer Science, College of Computer Science and Engineering, Taibah University, Madinah, Saudi Arabia
  • 3 Department of Computer Science and Information, Taibah University, Madinah, Saudi Arabia
  • 4 Department of Computer Science and Engineering, Graphic Era University, Dehradun, India

Abstract

This research concentrates on EfficientNetV2B1 deep learning model to classify garbage collection tasks, with both balanced and unbalanced dataset configurations. The dataset includes 7,260 images for the balanced dataset and 15,515 for the unbalanced one, both of the datasets are used to train and evaluate the model. Training and evaluation of the deep learning model with the standard performance variables: Accuracy, precision, recall, F1-score, and AUC score. The results indicate the unbalanced dataset performs excellently, with an accuracy of 95.22%, precision at 95.28%, recall of 95.22%, and F1-score of 95.21%. In contrast, the fully balanced data set yields slightly less but still excellent results: Accuracy of 91.46%, precision of 91.60%, recall of 91.46%, and F1 score of 91.43%. The test scores for loss and accuracy in both datasets are 0.4296 and 0.9522 for the unbalanced dataset and 0.60189 and 0.9146 for the balanced dataset respectively. A study with a dataset containing balanced classes is beneficial for assessing EfficientNetV2B1 deep learning performance across different classes evenly, providing a fair evaluation of the model's ability to generalize. On the other hand, a study with an unbalanced class distribution can be useful for evaluating how well EfficientNetV2B1 deep learning handles class imbalance and its performance in minority classes. Both types of studies offer valuable insights into model behavior under different data scenarios.

Journal of Computer Science
Volume 20 No. 10, 2024, 1310-1321

DOI: https://doi.org/10.3844/jcssp.2024.1310.1321

Submitted On: 3 May 2024 Published On: 27 August 2024

How to Cite: Singh, R., Prabha, C., Noor, A., Khan, M. Z., Srivastava, P. & Kaur, G. (2024). Performance Analysis of Garbage Classification Using Balanced and Unbalanced Dataset with EfficientNetV2B1 Architecture. Journal of Computer Science, 20(10), 1310-1321. https://doi.org/10.3844/jcssp.2024.1310.1321

  • 473 Views
  • 212 Downloads
  • 0 Citations

Download

Keywords

  • Garbage Classification
  • EfficientNetV2B1
  • Deep Learning
  • Image Classification
  • Performance Analysis