混合模型让垃圾图像分类更准更快,精度达100%且提速超95%
Towards Accurate and Efficient Waste Image Classification: A Hybrid Deep Learning and Machine Learning Approach
- 用深度模型提特征,传统分类器做判断,兼顾准确与效率
- 在三个数据集上最高达100%准确率,超越当前最佳水平
- 特征维度降低95%以上,适合低算力设备部署
自动化垃圾图像分类是全球废物管理的关键环节;然而,集成机器学习(ML)、深度学习(DL)及高效混合方案的系统性基准仍不完善。本研究全面比较三种范式:(1) 使用手工特征的机器学习算法,(2) 包括ResNet变体和EfficientNetV2S在内的深度学习架构,以及(3) 混合方法——利用深度模型提取特征,再结合支持向量机(SVM)和逻辑回归等经典分类器进行识别。在三个公开数据集(TrashNet、Garbage Classification 和经43个误标修正的家用垃圾数据集)上的实验表明,混合方法始终表现最优,在TrashNet和修正后家用数据集上达到100%准确率,在Garbage Classification上达99.87%,超过现有最佳基准。此外,特征选择将特征维度降低超过95%而不损失精度,显著提升训练与推理速度。本工作建立了更可靠的垃圾分类基准,并提出一种高效混合框架,实现高精度的同时降低推理成本,适用于资源受限环境下的可扩展部署。
原文摘要 · Abstract (English)
Automated image-based garbage classification is a critical component of global waste management; however, systematic benchmarks that integrate Machine Learning (ML), Deep Learning (DL), and efficient hybrid solutions remain underdeveloped. This study provides a comprehensive comparison of three paradigms: (1) machine learning algorithms using handcrafted features, (2) deep learning architectures, including ResNet variants and EfficientNetV2S, and (3) a hybrid approach that utilizes deep models for feature extraction combined with classical classifiers such as Support Vector Machine and Logistic Regression to identify the most effective strategy. Experiments on three public datasets - TrashNet, Garbage Classification, and a refined Household Garbage Dataset (with 43 corrected mislabels)- demonstrate that the hybrid method consistently outperforms the others, achieving up to 100% accuracy on TrashNet and the refined Household set, and 99.87% on Garbage Classification, thereby surpassing state-of-the-art benchmarks. Furthermore, feature selection reduces feature dimensionality by over 95% without compromising accuracy, resulting in faster training and inference. This work establishes more reliable benchmarks for waste classification and introduces an efficient hybrid framework that achieves high accuracy while reducing inference cost, making it suitable for scalable deployment in resource-constrained environments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。