arXiv:2411.02779cs.CV2024-11中稿 · the 6th Internatio…被引 12

对比多种深度学习模型,提升垃圾分类准确率。

Advancing Recycling Efficiency: A Comparative Analysis of Deep Learning Models in Waste Classification

  • 融合ResNet50与SVM,提升复杂类别分类精度。
  • Transformer模型平均准确率最高,适合复杂任务。
  • 数据增强与模型优化可改善低表现类别性能。

随着全球人口增长和消费习惯加剧,垃圾产量持续上升,给垃圾管理和回收优化带来巨大挑战。本研究针对回收中的垃圾分类问题,分析了卷积神经网络(CNN)、AlexNet、ResNet、ResNet50加支持向量机(SVM)以及Transformer等多种深度学习模型在多类垃圾上的表现。通过参数设置、类别准确率、总准确率和模型参数等指标进行系统比较,建立统一评估标准。研究提出一种将SVM与深度学习框架(特别是ResNet50)结合的新方法,显著提升复杂垃圾类别的分类准确率。同时,Transformer模型在平均准确率上优于其他模型,展现出处理复杂分类任务的优势。为改善低表现类别的性能,研究建议扩大数据集、使用数据增强,并采用Transformer等先进模型及优化训练方法。该研究为多类别垃圾回收的未来发展提供方向,凸显深度学习在推动环境可持续性中的关键作用。

原文摘要 · Abstract (English)

With the ongoing increase in the worldwide population and escalating consumption habits,there's a surge in the amount of waste produced.The situation poses considerable challenges for waste management and the optimization of recycling operations.The research tackles the pressing issue of waste classification for recycling by analyzing various deep learning models,including Convolutional Neural Network(CNN),AlexNet,ResNet,ResNet50 plus Support Vector Machine(SVM),and transformers,across a wide array of waste categories.The research meticulously compares these models on several targets like parameters settings,category accuracy,total accuracy and model parameters to establish a uniform evaluation criterion.This research presents a novel method that incorporates SVM with deep learning frameworks,particularly ResNet50.The results indicate the method significantly boosts accuracy in complex waste categories.Moreover,the transformer model outshines others in average accuracy,showcasing its aptitude for intricate classification tasks.To improve performance in poorly performing categories,the research advocates for enlarging the dataset,employing data augmentation,and leveraging sophisticated models such as transformers,along with refining training methodologies.The research paves the way for future advancements in multi-category waste recycling and underscores the pivotal role of deep learning in promoting environmental sustainability.

垃圾分类深度学习RecyclingTransformer

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。