Leveraging Deep Learning and Convolutional Neural Network for Digital Waste Image Classification

Universitas Muhammadiyah Magelang, Magelang, Indonesia

^* Corresponding author: pristi.sukmasetya@ummgl.ac.id

Abstract

Waste management is a significant environmental challenge, with 14 million tons of waste uncollected in 2023, accumulating at landfill sites. The mixing of household and industrial waste complicates adequate segregation. Baseline sorting techniques are costly in labor, time, and resources. This paper looks at the application of CNNs for the automated sorting of wastes to enhance speed, capacity, and precision. CNNs can be used to classify digital images of waste to minimize the manual work in waste management, especially on a large scale. The research evaluates two CNN architectures—ResNet101 and a custom-designed CNN model—aiming to enhance waste classification. The dataset includes images of 12 different waste categories: battery, biological waste, cardboard, clothes, green glass, metal, paper, plastic, shoes, trash, and white glass. The performance of models was measured by metrics such as accuracy, precision, recall, and F1-score during the model training and assessment periods. The fine-tuned ResNet101 model offered a validation accuracy of 97.74% and classification accuracy of 74.21%, though with lesser accuracy when compared to the custom model, which had 90.31% validation accuracy and classification accuracy of 52.5%. Such results have shown the ability of deep learning models, especially CNNs, to enhance waste classification accuracy. The study, therefore, provides perceptions of improved efficiency in waste management. Thus, it is recommended that future work improve these models, include more data, and explore more practical applications to encourage appropriate waste management and demarcate the impact on the environment.