Machine Learning (ML) for Waste Classification: A Comparative Study of Different Algorithms

Thiruvilan Mohan

Authors

Thiruvilan Mohan United Kingdom Author

Keywords:

Machine Learning, Waste Classification, Decision Trees, Vector Machines, SVM algorithm

Abstract

Effective waste classification is crucial for efficient waste management and sustainable environmental practices. Machine learning (ML) algorithms have been increasingly used to classify waste due to their ability to learn from large datasets and improve classification accuracy. This study compares the performance of various ML algorithms in waste classification, including Support Vector Machines (SVM), Random Forest (RF), Decision Trees (DT), K-Nearest Neighbors (KNN), and Convolutional Neural Networks (CNN). The algorithms were trained and tested on a dataset of 1,500 waste samples, each with 12 features representing physical and chemical properties. The results show that the RF algorithm achieved the highest accuracy (93.3%) and F1-score (0.93), followed closely by the SVM algorithm (92.7% accuracy and 0.92 F1-score). The DT and KNN algorithms performed moderately well, while the CNN algorithm struggled to classify waste effectively. The study highlights the potential of ML algorithms in improving waste classification accuracy and suggests that RF and SVM algorithms are promising choices for waste management applications.

References

Yasin, E. T., and M. Koklu. "A Comparative Analysis of ML Algorithms for Waste Classification." International Journal of Environmental Science and Technology, vol. 18, no. 5, 2021, pp. 4213–4222.

Rajadevi, R., R. S. Latha, and K. Logeswaran. "Transfer Learning in Solid Waste Classification." Proceedings of the IEEE International Conference on Electronics and Communication, 2021, pp. 92–97.

Navaz, K., R. Selvakumar, and A. D. Aguru. "Comparative Study of SVM and Random Forest for Municipal Waste Sorting." Proceedings of the IEEE International Conference on Smart Cities, 2020, pp. 110–115.

Joshi, P., P. Bagul, and A. Deshpande. "AI-Based Waste Identification and Valuation." Proceedings of the IEEE Conference on Computer Vision and Machine Learning, 2020, pp. 201–206.

Dichpally, T., and U. Vallabhaneni. "YOLO vs. EfficientDet in Smart Waste Segregation." Multimedia Tools and Applications, vol. 80, no. 24, 2021, pp. 36419–36433.

Mishra, D., and R. Kumar. "ML Models for Citizen-Driven Waste Reporting." Proceedings of the IEEE Sustainable Computing Conference, 2020, pp. 141–146.

Santhiya, K., and D. Murugan. "Waste Detection via ML for Smart Healthcare." IEEE Xplore, 2020, pp. 51–56.

Tao, X., L. Liu, and S. Yuan. "Comparative Review of YOLO for Waste Sorting." IEEE Journal on Vision and Machine Learning, vol. 12, no. 3, 2020, pp. 223–229.

Kumar, M., and A. K. Luhach. "AI in Waste Management and Segregation." Trends in Machine Learning Research, vol. 4, no. 2, 2021, pp. 97–105.

Sharma, R., and A. Singh. "Reinforcement Learning in Urban Waste Routing." Springer Urban Technology Journal, vol. 15, no. 1, 2019, pp. 45–54.

Kumar, D., and A. Sinha. "Machine Learning Techniques for Solid Waste Management: A Survey." International Journal of Environmental Science and Technology, vol. 18, no. 2, 2021, pp. 341–352.

Fernandez, A., and P. Romero. "Smart Waste Sorting System Based on Deep Learning and IoT Integration." Procedia Computer Science, vol. 184, 2021, pp. 567–574.

Raghav, P., and R. M. Singh. "Comparison of Traditional and Machine Learning Approaches in Plastic Waste Classification." Journal of Cleaner Production, vol. 275, 2020, pp. 122875.

Banerjee, S., and A. Kar. "Image-Based Waste Segregation Using Transfer Learning for Urban Management." Sustainable Cities and Society, vol. 59, 2020, pp. 102233.

Thakur, P., and S. Gupta. "AI-Driven Classification of Biodegradable and Non-Biodegradable Waste." International Journal of Scientific Research in Computer Science, Engineering and Information Technology, vol. 7, no. 6, 2019, pp. 123–128.