Performance of Naïve Bayes and Support Vector Machine for solid waste classification in automated sorting systems
##plugins.themes.academic_pro.article.main##
Abstract
This study evaluates the performance of two traditional machine learning models - Naïve Bayes and Support Vector Machines (SVM) - for classifying solid waste materials in an automated sorting system. A dataset of 284 JPEG images, categorized into five classes (cardboard, glass, metal, paper, and plastic), was utilized. Preprocessing involved resizing images to 512 × 384 pixels, normalizing pixel values, and extracting features using Histograms of Oriented Gradients (HOG) and Color Histograms. Based on the results, Naïve Bayes exhibited computational efficiency, achieving an accuracy of 98.90% and an F1-score of 0.908. However, it struggled with overlapping features, particularly between glass and metal, leading to misclassifications.
In contrast, SVM outperformed Naïve Bayes, achieving an accuracy of 99.80% and an F1-score of 0.965 by effectively handling complex and overlapping features through optimal decision boundaries. The findings highlight SVM’s superior performance for complex datasets, whereas Naïve Bayes remains a viable option for more straightforward classification tasks. This study underscores the potential of traditional machine learning in waste classification. Still, it suggests that integrating deep learning models may improve accuracy, scalability, and adaptability in real-world waste sorting systems.
Keywords
automated sorting system, Naïve Bayes, pollution, solid waste management, Support Vector Machine

