无需大量标注数据,用对比学习和多聚类投票实现高效垃圾分类。
Unsupervised Waste Classification By Dual-Encoder Contrastive Learning and Multi-Clustering Voting (DECMCV)
- 双编码器对比学习+多聚类投票,自动挖掘图像特征并聚类。
- 仅需50个标签样本,在真实数据集上提升准确率29.85%。
- 适合资源有限但需高泛化能力的垃圾分类场景。
垃圾分类对提升处理效率、减少环境污染至关重要。现有监督深度学习方法依赖大量标注数据,获取成本高且易受类别与风格偏差影响(如拍摄角度、光照、垃圾类型差异),制约模型泛化能力。构建无偏数据集困难,人工标注效率低。自监督学习虽缓解数据稀缺,仍需部分标签且精度低于监督方法。本研究提出一种新型无监督方法——双编码器对比学习与多聚类投票(DECMCV)。该方法采用预训练ConvNeXt进行图像编码,利用Vision Transformer生成正样本,结合多聚类投票机制应对标注与领域漂移问题。实验表明,DECMCV在TrashNet和Huawei Cloud数据集上分别达到93.78%和98.29%的分类准确率,优于或匹配监督模型;在包含4,169张真实垃圾图像的数据集上,仅用50个标注样本即完成数千张图像的准确标注,较监督模型提升29.85%。该方法有效缓解风格差异,增强模型泛化性,推动自动化垃圾分类发展。
原文摘要 · Abstract (English)
Waste classification is crucial for improving processing efficiency and reducing environmental pollution. Supervised deep learning methods are commonly used for automated waste classification, but they rely heavily on large labeled datasets, which are costly and inefficient to obtain. Real-world waste data often exhibit category and style biases, such as variations in camera angles, lighting conditions, and types of waste, which can impact the model's performance and generalization ability. Therefore, constructing a bias-free dataset is essential. Manual labeling is not only costly but also inefficient. While self-supervised learning helps address data scarcity, it still depends on some labeled data and generally results in lower accuracy compared to supervised methods. Unsupervised methods show potential in certain cases but typically do not perform as well as supervised models, highlighting the need for an efficient and cost-effective unsupervised approach. This study presents a novel unsupervised method, Dual-Encoder Contrastive Learning with Multi-Clustering Voting (DECMCV). The approach involves using a pre-trained ConvNeXt model for image encoding, leveraging VisionTransformer to generate positive samples, and applying a multi-clustering voting mechanism to address data labeling and domain shift issues. Experimental results demonstrate that DECMCV achieves classification accuracies of 93.78% and 98.29% on the TrashNet and Huawei Cloud datasets, respectively, outperforming or matching supervised models. On a real-world dataset of 4,169 waste images, only 50 labeled samples were needed to accurately label thousands, improving classification accuracy by 29.85% compared to supervised models. This method effectively addresses style differences, enhances model generalization, and contributes to the advancement of automated waste classification.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。