Implementation of Convolutional Neural Network for Waste Classification Using the TrashNet Dataset

Authors

  • Dika Universitas Pembangunan Panca Budi
  • Zulham Sitorus Universitas Pembangunan Panca Budi

Keywords:

Convolutional Neural Network, Deep Learning, Image Classification, TrashNet, Artificial Intelligence

Abstract

Effective waste management requires a fast and accurate waste sorting process. This study aims to develop a Convolutional Neural Network (CNN)-based model for automatic waste classification using the TrashNet dataset. The dataset consists of six categories: cardboard, glass, metal, paper, plastic, and trash. The research methodology includes data preprocessing, data augmentation, CNN model training, and performance evaluation using accuracy, precision, recall, and F1-score metrics. The experimental results demonstrate that the proposed CNN model achieves a high level of classification accuracy on the test dataset, indicating its potential for supporting automated waste sorting systems.

References

. Goodfellow, I., Bengio, Y., & Courville, A. (2016). Deep learning. MIT Press.

. He, K., Zhang, X., Ren, S., & Sun, J. (2016). Deep residual learning for image recognition. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 770–778. https://doi.org/10.1109/CVPR.2016.90

. Kaza, S., Yao, L., Bhada-Tata, P., & Van Woerden, F. (2018). What a waste 2.0: A global snapshot of solid waste management to 2050. World Bank. https://doi.org/10.1596/978-1-4648-1329-0

. Kingma, D. P., & Ba, J. (2015). Adam: A method for stochastic optimization. Proceedings of the 3rd International Conference on Learning Representations (ICLR).

. Krizhevsky, A., Sutskever, I., & Hinton, G. E. (2012). ImageNet classification with deep convolutional neural networks. Advances in Neural Information Processing Systems, 25, 1097–1105.

. LeCun, Y., Bengio, Y., & Hinton, G. (2015). Deep learning. Nature, 521(7553), 436–444. https://doi.org/10.1038/nature14539

. Shorten, C., & Khoshgoftaar, T. M. (2019). A survey on image data augmentation for deep learning. Journal of Big Data, 6(1), 60. https://doi.org/10.1186/s40537-019-0197-0

. Simonyan, K., & Zisserman, A. (2015). Very deep convolutional networks for large-scale image recognition. International Conference on Learning Representations (ICLR).

. Tan, M., & Le, Q. V. (2019). EfficientNet: Rethinking model scaling for convolutional neural networks. Proceedings of the 36th International Conference on Machine Learning (ICML), 6105–6114.

. Yang, M., & Thung, G. (2016). Classification of trash for recyclability status. Stanford University CS229 Machine Learning Project Report.

. Dhany, H. W., & Sapriadi, S. (2025). Pengembangan perangkat lunak penilaian otomatis ujian pilihan ganda menggunakan algoritma string matching. Jurnal Minfo Polgan, 14(2), 2768-2774.

. Sitorus, Z., Helmy, A., & Ardiya, D. (2025). Forecasting salary ranges for IT professional in marketplace employing the support vector machine technique. Proceedings of International Conference on Islamic Community Studies, 523–532.

. Aryza, S. S., Khowarizmi, A., Furqon, M., Lubis, Z., & Nasution, A. R. (2026). Explainable data-driven machine learning for identifying MBG program beneficiaries in Medan City. Journal of Computer Science, Information Technology and Telecommunication Engineering, 7(1), 1109–1118.

. Novelan, M. S., & Badawi, A. (2026). Penerapan metode Rapid Application Development (RAD) pada sistem absensi karyawan berbasis GPS di CV. Bambang Tetuko. Jurnal Nasional Teknologi Komputer, 6(2), 115–124.

. Wijaya, R. F., & Putra, R. R. (2025). Desain UI/UX mobile app mengenai pengelolaan konten dengan Search Engine Optimization menggunakan pendekatan User Centered Design. Jurnal Komputer Teknologi Informasi Sistem Informasi (JUKTISI), 4(2), 1497–1502.

Downloads

Published

2025-10-27

Most read articles by the same author(s)