Indonesian Journal of Artificial Intelligence and Data Mining
Vol. 9 No. 2 (2026): July 2026

Waste Identification Using a Hybrid Convolutional Neural Network and Vision Transformer on Visually Heterogeneous Images

Muhammad Fauzan Adzim (Universitas Lambung Mangkurat)
Dewi Sri Susanti (Universitas Lambung Mangkurat)
Sigit Dwi Prabowo (Universitas Lambung Mangkurat)



Article Info

Publish Date
16 Jul 2026

Abstract

In image classification, convolutional neural networks (CNNs) focus on local patterns, whereas vision Transformers (ViTs) emphasize global context. Combining the two in a hybrid CNN-ViT model may yield a more comprehensive image representation. Waste image classification with visually heterogeneous characteristics can be used to effectively evaluate the performance of the hybrid CNN-ViT model. In addition, such classification supports the crucial need for accurate waste-type identification to enable effective waste management systems. This study investigates a hybrid CNN-ViT model for classifying 24705 organic and recyclable waste images. The workflow involves resizing, an 80:10:10 split, and data augmentation, with models trained for 50 epochs using BCE loss and the Adam optimizer. Evaluation is conducted at the best epoch, defined as the epoch with the highest validation accuracy. For comparison, CNN and ViT models are also trained and evaluated separately. On the test set, the hybrid CNN-ViT model achieves an accuracy of 91.54%, the CNN achieves 91.78%, and the ViT achieves 87.17%. These findings show that CNNs provide an effective and efficient baseline, while the hybrid CNN-ViT model delivers performance competitive with CNNs and is worth considering as a robust alternative for image-based waste classification tasks.

Copyrights © 2026






Journal Info

Abbrev

IJAIDM

Publisher

Subject

Computer Science & IT

Description

Indonesian Journal of Artificial Intelligence and Data Mining (IJAIDM) is an electronic periodical publication published by Puzzle Research Data Technology (Predatech) Faculty of Science and Technology UIN Sultan Syarif Kasim Riau, Indonesia. IJAIDM provides online media to publish scientific ...