Journal of Machine and Computing


A Deep Learning Approach to Smart Waste Classification for Sustainable Environments



Journal of Machine and Computing

Received On : 24 November 2024

Revised On : 02 March 2025

Accepted On : 25 May 2025

Published On : 05 July 2025

Volume 05, Issue 03

Pages : 1503-1517


Abstract


A key element of sustainable development is efficient trash classification, which aims to minimize environmental damage and expedite recycling procedures. In addition to being time-consuming, traditional human sorting methods are prone to mistakes, which makes waste management systems less effective. Automated garbage classification has attracted so much attention as AI, especially ML and DL, has grown. However, because they frequently rely on small-scale datasets and traditional architectures, many of the models that are now in use have issues with generalization, poor performance, and high error rates. This work presents a hybrid deep learning system that combines an autoencoder with a vision transformer (ViT) to address these issues. By efficiently capturing local and global data, our design improves classification robustness and accuracy across various waste types. Our model was trained and assessed using a sizable and varied dataset to enhance generalization to real-world scenarios. According to experimental data, the suggested model achieves a precision of 96.72%, a recall of 96.21%, an F1-score of 96.46%, and a balanced accuracy of 96.48%, outperforming some cutting-edge CNN-based architectures. Furthermore, sophisticated measures like Cohen's Kappa (95.90%) and Matthews Correlation Coefficient (MCC = 94.91%) confirm the dependability of our solution. Lastly, by successfully implementing the model in an inference pipeline, we show that it is ready for real-world deployment and that it has the potential to promote sustainable development goals through scalable, intelligent waste management.


Keywords


Sustainable Development, Waste Classification, Vision Transformer, Autoencoder, Hybrid Model.


  1. S. Kaza, L. C. Yao, P. Bhada-Tata, and F. Van Woerden, What a Waste 2.0: A Global Snapshot of Solid Waste Management to 2050. Washington, DC: World Bank, 2018. doi: 10.1596/978-1-4648-1329-0.
  2. M. S. Rad et al., “A Computer Vision System to Localize and Classify Wastes on the Streets,” Computer Vision Systems, pp. 195–204, 2017, doi: 10.1007/978-3-319-68345-4_18.
  3. G. Qi, Y. Jia, and H. Zou, “Is institutional pressure the mother of green innovation? Examining the moderating effect of absorptive capacity,” Journal of Cleaner Production, vol. 278, p. 123957, Jan. 2021, doi: 10.1016/j.jclepro.2020.123957.
  4. A. Krizhevsky, I. Sutskever, and G. E. Hinton, “ImageNet classification with deep convolutional neural networks,” Communications of the ACM, vol. 60, no. 6, pp. 84–90, May 2017, doi: 10.1145/3065386.
  5. K. Han et al., “A Survey on Vision Transformer,” IEEE Transactions on Pattern Analysis and Machine Intelligence, vol. 45, no. 1, pp. 87–110, Jan. 2023, doi: 10.1109/tpami.2022.3152247.
  6. S. J. Pan and Q. Yang, “A Survey on Transfer Learning,” IEEE Transactions on Knowledge and Data Engineering, vol. 22, no. 10, pp. 1345–1359, Oct. 2010, doi: 10.1109/tkde.2009.191.
  7. A. Dosovitskiy., “An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale,” Jun. 03, 2021, arXiv: arXiv:2010.11929. doi: 10.48550/arXiv.2010.11929.
  8. K. He, X. Chen, S. Xie, Y. Li, P. Dollar, and R. Girshick, “Masked Autoencoders Are Scalable Vision Learners,” 2022 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Jun. 2022, doi: 10.1109/cvpr52688.2022.01553.
  9. W. Qiu, C. Xie, and J. Huang, “An improved EfficientNetV2 for garbage classification,” Mar. 27, 2025, arXiv: arXiv:2503.21208. doi: 10.48550/arXiv.2503.21208.
  10. Md. Nahiduzzaman et al., “An automated waste classification system using deep learning techniques: Toward efficient waste recycling and environmental sustainability,” Knowledge-Based Systems, vol. 310, p. 113028, Feb. 2025, doi: 10.1016/j.knosys.2025.113028.
  11. S. Kunwar, B. R. Owabumoye, and A. S. Alade, “Plastic Waste Classification Using Deep Learning: Insights from the WaDaBa Dataset,” Dec. 28, 2024, arXiv: arXiv:2412.20232. doi: 10.48550/arXiv.2412.20232.
  12. M. I. Ahmed et al., “Deep Learning Approach to Recyclable Products Classification: Towards Sustainable Waste Management,” Sustainability, vol. 15, no. 14, 2023, doi: 10.3390/su151411138.
  13. S. Majchrowska et al., “Deep learning-based waste detection in natural and urban environments,” Waste Manag, vol. 138, pp. 274–284, Feb. 2022, doi: 10.1016/j.wasman.2021.12.001.
  14. Y. Narayan, “DeepWaste: Applying Deep Learning to Waste Classification for a Sustainable Planet,” Jan. 15, 2021, arXiv: arXiv:2101.05960. doi: 10.48550/arXiv.2101.05960.
  15. J. Bobulski and M. Kubanek, “Deep Learning for Plastic Waste Classification System,” Applied Computational Intelligence and Soft Computing, vol. 2021, pp. 1–7, May 2021, doi: 10.1155/2021/6626948.
  16. G. White, C. Cabrera, A. Palade, F. Li, and S. Clarke, “WasteNet: Waste Classification at the Edge for Smart Bins,” Jun. 10, 2020, arXiv: arXiv:2006.05873. doi: 10.48550/arXiv.2006.05873.
  17. D. Gyawali, A. Regmi, A. Shakya, A. Gautam, and S. Shrestha, “Comparative Analysis of Multiple Deep CNN Models for Waste Classification,” Aug. 14, 2020, arXiv: arXiv:2004.02168. doi: 10.48550/arXiv.2004.02168.
  18. O. Adedeji and Z. Wang, “Intelligent Waste Classification System Using Deep Learning Convolutional Neural Network,” Procedia Manufacturing, vol. 35, pp. 607–612, 2019, doi: 10.1016/j.promfg.2019.05.086.
  19. N. N. Islam Prova, “Garbage Intelligence: Utilizing Vision Transformer for Smart Waste Sorting,” 2024 Second International Conference on Intelligent Cyber Physical Systems and Internet of Things (ICoICI), pp. 1213–1219, Aug. 2024, doi: 10.1109/icoici62503.2024.10696177.
  20. N. N. Islam Prova, “Enhancing Fish Disease Classification in Bangladeshi Aquaculture through Transfer Learning, and LIME Interpretability Techniques,” 2024 4th International Conference on Sustainable Expert Systems (ICSES), pp. 1157–1163, Oct. 2024, doi: 10.1109/icses63445.2024.10763350.
  21. T. Sattler, Q. Zhou, M. Pollefeys, and L. Leal-Taixe, “Understanding the Limitations of CNN-Based Absolute Camera Pose Regression,” 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), pp. 3297–3307, Jun. 2019, doi: 10.1109/cvpr.2019.00342.
  22. Wang, Y., Zhao, W. J., Xu, J., & Hong, R. (2020). Recyclable waste identification using cnn image recognition and gaussian clustering. arXiv preprint arXiv:2011.01353.
  23. Nakib, A. A., Talukder, M. N., Majumder, C., Biswas, S., & Hassan, J. (2021). Deep learning-based waste classification system for efficient waste management (Doctoral dissertation, Brac University).
  24. G. I. Sayed, M. Abd Elfattah, A. Darwish, and A. E. Hassanien, “Intelligent and sustainable waste classification model based on multi-objective beluga whale optimization and deep learning,” Environmental Science and Pollution Research, vol. 31, no. 21, pp. 31492–31510, Apr. 2024, doi: 10.1007/s11356-024-33233-w.

