Deep Learning-Based Waste Classification with Transfer Learning Using EfficientNet-B0 Model
Recycling of waste is a significant challenge in modern waste management. Conventional techniques that use inductive and capacitive proximity sensors exhibit limitations in accuracy and flexibility for the detection of various types of waste. Indonesia generates approximately 175,000 tons of waste p...
Saved in:
| Main Authors: | , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
Ikatan Ahli Informatika Indonesia
2024-08-01
|
| Series: | Jurnal RESTI (Rekayasa Sistem dan Teknologi Informasi) |
| Subjects: | |
| Online Access: | https://jurnal.iaii.or.id/index.php/RESTI/article/view/5875 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1841544040190836736 |
|---|---|
| author | Risfendra Risfendra Gheri Febri Ananda Herlin Setyawan |
| author_facet | Risfendra Risfendra Gheri Febri Ananda Herlin Setyawan |
| author_sort | Risfendra Risfendra |
| collection | DOAJ |
| description | Recycling of waste is a significant challenge in modern waste management. Conventional techniques that use inductive and capacitive proximity sensors exhibit limitations in accuracy and flexibility for the detection of various types of waste. Indonesia generates approximately 175,000 tons of waste per day, highlighting the urgent need for efficient waste management solutions. This study develops a waste classification system based on deep learning, leveraging the powerful EfficientNet-B0 model through transfer learning. EfficientNet-B0 is designed with a compound scaling method, which uniformly scales network depth, width, and resolution, providing an optimal balance between accuracy and computational efficiency. The model was trained on a dataset containing six classes of waste—glass, cardboard, paper, metal, plastic, and residue—totalling 7014 images. The model was trained using data augmentation and fine-tuning techniques. The training results show a test accuracy of 91.94%, a precision of 92.10%, and a recall of 91.94%, resulting in an F1-score of 91.96%. Visualization of predictions demonstrates that the model effectively classifies waste in new test data. Implementing this model in the industry can automate the waste sorting process more efficiently and accurately than methods based on inductive and capacitive proximity sensors. This study underscores the significant potential of deep learning models, particularly EfficientNet-B0, in industrial waste classification applications and opens opportunities for further integration with sensor and robotic systems for more advanced waste management solutions. |
| format | Article |
| id | doaj-art-562af52da426404f9467953f47e07248 |
| institution | Kabale University |
| issn | 2580-0760 |
| language | English |
| publishDate | 2024-08-01 |
| publisher | Ikatan Ahli Informatika Indonesia |
| record_format | Article |
| series | Jurnal RESTI (Rekayasa Sistem dan Teknologi Informasi) |
| spelling | doaj-art-562af52da426404f9467953f47e072482025-01-13T03:33:02ZengIkatan Ahli Informatika IndonesiaJurnal RESTI (Rekayasa Sistem dan Teknologi Informasi)2580-07602024-08-018453554110.29207/resti.v8i4.58755875Deep Learning-Based Waste Classification with Transfer Learning Using EfficientNet-B0 ModelRisfendra Risfendra0Gheri Febri Ananda1Herlin Setyawan2Universitas Negeri PadangUniversitas Gadjah MadaUniversitas Negeri PadangRecycling of waste is a significant challenge in modern waste management. Conventional techniques that use inductive and capacitive proximity sensors exhibit limitations in accuracy and flexibility for the detection of various types of waste. Indonesia generates approximately 175,000 tons of waste per day, highlighting the urgent need for efficient waste management solutions. This study develops a waste classification system based on deep learning, leveraging the powerful EfficientNet-B0 model through transfer learning. EfficientNet-B0 is designed with a compound scaling method, which uniformly scales network depth, width, and resolution, providing an optimal balance between accuracy and computational efficiency. The model was trained on a dataset containing six classes of waste—glass, cardboard, paper, metal, plastic, and residue—totalling 7014 images. The model was trained using data augmentation and fine-tuning techniques. The training results show a test accuracy of 91.94%, a precision of 92.10%, and a recall of 91.94%, resulting in an F1-score of 91.96%. Visualization of predictions demonstrates that the model effectively classifies waste in new test data. Implementing this model in the industry can automate the waste sorting process more efficiently and accurately than methods based on inductive and capacitive proximity sensors. This study underscores the significant potential of deep learning models, particularly EfficientNet-B0, in industrial waste classification applications and opens opportunities for further integration with sensor and robotic systems for more advanced waste management solutions.https://jurnal.iaii.or.id/index.php/RESTI/article/view/5875waste classificationtransfer learningefficientnet-b0 |
| spellingShingle | Risfendra Risfendra Gheri Febri Ananda Herlin Setyawan Deep Learning-Based Waste Classification with Transfer Learning Using EfficientNet-B0 Model Jurnal RESTI (Rekayasa Sistem dan Teknologi Informasi) waste classification transfer learning efficientnet-b0 |
| title | Deep Learning-Based Waste Classification with Transfer Learning Using EfficientNet-B0 Model |
| title_full | Deep Learning-Based Waste Classification with Transfer Learning Using EfficientNet-B0 Model |
| title_fullStr | Deep Learning-Based Waste Classification with Transfer Learning Using EfficientNet-B0 Model |
| title_full_unstemmed | Deep Learning-Based Waste Classification with Transfer Learning Using EfficientNet-B0 Model |
| title_short | Deep Learning-Based Waste Classification with Transfer Learning Using EfficientNet-B0 Model |
| title_sort | deep learning based waste classification with transfer learning using efficientnet b0 model |
| topic | waste classification transfer learning efficientnet-b0 |
| url | https://jurnal.iaii.or.id/index.php/RESTI/article/view/5875 |
| work_keys_str_mv | AT risfendrarisfendra deeplearningbasedwasteclassificationwithtransferlearningusingefficientnetb0model AT gherifebriananda deeplearningbasedwasteclassificationwithtransferlearningusingefficientnetb0model AT herlinsetyawan deeplearningbasedwasteclassificationwithtransferlearningusingefficientnetb0model |