This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.
References
- Abood, I. N., & Al-Talib, G. A. A. (2023). Waste classification using artificial intelligence techniques: A literature review. Technium, 5, 49-59. https://doi.org/10.47577/technium.v5i.8345 DOI: https://doi.org/10.47577/technium.v5i.8345
- Ahmed, R., Malik, F., & Khan, S. (2022). Handwritten digit recognition with hybrid naïve Bayes. Pattern Recognition and Applications, 32(1), 145-159. https://doi.org/10.1016/j.pra.2022.0321
- Bian, S., Ouyang, X., Fan, Z., & Koutris, P. (2024). Naïve Bayes classifiers over missing data: Decision and poisoning. In Proceedings of the 41st International Conference on Machine Learning (Vol. 235, pp. 3913-3934). PMLR. https://proceedings.mlr.press/v235/bian24b.html
- Chollet, F. (2017). Deep learning with Python. Manning Publications.
- Commission on Audit - Philippines. (2023). Performance audit report on solid waste management program (PAO-2023-01). https://www.coa.gov.ph/wpfd_file/solid-waste-management-program-pao-2023-01/
- Dalal, N., & Triggs, B. (2005). Histograms of oriented gradients for human detection. In Proceedings of the IEEE computer society conference on computer vision and pattern recognition (pp. 886-893). IEEE. DOI: https://doi.org/10.1109/CVPR.2005.177
- Department of Environment and Natural Resources. (2019). Impact of improper waste disposal on the environment. Republic of the Philippines.
- Domingos, P., & Pazzani, M. (1997). On the optimality of the simple Bayesian classifier under zero‐one loss. Machine Learning, 29(2/3), 103-130. https://doi.org/10.1023/A:1007413511361 DOI: https://doi.org/10.1023/A:1007413511361
- Fatimah, Y. A., Govindan, K., Murniningsih, R., & Setiawan, A. (2020). Predictive analytics and Internet of things in waste management: A systematic review. Sustainable Cities and Society, 62, Article 102356. https://doi.org/10.1016/j.scs.2020.102356 DOI: https://doi.org/10.1016/j.scs.2020.102356
- García-Solla, D. (2022). Advanced waste classification with machine learning [Blog post]. Towards Data Science. https://towardsdatascience.com/advanced-waste-classification-with-machine-learning-6445bff1304f
- Gonzalez, R. C., & Woods, R. E. (2018). Digital image processing (4th ed.). Pearson.
- Goodfellow, I., Bengio, Y., & Courville, A. (2016). Deep learning. MIT Press.
- Gupta, P., Sharma, A., & Bhardwaj, L. K. (2023). Solid Waste Management (SWM) and its effect on environment & human health [Preprint]. https://doi.org/10.20944/preprints202309.0384.v1 DOI: https://doi.org/10.20944/preprints202309.0384.v1
- Hanbal, I. F., Ingosan, J. S., Oyam, N. A. A., & Hu, Y. (2020). Classifying wastes using random forests, Gaussian naïve Bayes, support vector machine, and multilayer perceptron. IOP Conference Series: Materials Science and Engineering, 803(1), Article 012017. DOI: https://doi.org/10.1088/1757-899X/803/1/012017
- Hossain, E., Karim, M. M., Hasan, M. K., & Rahman, A. (2019). Traditional machine learning techniques for yoga pose classification. International Journal of Computer Vision and Signal Processing, 8(1), 12-21.
- Kanade, V. (2021). Support vector machine formulation and derivation. https://towardsdatascience.com
- Kesumawati, A., & Utari, D. T. (2018). Predicting patterns of student graduation rates using naïve Bayes classifier and support vector machine. AIP Conference Proceedings, 2021(1), Article 060005. https://doi.org/10.1063/1.5062769 DOI: https://doi.org/10.1063/1.5062769
- Li, N., & Chen, Y. (2023). Municipal solid waste classification and real-time detection using deep learning methods. Urban Climate, 49, Article 101462. DOI: https://doi.org/10.1016/j.uclim.2023.101462
- Liu, G., Luo, Y., & Sheng, J. (2022). Research on application of Naïve Bayes algorithm based on attribute correlation to unmanned driving ethical dilemma. Mathematical Problems in Engineering, 1, Article 1104718. https://doi.org/10.1155/2022/1104718 DOI: https://doi.org/10.1155/2022/4163419
- Mohan, A., & Varghese, R. (2021). A comparative analysis of image classification techniques using machine learning. Journal of Computational and Applied Mathematics, 10(4), 45-58.
- Murphy, K. P. (2012). Machine learning: A probabilistic perspective. MIT Press.
- Namoun, A., Tufail, A., Khan, M. Y., Alrehaili, A., Syed, T. A., & BenRhouma, O. (2022). Solid waste generation and disposal using machine learning approaches: A survey of solutions and challenges. Sustainability, 14(20), Article 13578. DOI: https://doi.org/10.3390/su142013578
- Park, S., & Choi, K. (2019). Environmental impacts of poor waste segregation: A global overview. International Journal of Environmental Policy, 12(3), 233-245.
- Patel, R., & Singh, D. (2021). Satellite image classification using the naïve Bayes algorithm. Geospatial Analysis and Data Science, 12(4), 200-215. https://doi.org/10.1007/s00394-021-00417-3
- Poudel, S., & Poudyal, P. (2022). Classification of waste materials using CNN based on transfer learning. In Forum for Information Retrieval Evaluation (FIRE ’22) (pp. 1-5). ACM. https://doi.org/10.1145/3574318.3574345 DOI: https://doi.org/10.1145/3574318.3574345
- Powers, D. M. W. (2020). Evaluation: From precision, recall, and F‐measure to ROC, informedness, markedness, and correlation. https://arxiv.org/abs/2010.16061
- Purnomo, A., Barata, M. A., Soeleman, M. A., & Alzami, F. (2020). Adding feature selection on Naïve Bayes to increase accuracy in classifying heart attack disease. Journal of Physics: Conference Series, 1511(1), Article 012001. https://doi.org/10.1088/1742-6596/1511/1/012001 DOI: https://doi.org/10.1088/1742-6596/1511/1/012001
- Russakovsky, O., Deng, J., Su, H., Krause, J., Satheesh, S., Ma, S., Huang, Z., Karpathy, A., Khosla, A., Bernstein, M., Berg, A. C., & Fei-Fei, L. (2015). ImageNet large scale visual recognition challenge. International Journal of Computer Vision, 115(3), 211-252. https://doi.org/10.1007/s11263-015-0816-y DOI: https://doi.org/10.1007/s11263-015-0816-y
- Russell, S., & Norvig, P. (2020). Artificial intelligence: A modern approach (4th ed.). Pearson.
- Schölkopf, B., & Smola, A. J. (2002). Learning with kernels: Support vector machines, regularization, optimization, and beyond. MIT Press. DOI: https://doi.org/10.7551/mitpress/4175.001.0001
- Solikhatun, A., & Sugiharti, E. (2020). Application of the Naïve Bayes classifier algorithm using N‐Gram and information gain to improve the accuracy of restaurant review sentiment analysis. Journal of Advanced Information Systems Technology, 2(2), 11-20.
- Wilts, H. A., von Gries, N., & Bahn-Walkowiak, B. (2016). From waste management to resource efficiency - The need for policy mixes. Sustainability, 8(7), Article 622. https://doi.org/10.3390/su8070622mdpi.com+1researchgate.net+1 DOI: https://doi.org/10.3390/su8070622
- World Bank. (2018). What a waste 2.0: A global snapshot of solid waste management to 2050. World Bank Publications.
- Wu, X., Kumar, V., Quinlan, J. R., Ghosh, J., Yang, Q., Motoda, H., McLachlan, G. J., Ng, A., Liu, B., Yu, P. S., Zhou, Z. H., Michael, S., David, J. H., & Steinberg, D. (2008). Top 10 algorithms in data mining. Knowledge and Information Systems, 14(1), 1-37. https://doi.org/10.1007/s10115-007-0114-2 DOI: https://doi.org/10.1007/s10115-007-0114-2
- Yulita, I. N., Ardiansyah, F., Prabuwono, A. S., Ramdhani, M. R., Wicaksono, M. A., Trisanto, A., & Sholahuddin, A. (2023). Recyclable waste classification using SqueezeNet and XGBoost. International Journal of Advanced Computer Science and Applications, 14(10), 345-352. https://doi.org/10.14569/IJACSA.2023.0141037 DOI: https://doi.org/10.14569/IJACSA.2023.0141037
- Zhang, M., Liu, Y., & Chen, J. (2020). Manual vs. automated waste sorting: Comparative impacts on recycling rates. Waste Management Journal, 102, 182-190.
- Zhou, Z. H., & Feng, J. (2017). Deep forest: Towards an alternative to deep neural networks. Proceedings of the 26th International Joint Conference on Artificial Intelligence (IJCAI-17) (pp.3553-3559). https://doi.org/10.24963/ijcai.2017/497 DOI: https://doi.org/10.24963/ijcai.2017/497