Abstract
Efficient waste management is essential for human well-being and environmental health, as neglecting proper disposal practices can lead to financial losses and the depletion of natural resources. Given the rapid urbanization and population growth, developing an automated, innovative waste classification model becomes imperative. To address this need, our paper introduces a novel and robust solution — a smart waste classification model that leverages a hybrid deep learning model (Optimized DenseNet-121 + SVM) to categorize waste items using the TrashNet datasets. Our proposed approach uses the advanced deep learning model DenseNet-121, optimized for superior performance, to extract meaningful features from an expanded TrashNet dataset. These features are subsequently fed into a support vector machine (SVM) for precise classification. Employing data augmentation techniques further enhances classification accuracy while mitigating the risk of overfitting, especially when working with limited TrashNet data. The results of our experimental evaluation of this hybrid deep learning model are highly promising, with an impressive accuracy rate of 99.84%. This accuracy surpasses similar existing models, affirming the efficacy and potential of our approach to revolutionizing waste classification for a sustainable and cleaner future.
Similar content being viewed by others
Data availability
The data and materials are available within the manuscript.
Code availability
Not applicable
References
Adekunle, T. S., Adeleke, T. A., Sunday, O., Ebong, G. N., & Bamisaye, T. A. (2023). A framework for robust attack detection and classification using rap-densenet. ParadigmPlus, 1–17.
Ajagbe, S. A., & Adigun, M. O. (2023). Deep learning techniques for detection and prediction of pandemic diseases: A systematic literature review. Multimedia Tools and Applications, 1–35.
Al Duhayyim, M., Eisa, T. A. E., Al-Wesabi, F. N., Abdelmaboud, A., Hamza, M. A., Zamani, A. S., ... Marzouk, R. (2022). Deep reinforcement learning enabled smart city recycling waste object classification. Computers, Materials and Continua, 71, 5699–5715.
Ali, M. A., PP, F. R., & Salama Abd Elminaam, D. (2022). A feature selection based on improved artificial hummingbird algorithm using random opposition-based learning for solving waste classification problem. Mathematics, 10(15), 2675.
Alqahtani, F., Al-Makhadmeh, Z., Tolba, A., & Said, W. (2020a). Internet of things-based urban waste management system for smart cities using a cuckoo search algorithm. Cluster Computing, 23, 1769–1780.
Alqahtani, F., Al-Makhadmeh, Z., Tolba, A., & Said, W. (2020b). Internet of things-based urban waste management system for smart cities using a cuckoo search algorithm. Cluster Computing, 23, 1769–1780.
Al-Salem, S., Lettieri, P., & Baeyens, J. (2009). Recycling and recovery routes of plastic solid waste (psw): A review. Waste Management, 29(10), 2625–2643.
Alsubaei, F. S., Al-Wesabi, F. N., & Hilal, A. M. (2022). Deep learning-based small object detection and classification model for garbage waste management in smart cities and iot environment. Applied Sciences, 12(5), 2281.
Aral, R. A., Keskin, Ş. R., Kaya, M., & Hacıömeroğlu, M. (2018). Classification of trashnet dataset based on deep learning models. 2018 ieee international conference on big data (big data) (pp. 2058–2062).
Bircanoğlu, C., Atay, M., Beşer, F., Genç, Ö., & Kızrak, M. A. (2018). Recyclenet: Intelligent waste sorting using deep neural networks. 2018 innovations in intelligent systems and applications (inista) (pp. 1–7).
Cheng, C., Ahmad, S. F., Irshad, M., Alsanie, G., Khan, Y., Ahmad, A. Y. B., & Aleemi, A. R. (2023). Impact of green process innovation and productivity on sustainability: The moderating role of environmental awareness. Sustainability, 15(17), 12945.
Chu, Y., Huang, C., Xie, X., Tan, B., Kamal, S., Xiong, X., et al. (2018). Multilayer hybrid deep-learning method for waste classification and recycling. Computational Intelligence and Neuroscience, 2018.
Costa, B. S., Bernardes, A. C., Pereira, J. V., Zampa, V. H., Pereira, V. A., Matos, G. F., ... Silva, A. F. (2018). Artificial intelligence in automated sorting in trash recycling. Anais do xv encontro nacional de inteligência artificial e computacional (pp. 198–205).
He, K., Zhang, X., Ren, S., & Sun, J. (2016). Deep residual learning for image recognition. Proceedings of the ieee conference on computer vision and pattern recognition (pp. 770–778).
Hoornweg, D., & Bhada-Tata, P. (2012). What a waste: A global review of solid waste management.
Huang, G., Liu, Z., Van Der Maaten, L., & Weinberger, K. Q. (2017). Densely connected convolutional networks. Proceedings of the ieee conference on computer vision and pattern recognition (pp. 4700–4708).
Ji, Q., Huang, J., He, W., & Sun, Y. (2019). Optimized deep convolutional neural networks for identification of macular diseases from optical coherence tomography images. Algorithms, 12(3), 51.
Kaza, S., Yao, L., Bhada-Tata, P., & Van Woerden, F. (2018). What a waste 2.0: a global snapshot of solid waste management to 2050. World Bank Publications
Kollikkathara, N., Feng, H., & Stern, E. (2009). A purview of waste management evolution special emphasis on usa. Waste Management, 29(2), 974–985.
Krizhevsky, A., Sutskever, I., & Hinton, G. E. (2017). Imagenet classification with deep convolutional neural networks. Communications of the ACM, 60(6), 84–90.
Lin, K., Zhou, T., Gao, X., Li, Z., Duan, H., Wu, H., ... Zhao, Y. (2022). Deep convolutional neural networks for construction and demolition waste classification: Vggnet structures, cyclical learning rate, and knowledge transfer. Journal of Environmental Management, 318, 115501.
Mao, W.-L., Chen, W.-C., Wang, C.-T., & Lin, Y.-H. (2021). Recycling waste classification using optimized convolutional neural network. Resources, Conservation and Recycling, 164, 105132.
Meng, S., & Chu, W.-T. (2020). A study of garbage classification with convolutional neural networks. 2020 indo–taiwan 2nd international conference on computing, analytics and networks (indo-taiwan ican) (pp. 152–157).
Meyer, D., & Wien, F. (2001). Support vector machines. R News, 1(3), 23–26.
Ni, L., Ahmad, S. F., Alshammari, T. O., Liang, H., Alsanie, G., Irshad, M., ... others (2023). The role of environmental regulation and green human capital towards sustainable development: The mediating role of green innovation and industry upgradation. Journal of Cleaner Production, 421, 138497.
Ojo, O. S., Oyediran, M. O., Bamgbade, B. J., Adeniyi, A. E., Ebong, G. N., & Ajagbe, S. A. (2023). Development of an improved convolutional neural network for an automated face based university attendance system. ParadigmPlus, 4(1), 18–28.
Ruiz, V., Sánchez, Á., Vélez, J. F., & Raducanu, B. (2019). Automatic image-based waste classification. From bioinspired systems and biomedical applications to machine learning: 8th international work-conference on the interplay between natural and artificial computation, iwinac 2019, almería, spain, june 3–7, 2019, proceedings, part ii 8 (pp. 422–431).
Satvilkar, M. (2018). Image based trash classification using machine learning algorithms for recyclability status (Unpublished doctoral dissertation). National College of Ireland: Dublin.
Shi, C., Tan, C., Wang, T., & Wang, L. (2021). A waste classification method based on a multilayer hybrid convolution neural network. Applied Sciences, 11(18), 8572.
Simonyan, K., & Zisserman, A. (2014). Very deep convolutional networks for large-scale image recognition. arXiv preprintarXiv:1409.1556
Sousa, J., Rebelo, A., & Cardoso, J. S. (2019). Automation of waste sorting with deep learning. 2019 xv workshop de visão computacional (wvc) (pp. 43–48).
Sudha, S., Vidhyalakshmi, M., Pavithra, K., Sangeetha, K., & Swaathi, V. (2016). An automatic classification method for environment: Friendly waste segregation using deep learning. 2016 ieee technological innovations in ict for agriculture and rural development (tiar) (p. 65–70).
Uganya, G., Rajalakshmi, D., Teekaraman, Y., Kuppusamy, R., & Radhakrishnan, A. (2022). A novel strategy for waste prediction using machine learning algorithm with iot based intelligent waste management system. Wireless Communications and Mobile Computing, 2022
Vo, A. H., Vo, M. T., Le, T., et al. (2019). A novel framework for trash classification using deep transfer learning. IEEE Access, 7, 178631–178639.
Williams, P. T. (2005). Waste treatment and disposal. John Wiley & Sons.
Xie, S., Girshick, R., Dollár, P., Tu, Z., & He, K. (2017). Aggregated residual transformations for deep neural networks. Proceedings of the ieee conference on computer vision and pattern recognition (pp. 1492–1500).
Yang, M., & Thung, G. (2016). Classification of trash for recyclability status. CS229 project report, 2016(1), 3.
Yuan, Z., & Liu, J. (2022). A hybrid deep learning model for trash classification based on deep trasnsfer learning. Journal of Electrical and Computer Engineering, 2022
Zhu, S., Chen, H., Wang, M., Guo, X., Lei, Y., & Jin, G. (2019). Plastic solid waste identification system based on near infrared spectroscopy in combination with support vector machine. Advanced Industrial and Engineering Polymer Research, 2(2), 77–81.
Author information
Authors and Affiliations
Contributions
All authors have contributed significantly to the conception and design. The initial manuscript was written by Shivendu Mishra, Ritika Yaduvanshi, and Prince Rajpoot from data collection, analysis, and model development. Shard Verma has contributed to the evaluation and analysis of the model result. Amit Kumar Pandey and Digvijay Pandey participated in reviewing and revising the manuscript’s content and gave their final approval for the published version. Each author has contributed sufficiently to the work to accept public responsibility for appropriate portions of the content. The final manuscript was read and approved by all authors.
Corresponding author
Ethics declarations
All authors have read, understood, and have complied as applicable with the statement on “Ethical responsibilities of Authors” as found in the Instructions for Authors.
Ethics approval
Not applicable
Consent to participate
All the authors declare their consent to participate in this research article.
Consent for publication
All the authors declare their consent for publication of the article on acceptance.
Competing interests
The authors declare no competing interests.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
About this article
Cite this article
Mishra, S., Yaduvanshi, R., Rajpoot, P. et al. An integrated deep-learning model for smart waste classification. Environ Monit Assess 196, 279 (2024). https://doi.org/10.1007/s10661-024-12410-x
Received:
Accepted:
Published:
DOI: https://doi.org/10.1007/s10661-024-12410-x