Image-Based Waste Classification Using a Hybrid Deep Learning Architecture with Transfer Learning and Edge AI Deployment

Growing amounts of municipal waste and the need for efficient recycling demand automated and accurate classification systems. This paper investigates deep learning approaches for multi-class waste sorting based on image data, comparing three widely used convolutional neural network architectures (ResNet-50, EfficientNet-B0, and MobileNet V3) with a custom hybrid model (CustomNet). The dataset comprises 13,933 RGB images across 10 waste categories, combining publicly available samples from the Kaggle Garbage Classification dataset (61.1%) with images collected in house (38.9%). The three glass sub-categories (brown, green, and white glass) were merged into a single glass class to ensure consistent class representation across all dataset splits. Preprocessing steps include normalization, resizing, and extensive data augmentation to improve robustness and mitigate class imbalance. Transfer learning is applied to pretrained models, while CustomNet integrates feature representations from multiple backbones using projection layers and attention mechanisms. Performance is evaluated using accuracy, macro-F1, and ROC–AUC on a held-out test set. Statistical significance was assessed using paired t-tests and Wilcoxon signed-rank tests with Bonferroni correction across five-fold cross-validation runs. The results show that CustomNet achieves 97.79% accuracy, a macro-F1 score of 0.973, and a ROC–AUC of 0.992. CustomNet significantly outperforms EfficientNet-B0 and MobileNet V3 (p<0.001, Bonferroni corrected), and it achieves performance parity with ResNet-50 (p=0.383) at a substantially lower parameter count in the classification head (9.7 M vs. 25.6 M). These findings indicate that combining multiple feature extractors with attention mechanisms improves classification performance, supports qualitative model explainability via saliency visualization (Grad-CAM), and enables practical deployment on heterogeneous Edge AI platforms. Inference benchmarking on an NVIDIA Jetson Orin Nano demonstrated real-world deployment feasibility at 86.70 ms per image (11.5 FPS).