Smart waste classification in IoT-enabled smart cities using VGG16 and Cat Swarm Optimized random forest

  • Akshat Gaurav
  • , Brij Bhooshan Gupta
  • , Varsha Arya
  • , Razaz Waheeb Attar
  • , Shavi Bansal
  • , Ahmed Alhomoud
  • , Kwok Tai Chui

Research output: Contribution to journalArticlepeer-review

3 Citations (Scopus)

Abstract

Effective waste management is becoming a crucial component of sustainable urban development as smart technologies are used by smart cities more and more. Smart trash categorization systems provided by IoT may greatly enhance garbage sorting and recycling mechanisms. In this context, this work presents a waste categorization model based on transfer learning using the VGG16 model for feature extraction and a Random Forest classifier tuned by Cat Swarm Optimization (CSO). On a Kaggle garbage categorization dataset, the model outperformed conventional models like SVM, XGBoost, and logistic regression. With an accuracy of 85% and a high AUC of 0.85 the Random Forest model shows better performance in precision, recall, and F1-score as compared to standard machine learning models.

Original languageEnglish
Article numbere0316930
JournalPLoS ONE
Volume20
Issue number2 February
DOIs
Publication statusPublished - Feb 2025

Fingerprint

Dive into the research topics of 'Smart waste classification in IoT-enabled smart cities using VGG16 and Cat Swarm Optimized random forest'. Together they form a unique fingerprint.

Cite this