CRediT Author Statement


The authors confirm contribution to the paper as follows:

Conceptualization: Vishnu Tej Y, Ashwitha A, Lakshmi H N, Vuppala Balaji, Suryanarayana G and Sirish Kumar M; Methodology: Vuppala Balaji, Suryanarayana G and Sirish Kumar M; Visualization: Vishnu Tej Y, Ashwitha A and Lakshmi H N; Investigation: Vuppala Balaji, Suryanarayana G and Sirish Kumar M; Supervision: Vuppala Balaji and Suryanarayana G; Validation: Vishnu Tej Y, Ashwitha A and Lakshmi H N; Writing- Reviewing and Editing: Vishnu Tej Y, Ashwitha A, Lakshmi H N, Vuppala Balaji, Suryanarayana G and Sirish Kumar M; All authors reviewed the results and approved the final version of the manuscript.


Acknowledgements


Authors thank Reviewers for taking the time and effort necessary to review the manuscript.


Funding


No funding was received to assist with the preparation of this manuscript.


Ethics declarations


Conflict of interest

The authors have no conflicts of interest to declare that are relevant to the content of this article.


Availability of data and materials


Data sharing is not applicable to this article as no new data were created or analysed in this study.


Author information


Contributions

All authors have equal contribution in the paper and all authors have read and agreed to the published version of the manuscript.


Corresponding author


Rights and permissions


Open Access This article is licensed under a Creative Commons Attribution NoDerivs is a more restrictive license. It allows you to redistribute the material commercially or non-commercially but the user cannot make any changes whatsoever to the original, i.e. no derivatives of the original work. To view a copy of this license, visit https://creativecommons.org/licenses/by-nc-nd/4.0/


Cite this article


Vishnu Tej Y, Ashwitha A, Lakshmi H N, Vuppala Balaji, Suryanarayana G and Sirish Kumar M, “A Deep Learning Approach to Smart Waste Classification for Sustainable Environments”, Journal of Machine and Computing, vol.5, no.3, pp. 1503-1517, July 2025, doi: 10.53759/7669/jmc202505119.


Copyright


© 2025 Vishnu Tej Y, Ashwitha A, Lakshmi H N, Vuppala Balaji, Suryanarayana G and Sirish Kumar M. This is an open access article distributed under the terms of the Creative Commons Attribution License, which permits unrestricted use, distribution, and reproduction in any medium, provided the original author and source are credited.