Impact of Dataset Background on Deep Learning-Based Waste Classification
DOI:
https://doi.org/10.29207/resti.v10i3.6965Keywords:
convolutional neural network, deep learning, densenet169, image classification, mobilenetv2Abstract
Accurate waste classification plays a vital role in supporting effective waste management and promoting environmental sustainability, especially amid the continuing increase in global waste generation. This study investigates how the presence and removal of image backgrounds influence the performance of deep learning models in automated waste classification. Two Convolutional Neural Network architectures, namely MobileNetV2 and DenseNet169, were evaluated using a dataset comprising 5,054 images across six waste categories: cardboard, glass, metal, paper, plastic, and trash. Each architecture was trained and tested on two dataset variants: original images with backgrounds and images with the backgrounds removed. Model performance was assessed using accuracy, precision, recall, F1-score, and ROC AUC. The results indicate that DenseNet169 consistently outperformed MobileNetV2 across all evaluation metrics. The highest accuracy, reaching 88.33%, was achieved by DenseNet169 when trained on images retaining their original backgrounds. This suggests that background information may provide meaningful contextual features that enhance classification performance. Conversely, removing backgrounds can limit the visual information available to the model and potentially reduce predictive effectiveness. These findings emphasize the importance of carefully considering background characteristics during dataset preparation and model training. Moreover, the study demonstrates that selecting an appropriate model architecture in relation to dataset properties is essential for optimizing classification outcomes. Overall, this research offers practical insights for improving dataset design and model selection in future automated waste classification systems, while contributing to the advancement of scalable and intelligent deep learning-based waste management solutions.
Downloads
References
Z. Kang, J. Yang, G. Li, and Z. Zhang, “An Automatic Garbage Classification System Based on Deep Learning,” IEEE Access, vol. 8, pp. 140019–140029, 2020, doi: 10.1109/ACCESS.2020.3010496.
Sunardi, A. Yudhana, and M. Fahmi, “SVM-CNN Hybrid Classification for Waste Image Using Morphology and HSV Color Model Image Processing,” Trait. du Signal, vol. 40, no. 4, pp. 1763–1769, 2023, doi: 10.18280/ts.400446.
F. Hamami, “Garbage classification using convolutional neural network based transfer learning,” 2023, p. 20012. doi: 10.1063/5.0128946.
F. Wu and H. Lin, “Effect of transfer learning on the performance of VGGNet-16 and ResNet-50 for the classification of organic and residual waste,” Front. Environ. Sci., vol. 10, 2022, doi: 10.3389/fenvs.2022.1043843.
K. Lin et al., “Applying a deep residual network coupling with transfer learning for recyclable waste sorting,” Environ. Sci. Pollut. Res., vol. 29, no. 60, pp. 91081–91095, 2022, doi: 10.1007/s11356-022-22167-w.
M. I. B. Ahmed et al., “Deep Learning Approach to Recyclable Products Classification: Towards Sustainable Waste Management,” Sustain., vol. 15, no. 14, 2023, doi: 10.3390/su151411138.
A. Sevinç and F. Ozyurt, “Classification of recyclable waste using deep learning architectures,” FIRAT Univ. J. Exp. Comput. Eng., vol. 1, no. 3, pp. 122–128, 2022, doi: 10.5505/fujece.2022.83997.
W. L. Mao, W. C. Chen, C. T. Wang, and Y. H. Lin, “Recycling waste classification using optimized convolutional neural network,” Resour. Conserv. Recycl., vol. 164, 2021, doi: 10.1016/j.resconrec.2020.105132.
A. H. Vo, L. H. Son, M. T. Vo, and T. Le, “A Novel Framework for Trash Classification Using Deep Transfer Learning,” IEEE Access, vol. 7, pp. 178631–178639, 2019, doi: 10.1109/ACCESS.2019.2959033.
O. Campesato, Artificial Intelligence, Machine Learning, and Deep Learning. Mercury Learning and Information LLC, 2020. [Online]. Available: https://books.google.co.id/books?hl=id&lr=&id=pqnNDwAAQBAJ&oi=fnd&pg=PT16&dq=Artificial+Intelligence,+Machine+Learning+and+Deep+Learning.+Dulles:+Mercury+Learning+and+Information+LLC.&ots=I2SxMON0UU&sig=Js5avan1NBgUuzkgWiBDlMrIHl8&redir_esc=y#v=onepage&q=
A. Fadlil, R. Umar, Sunardi, and A. S. Nugroho, “Comparison of Machine Learning Approach for Waste Bottle Classification,” Emerg. Sci. J., vol. 6, no. 5, pp. 1075–1085, 2022, doi: 10.28991/ESJ-2022-06-05-011.
Sunardi, A. Yudhana, and M. Fahmi, “Improving Waste Classification Using Convolutional Neural Networks: An Application of Machine Learning for Effective Environmental Management,” Rev. d’Intelligence Artif., vol. 37, no. 4, pp. 845–855, 2023, doi: 10.18280/ria.370404.
C. Shi, C. Tan, T. Wang, and L. Wang, “A waste classification method based on a multilayer hybrid convolution neural network,” Appl. Sci., vol. 11, no. 18, 2021, doi: 10.3390/app11188572.
L. Chen, S. Li, Q. Bai, J. Yang, S. Jiang, and Y. Miao, “Review of image classification algorithms based on convolutional neural networks,” 2021, MDPI. doi: 10.3390/rs13224712.
F. Nashrullah, S. A. Wibowo, and D. G. Budiman, “Investigasi Parameter Epoch Pada Arsitektur ResNet-50 Untuk Klasifikasi Pornografi,” J. Comput. Electron. Telecommun., 2020, doi: 10.52435/complete.v1i1.51.
Y. Sun, B. Xue, M. Zhang, G. G. Yen, and J. Lv, “Automatically Designing CNN Architectures Using the Genetic Algorithm for Image Classification,” IEEE Trans. Cybern., vol. 50, no. 9, pp. 3840–3854, 2020, doi: 10.1109/TCYB.2020.2983860.
A. Mabrouk, R. P. D. Redondo, A. Dahou, M. A. Elaziz, and M. Kayed, “Pneumonia Detection on Chest X-ray Images Using Ensemble of Deep Convolutional Neural Networks,” Appl. Sci., vol. 12, no. 13, 2022, doi: 10.3390/app12136448.
L. Yong, L. Ma, D. Sun, and L. Du, “Application of MobileNetV2 to waste classification,” PLoS One, vol. 18, no. 3 March, 2023, doi: 10.1371/journal.pone.0282336.
D. Yu, Q. Xu, H. Guo, C. Zhao, Y. Lin, and D. Li, “An Efficient and Lightweight Convolutional Neural Network for Remote Sensing Image Scene Classification,” Sensors, 2020, doi: 10.3390/s20071999.
G. S. Susanth, L. M. J. Livingston, and L. G. X. A. Livingston, “Garbage Waste Segregation Using Deep Learning Techniques,” IOP Conf. Ser. Mater. Sci. Eng., vol. 1012, no. 1, p. 12040, 2021, doi: 10.1088/1757-899x/1012/1/012040.
M. Zeng, X. Lu, W. Xu, T. Zhou, and Y. Liu, “PublicGarbageNet : A Deep Learning Framework for Public Garbage Classification,” in Proceedings of the 39th Chinese Control Conference, 2020, doi: 10.23919/CCC50068.2020.9189561.
J. Bobulski and M. Kubanek, “Deep Learning for Plastic Waste Classification System,” Appl. Comput. Intell. Soft Comput., vol. 2021, 2021, doi: 10.1155/2021/6626948.
Q. Zhang, Q. Yang, X. Zhang, Q. Bao, J. Su, and X. Liu, “Waste image classification based on transfer learning and convolutional neural network,” Waste Manag., vol. 135, pp. 150–157, 2021, doi: 10.1016/j.wasman.2021.08.038.
Risfendra, G. F. Ananda, and H. Setyawan, “Deep Learning-Based Waste Classification with Transfer Learning Using EfficientNet-B0 Model,” J. RESTI, vol. 8, no. 4, pp. 535–541, 2024, doi: 10.29207/resti.v8i4.5875.
X. Qin, Z. Zhang, C. Huang, M. Dehghan, O. R. Zaiane, and M. Jagersand, “U2-Net: Going deeper with nested U-structure for salient object detection,” Pattern Recognit., vol. 106, p. 107404, 2020, doi: 10.1016/j.patcog.2020.107404.
T. Jahan and A. Rajee, “Image Background Excluder: A Deep Learning Approach to Object Isolation,” no. February, 2025, doi: 10.13140/RG.2.2.18436.64644.
J. Filter, “Split-folders 0.5.1,” 3 February 2022. [Interactiv]. Available: https://pypi.org/project/split-folders/.
J. Brown, Z. Gharineiat, and N. Raj, “CNN Based Image Classification of Malicious UAVs,” Appl. Sci., vol. 13, no. 1, 2023, doi: 10.3390/app13010240.
H. Zheng and Y. Gu, “Encnn-upmws: Waste classification by a CNN ensemble using the UPM weighting strategy,” Electron., vol. 10, no. 4, pp. 1–21, 2021, doi: 10.3390/electronics10040427.
Y. Pang, W. Yu, Y. Zhang, C. Xuan, and P. Wu, “Sheep face recognition and classification based on an improved MobilenetV2 neural network,” Int. J. Adv. Robot. Syst., vol. 20, no. 1, 2023, doi: 10.1177/17298806231152969.
E. Amoako, Y. Sheng, M. Khalid, and M. Bock, “Application of SSD-MobileNetV2 for automated defect detection in masonry bridges using AI and IoT,” Adv. Bridg. Eng., vol. 6, no. 1, 2025, doi: 10.1186/s43251-025-00179-z.
R. P. Narmadha, N. Sengottaiyan, and R. J. Kavitha, “Deep transfer learning based rice plant disease detection model,” Intell. Autom. Soft Comput., vol. 31, no. 2, pp. 1257–1271, 2022, doi: 10.32604/iasc.2022.020679.
T. Chauhan, H. Palivela, and S. Tiwari, “Optimization and fine-tuning of DenseNet model for classification of COVID-19 cases in medical imaging,” Int. J. Inf. Manag. Data Insights, vol. 1, no. 2, 2021, doi: 10.1016/j.jjimei.2021.100020.
B. A. S. Al-rimy, F. Saeed, M. Al-Sarem, A. M. Albarrak, and S. N. Qasem, “An Adaptive Early Stopping Technique for DenseNet169-Based Knee Osteoarthritis Detection Model,” Diagnostics, vol. 13, no. 11, 2023, doi: 10.3390/diagnostics13111903.
A. Yadav and D. S. Yadav, “Enhancing Oral Squamous Cell Carcinoma Detection: A Transfer Learning Perspective on Histopathological Analysis Using ResNet-18, AlexNet, DenseNet-169, and DenseNet-201 with Cyclic Learning Rate,” Int. J. Intell. Syst. Appl. Eng., vol. 12, no. 17s, pp. 689–699, 2024, [Online]. Available: https://ijisae.org/index.php/IJISAE/article/view/4937
M. S. Amrullah and S. F. Pane, Analisis Sentimen Masyarakat Terhadap Kebijakan Polisi Tilang Manual Di Indonesia. Penerbit Buku Pedia, 2023. Available: https://books.google.co.id/books?id=_Uq5EAAAQBAJ&printsec=copyright&hl=id#v=onepage&q&f=false.
F. Rahmad, Y. Suryanto, and K. Ramli, “Performance Comparison of Anti-Spam Technology Using Confusion Matrix Classification,” in IOP Conference Series: Materials Science and Engineering, IOP Publishing Ltd, 2020. doi: 10.1088/1757-899X/879/1/012076.
A. U. Zailani and N. L. Hanun, “Penerapan Algoritma Klasifikasi Random Forest Untuk Penentuan Kelayakan Pemberian Kredit Di Koperasi Mitra Sejahtera,” Infotech J. Technol. Inf., vol. 6, no. 1, pp. 7–14, 2020, doi: 10.37365/jti.v6i1.61.
F. S. Nahm, “Receiver operating characteristic curve: overview and practical use for clinicians,” Korean J. Anesthesiol., vol. 75, no. 1, pp. 25–36, 2022, doi: 10.4097/kja.21209.
K. Xiao, L. Engstrom, A. Ilyas, and A. Madry, “Noise or Signal: The Role of Image Backgrounds in Object Recognition,” 2020, [Online]. Available: http://arxiv.org/abs/2006.09994
R. Aljundi, K. Kelchtermans, and T. Tuytelaars, “Task-free continual learning,” in Proceedings of the IEEE Computer Society Conference on Computer Vision and Pattern Recognition, IEEE Computer Society, 2019, pp. 11246–11255. doi: 10.1109/CVPR.2019.01151.
E. H. E. Yasin and C. Kornel, “Evaluating Satellite Image Classification: Exploring Methods and Techniques,” 2024. doi: 10.5772/intechopen.1003196
S. T. Erukude, A. Joshi, and L. Shamir, “Identifying Bias in Deep Neural Networks Using Image Transforms,” Computers, vol. 13, no. 12, 2024, doi: 10.3390/computers13120341.
E. M. Aminoff, S. Baror, E. W. Roginek, and D. D. Leeds, “Contextual associations represented both in neural networks and human behavior,” Sci. Rep., vol. 12, no. 1, pp. 1–12, 2022, doi: 10.1038/s41598-022-09451-y.
P. R. A. S. Bassi, S. S. J. Dertkigil, and A. Cavalli, “Improving deep neural network generalization and robustness to background bias via layer-wise relevance propagation optimization,” Nat. Commun., vol. 15, no. 1, pp. 1–16, 2024, doi: 10.1038/s41467-023-44371-z.
S. Uhlig, I. Alkhasli, F. Schubert, C. Tschöpe, and M. Wolff, “A review of synthetic and augmented training data for machine learning in ultrasonic non-destructive evaluation,” Ultrasonics, vol. 134, no. June 2022, 2023, doi: 10.1016/j.ultras.2023.107041.
Downloads
Published
Issue
Section
License
Copyright in each article belongs to the author
- The author acknowledges that the RESTI Journal (System Engineering and Information Technology) is the first publisher to publish with a license Creative Commons Attribution 4.0 International License.
- Authors can enter writing separately, arrange the non-exclusive distribution of manuscripts that have been published in this journal into other versions (eg sent to the author's institutional repository, publication in a book, etc.), by acknowledging that the manuscript has been published for the first time in the RESTI (Rekayasa Sistem dan Teknologi Informasi) journal ;
