An Efficient Model for Waste Image Classification Using EfficientNet-B0
DOI:
https://doi.org/10.52436/1.jutif.2025.6.3.4417Keywords:
Classification, CNN, Efficientnet, Waste Image, Waste SortingAbstract
Waste management remains a significant challenge, particularly in developing countries. To address this issue, artificial intelligence can be leveraged to develop a waste image classifier that facilitates automatic waste sorting. Previous studies have explored the use of Convolutional Neural Networks (CNNs) for waste image classification. However, CNNs typically require a large number of parameters, leading to increased computational time. For practical applications, a waste image classifier must not only achieve high accuracy but also operate efficiently. Therefore, this study aims to develop an accurate and computationally efficient waste image classification model using EfficientNet-B0. EfficientNet-B0 is a CNN architecture designed to achieve high accuracy while maintaining an efficient number of parameters. This study utilized the publicly available TrashNet dataset and investigated the impact of image augmentation in addressing imbalance data issues. The highest performance was achieved by the model trained on the unbalanced dataset with the addition of a Dense(32) layer, a dropout rate of 0.3, and a learning rate of 1e-4. This configuration achieved an accuracy of 0.885 and an F1-score of 0.87. These results indicate that the inclusion of a Dense(32) layer prior to the output layer consistently improves model performance, whereas image augmentation does not yield a significant enhancement. Furthermore, our proposed model achieved the highest accuracy while maintaining a significantly lower number of parameters compared to other CNN architectures with comparable accuracy, such as ResNet-50 and Xception. The resulting waste classification model can then be further implemented to build an automatic waste sorter.
Downloads
References
Z. Zhang et al., “Municipal solid waste management challenges in developing regions: A comprehensive review and future perspectives for Asia and Africa,” Jun. 20, 2024, Elsevier B.V. doi: 10.1016/j.scitotenv.2024.172794.
United Nations Environment Programme, Global Waste management Outlook 2024: Beyond an age of waste - Turning rubbish into resource. Nairobi: United Nations Environment Programme, 2024. doi: https://doi.org/10.59117/20.500.11822/44939.
D. V. Yevle and P. S. Mann, “Artificial intelligence based classification for waste management: A survey based on taxonomy, classification & future direction,” May 01, 2025, Elsevier Ireland Ltd. doi: 10.1016/j.cosrev.2024.100723.
Khadijah, S. N. Endah, R. Kusumaningrum, Rismiyati, P. S. Sasongko, and I. Z. Nisa, “Solid waste classification using pyramid scene parsing network segmentation and combined features,” Telkomnika (Telecommunication Computing Electronics and Control), vol. 19, no. 6, pp. 1902–1912, 2021, doi: 10.12928/TELKOMNIKA.v19i6.18402.
L. Chen, S. Li, Q. Bai, J. Yang, S. Jiang, and Y. Miao, “Review of image classification algorithms based on convolutional neural networks,” Nov. 01, 2021, MDPI. doi: 10.3390/rs13224712.
R. Hanieh, S. Elham, and S. B. Mehdi, “Enhancing breast cancer detection in thermographic images using deep hybrid networks,” Imaging and Radiation Research, vol. 7, no. 1, p. 6195, Jul. 2024, doi: 10.24294/irr6195.
A. Peryanto, A. Yudhana, and R. Umar, “Convolutional Neural Network and Support Vector Machine in Classification of Flower Images,” Khazanah Informatika: Jurnal Ilmu Komputer dan Informatika, vol. 8, no. 1, pp. 1–7, 2022, doi: 10.23917/khif.v8i1.15531.
Y. Lecun, Y. Bengio, and G. Hinton, “Deep learning,” Nature, vol. 521, no. 7553, pp. 436–444, 2015, doi: 10.1038/nature14539ï.
R. Sidharth, P. Rohit, S. Vishagan, R. Karthika, and M. Ganesan, “Deep Learning based Smart Garbage Classifier for Effective Waste Management,” in 2020 5th International Conference on Communication and Electronics Systems (ICCES), Coimbatore, India: IEEE, Jun. 2020, pp. 1086–1089. doi: 10.1109/ICCES48766.2020.9137938.
A. Altikat, A. Gulbe, and S. Altikat, “Intelligent solid waste classification using deep convolutional neural networks,” International Journal of Environmental Science and Technology, vol. 19, no. 3, pp. 1285–1292, Mar. 2022, doi: 10.1007/s13762-021-03179-4.
R. Faria, F. Ahmed, A. Das, and A. Dey, “Classification of Organic and Solid Waste Using Deep Convolutional Neural Networks,” in 2021 IEEE 9th Region 10 Humanitarian Technology Conference (R10-HTC), Bangalore, India: IEEE, Sep. 2021, pp. 01–06. doi: 10.1109/R10-HTC53172.2021.9641560.
A. Gaurav et al., “Smart waste classification in IoT-enabled smart cities using VGG16 and Cat Swarm Optimized random forest,” PLoS One, vol. 20, no. 2 February, Feb. 2025, doi: 10.1371/journal.pone.0316930.
D. Gyawali, A. Regmi, A. Shakya, A. Gautam, and S. Shrestha, “Comparative Analysis of Multiple Deep CNN Models for Waste Classification,” Apr. 2020, [Online]. Available: http://arxiv.org/abs/2004.02168
O. Adedeji and Z. Wang, “Intelligent waste classification system using deep learning convolutional neural network,” in Procedia Manufacturing, Elsevier B.V., 2019, pp. 607–612. doi: 10.1016/j.promfg.2019.05.086.
Rismiyati, S. N. Endah, Khadijah, and I. N. Shiddiq, “Xception Architecture Transfer Learning for Garbage Classification,” in The 4th International Conference on Informatics and Computational Sciences (ICICoS 2020), IEEE, 2020. doi: 10.1109/ICICoS51170.2020.9299017.
M. Fan, K. Zuo, J. Wang, and J. Zhu, “A lightweight multiscale convolutional neural network for garbage sorting,” Systems and Soft Computing, vol. 5, Dec. 2023, doi: 10.1016/j.sasc.2023.200059.
M. Tan and Q. V. Le, “EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks,” in International Conference on Machine Learning, Long Beach, USA, Jun. 2019, pp. 6105–6114. Accessed: May 14, 2024. [Online]. Available: https://proceedings.mlr.press/v97/tan19a.html
S. M. Baik, K. S. Hong, and D. J. Park, “Deep learning approach for early prediction of COVID-19 mortality using chest X-ray and electronic health records,” BMC Bioinformatics, vol. 24, no. 1, Dec. 2023, doi: 10.1186/s12859-023-05321-0.
Khadijah, R. Kusumaningrum, Rismiyati, and A. Mujadidurrahman, “An Efficient Masked Face Classifier Using EfficientNet,” in 2021 5th International Conference on Informatics and Computational Sciences (ICICoS), Semarang: IEEE, Nov. 2021, pp. 277–281. doi: 10.1109/ICICoS53627.2021.9651883.
Ü. Atila, M. Uçar, K. Akyol, and E. Uçar, “Plant leaf disease classification using EfficientNet deep learning model,” Ecol Inform, vol. 61, Mar. 2021, doi: 10.1016/j.ecoinf.2020.101182.
D. R. Nayak, N. Padhy, P. K. Mallick, M. Zymbler, and S. Kumar, “Brain Tumor Classification Using Dense Efficient-Net,” Axioms, vol. 11, no. 1, Jan. 2022, doi: 10.3390/axioms11010034.
K. M. Chung et al., “Deep Learning-Based Knee MRI Classification for Common Peroneal Nerve Palsy with Foot Drop,” Biomedicines, vol. 11, no. 12, Dec. 2023, doi: 10.3390/biomedicines11123171.
V. H. Barella, L. P. F. Garcia, M. C. P. de Souto, A. C. Lorena, and A. C. P. L. F. de Carvalho, “Assessing the data complexity of imbalanced datasets,” Inf Sci (N Y), vol. 553, pp. 83–109, Apr. 2021, doi: 10.1016/j.ins.2020.12.006.
G. E. A. P. A. Batista, R. C. Prati, and M. C. Monard, “A study of the behavior of several methods for balancing machine learning training data,” ACM SIGKDD Explorations Newsletter, vol. 6, no. 1, pp. 20–29, Jun. 2004, doi: 10.1145/1007730.1007735.
N. E. Khalifa, M. Loey, and S. Mirjalili, “A comprehensive survey of recent trends in deep learning for digital images augmentation,” Artif Intell Rev, vol. 55, no. 3, pp. 2351–2377, Mar. 2022, doi: 10.1007/s10462-021-10066-4.
M. Yang and G. Thung, “Classification of Trash for Recyclability Status,” 2016. Accessed: May 13, 2024. [Online]. Available: https://cs229.stanford.edu/proj2016/report/ThungYang-ClassificationOfTrashForRecyclabilityStatus-report.pdf
M. Yani, B. Irawan, and C. Setiningsih, “Application of Transfer Learning Using Convolutional Neural Network Method for Early Detection of Terry’s Nail,” in Journal of Physics: Conference Series, Institute of Physics Publishing, May 2019. doi: 10.1088/1742-6596/1201/1/012052.
C. Shorten and T. M. Khoshgoftaar, “A survey on Image Data Augmentation for Deep Learning,” J Big Data, vol. 6, no. 1, Dec. 2019, doi: 10.1186/s40537-019-0197-0.
S. Ravichandiran, Hands-On Deep Learning Algorithms with Python. Packt Publishing Ltd, 2019.
T. Ganegedara, Natural Language Processing with TensorFlow. Birmingham: Packt Publishing Ltd, 2018.
X. Deng, Q. Liu, Y. Deng, and S. Mahadevan, “An improved method to construct basic probability assignment based on the confusion matrix for classification problem,” Inf Sci (N Y), vol. 340–341, pp. 250–261, May 2016, doi: 10.1016/j.ins.2016.01.033.
Additional Files
Published
How to Cite
Issue
Section
License
Copyright (c) 2025 Teofilus Kurniawan, Khadijah, Retno Kusumaningrum
This work is licensed under a Creative Commons Attribution 4.0 International License.
