Multi-class Waste Classification Using Convolutional Neural Network
Article Sidebar
Main Article Content
Abstract
This study explores a method of waste classification using deep learning, specifically employing the Convolutional Neural Network (CNN). This research involves the creation of a unique dataset, a hybrid of publicly accessible data and a newly compiled collection of images across 13 waste classes: paper, glass, wood, metal, clothes, PCB e-waste, non-PCB e-waste, PET, HDPE, LDPE, PP, PVC, and PS. The development of the CNN model was approached in two ways: transfer learning and full learning. In the transfer learning approach, two pre-trained models, MobileNetV2 and DenseNet121, were utilized. While in the full learning approach, the architecture is constructed using the sequential method. The experimental results indicated that the DenseNet121 transfer learning model outperformed others, achieving an impressive accuracy of 95.2% and an average F-1 score of 0.95 on test data. This was closely followed by the MobileNetV2 transfer learning model, which attained an accuracy of 92% and an average F-1 score of 0.92. In comparison, the full learning model reached an accuracy of 65% and an average F-1 score of 0.65. Generally, transfer learning models yielded more optimal results than those full learning model. This efficiency can be attributed to the pre-existing knowledge in the transfer learning models, which eliminates the need to learn input patterns from the ground up. However, it's important to note that the dataset size of 4586 images across 13 classes may not be sufficient for developing a robust machine learning model from scratch.
Article Details
This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.
Published articles are under the copyright of the Applied Environmental Research effective when the article is accepted for publication thus granting Applied Environmental Research all rights for the work so that both parties may be protected from the consequences of unauthorized use. Partially or totally publication of an article elsewhere is possible only after the consent from the editors.
References
Indonesian Ministry of Environment and Forestry. National Waste Management Information System, 2023. [Online] Available from: https://sipsn.menlhk.go.id/sipsn/public/data/komposisi [Accessed 15 July 2023]
Lai, Y.Y., Lee, Y.M. Management strategy of plastic wastes in Taiwan. Sustainable Environment Research, 2022, 32(1), 1-10. https://doi.org/10.1186/s42834-022-00123-0.
Hidayat, Y.A., Kiranamahsa, S., Zamal, M.A. A study of plastic waste management effectiveness in Indonesia industries. AIMS Energy, 2019, 7(3), 350–370. https://doi.org/10.3934/ENERGY.2019.3.350.
Shamkhalichenar, H., Bueche, C.J., Choi, J.W. Printed Circuit Board (PCB) Technology for Electrochemical Sensors and Sensing Platforms. Biosensors, 2020, 10(11). https://doi.org/10.3390/bios10110159.
Baldé, C.P., Forti V., Gray, V., Kuehr, R., Stegmann, P. The Global E-waste Monitor – 2017. United Nations University (UNU), International Telecommunication Union (ITU) & International Solid Waste Association (ISWA), Bonn/Geneva/Vienna, 2017. [Online] Available from: https://collections.unu.edu/eserv/UNU:6341/Global-E-waste_Monitor_2017__electronic_single_pages_.pdf [Accessed 20 July 2023]
Santoso, S., Zagloel, T., Yuri M., Ardi, R., Suzianti, A. Estimating the Amount of Electronic Waste Generated in Indonesia: Population Balance Model. In IOP Conference Series: Earth and Environmental Science, 2019, 219(1). https://doi.org/10.1088/1755-1315/219/1/012006.
Cruvinel, V., Marques, C., Cardoso, V., Novaes, M., Araújo, W., Angulo-Tuesta, A., …, Da Silva, E. Health conditions and occupational risks in a novel group: Waste pickers in the largest open garbage dump in Latin America. BMC Public Health, 2019, 19(1), 1–15. https://doi.org/10.1186/s12889-019-6879-x.
Shah, J., Kamat, S. A Method for Waste Segregation using a Convolutional Neural Networks. In 2022 2nd International Conference on Advances in Electrical, Computing, Communication and Sustainable Technologies, ICAECT, 2022, 1-5. https://doi.org/10.1109/ICAECT54875.2022.9807969.
Liu, C., Sharan, L., Rosenholtz, R. Exploring features in a Bayesian. In 2010 IEEE computer society conference on computer vision and pattern recognition, 2010, 239–246.
Mohammed, M.A., Abdulhasan, M., Kumar, N., Abdulkareem, K., Mostafa, S., Maashi, M., …, Chopra, S. Automated waste-sorting and recycling classification using artificial neural network and features fusion: a digital-enabled circular economy vision for smart cities. Multimedia Tools and Applications, 2023, 82, 39617–39632. https://doi.org/10.1007/s11042-021-11537-0.
Hanbal, I. F., Ingosan, J., Oyam, N., Hu, Y. Classifying Wastes Using Random Forests, Gaussian Naïve Bayes, Support Vector Machine and Multilayer Perceptron. In IOP Conference Series: Materials Science and Engineering, 2020, 803(1). https://doi.org/10.1088/1757-899X/803/1/012017.
Wang, T., Chen, Y., Qiao, M., Snoussi, H. A fast and robust convolutional neural network-based defect detection model in product quality control. International Journal of Advanced Manufacturing Technology, 2018, 94(9–12), 3465–3471. https://doi.org/10.1007/s00170-017-0882-0.
Sami, K.N., Amin, Z.M.A., Hassan, R. Waste management using machine learning and deep learning algorithms. International Journal on Perceptive and Cognitive Computing, 2020, 6(2), 97-106.
Peng, H., Shen, N., Ying, H., Wang, Q. Factor analysis and policy simulation of domestic waste classification behavior based on a multiagent study—Taking Shanghai’s garbage classification as an example. Environmental Impact Assessment Review, 2021, 89. https://doi.org/10.1016/j.eiar.2021.106598.
Yang, M., Thung, G. Classification of Trash for Recyclability Status. Project Report CS229, 2016, 940–945. [Online] Available from: https://cs229.stanford.edu/proj2016/report/ThungYang-ClassificationOfTrashForRecyclabilityStatus-report.pdf [Accessed 21 July 2023]
Bircanoglu, C., Atay, M., Beser, F., Genc, O., Kizrak, M. RecycleNet: Intelligent Waste Sorting Using Deep Neural Networks. In Proceeding of 2018 IEEE (SMC) International Conference on Innovations in Intelligent Systems and Applications, INISTA, 2018, 1-7. https://doi.org/10.1109/INISTA.2018.8466276.
Yujie H., Qinyue G., Maguo S. Trash Classification Using Convolutional Neural Networks Project Category: Computer Vision. CS230: Deep Learning, 2020. [Online] Available from: http://cs230.stanford.edu/projects_spring_2020/reports/38847029.pdf [Accessed 21 July 2023]
Ziouzios, D., Tsiktsiris, D., Baras, N., Dasygenis, M. A Distributed Architecture for Smart Recycling Using Machine Learning. Future Internet, 2020, 12(9). https://doi.org/10.3390/FI12090141.
Mao, W.L., Chen, W., Wang, C., Lin, Y. Recycling waste classification using optimized convolutional neural network. Resources, Conservation and Recycling, 2021, 164, 105132. https://doi.org/10.1016/j.resconrec.2020.105132.
Zhang, Q., Yang, Q., Zhang, X., Bao, Q., Su, J., Liu, X. Waste image classification based on transfer learning and convolutional neural network. Waste Management, 2021, 135, 150–157. https://doi.org/10.1016/j.wasman.2021.08.038.
Patrizi, A., Gambosi, G., Zanzotto, F.M. Data augmentation using background replacement for automated sorting of littered waste. Journal of Imaging, 2021, 7(8). https://doi.org/10.3390/jimaging7080144.
Huynh, M. H., Pham-Hoai, P., Tran, A., Nguyen, T. Automated Waste Sorting Using Convolutional Neural Network. In Proceedings of the 7th NAFOSTED Conference on Information and Computer Science, NICS, 2020, 102–107. https://doi.org/10.1109/NICS51282.2020.9335897.
Majchrowska, S., Mikołajczyk, A., Ferlin, M., Klawikowska, Z., Plantykow, M., Kwasigroch, A., Majek, K. Deep learning-based waste detection in natural and urban environments. Waste Management, 2022, 138, 274–284. https://doi.org/10.1016/j.wasman.2021.12.001.
Karthikeyan, S., Sivakumar, M., Jeysiva, A., Maheshkumar, C. Application of Deep Learning for Solid Waste Trash Classification using Deep CNN. In Proceedings of ACM/CSI/IEEECS Research & Industry Symposium on IoT Cloud for Societal Applications, 2021, 126–131.
Bobulski, J., Kubanek, M. Deep Learning for Plastic Waste Classification System. Applied Computational Intelligence and Soft Computing, 2021, 1-7. https://doi.org/10.1155/2021/6626948.
Chazhoor, A.A.P., Edmond, H., Gao, B., Woo, W. Deep transfer learning benchmark for plastic waste classification. Intelligence & Robotics, 2022, 2, 1–19. https://doi.org/10.20517/ir.2021.15.
Baker, N., Szabo-Müller, P., Handmann, U. Transfer learning-based method for automated e-waste recycling in smart cities. EAI Endorsed Transactions on Smart Cities, 2018, 5(16), 169337. https://doi.org/10.4108/eai.16-4-2021.169337.
Sandler, M., Howard, A., Zhu, M., Zhmoginov, A., Chen, L.C. Mobilenetv2: Inverted residuals and linear bottlenecks. In Proceedings of the IEEE conference on computer vision and pattern recognition, 2018, 4510-4520.
Huang, G., Liu, Z., Van Der Maaten, L., Weinberger, K.Q. Densely Connected Convolutional Networks. In Proceedings of the IEEE conference on computer vision and pattern recognition, 2017, 2261–2269.