用集成学习提升垃圾检测,少标注也能达到高精度。
Robust and Label-Efficient Deep Waste Detection
- 用集成预测加空间权重生成伪标签,减少人工标注依赖。
- 在零标注数据上训练,性能超过全监督模型。
- 为真实垃圾分拣场景建立新基准,适合工业落地应用。
高效垃圾分类对可持续回收至关重要,但当前人工智能研究因数据有限且依赖传统目标检测器而滞后于商业系统。本文在真实世界数据集ZeroWaste上基准测试了先进开放词汇目标检测(OVOD)模型,发现仅用类别提示效果差,而经大语言模型优化的提示显著提升零样本准确率。为克服领域特异性限制,我们微调基于Transformer的现代检测器,达成51.6 mAP的新基准。随后提出一种融合空间与共识感知权重的软伪标签策略,实现稳健的半监督训练。应用于未标注的ZeroWaste-s子集时,伪标注带来性能提升,超越全监督训练,验证了可扩展标注流程的有效性。本工作为研究社区贡献了严格基线、鲁棒集成伪标签流程、高质量伪标注数据及真实垃圾分拣条件下的系统评估。代码已开源:https://github.com/h-abid97/robust-waste-detection。
原文摘要 · Abstract (English)
Effective waste sorting is critical for sustainable recycling, yet AI research in this domain continues to lag behind commercial systems due to limited datasets and reliance on legacy object detectors. In this work, we advance AI-driven waste detection by establishing strong baselines and introducing an ensemble-based semi-supervised learning framework. We first benchmark state-of-the-art Open-Vocabulary Object Detection (OVOD) models on the real-world ZeroWaste dataset, demonstrating that while class-only prompts perform poorly, LLM-optimized prompts significantly enhance zero-shot accuracy. Next, to address domain-specific limitations, we fine-tune modern transformer-based detectors, achieving a new baseline of 51.6 mAP. We then propose a soft pseudo-labeling strategy that fuses ensemble predictions using spatial and consensus-aware weighting, enabling robust semi-supervised training. Applied to the unlabeled ZeroWaste-s subset, our pseudo-annotations achieve performance gains that surpass fully supervised training, underscoring the effectiveness of scalable annotation pipelines. Our work contributes to the research community by establishing rigorous baselines, introducing a robust ensemble-based pseudo-labeling pipeline, generating high-quality annotations for the unlabeled ZeroWaste-s subset, and systematically evaluating OVOD models under real-world waste sorting conditions. Our code is available at: https://github.com/h-abid97/robust-waste-detection.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。