arXiv:2512.20746cs.CVcs.LG2025-12中稿 · the WACV 2026 work…

针对边缘设备设计高效垃圾检测模型,实现低功耗高精度部署。

TrashDet: Iterative Neural Architecture Search for Efficient Waste Detection

  • 采用迭代式硬件感知架构搜索,分步优化主干与颈部结构。
  • 在TACO数据集上最高达19.5 mAP50,参数仅30.5M,精度提升3.6点。
  • 适配微控制器,能耗降低88%,适合资源受限的物联网场景。

本文针对边缘和物联网设备上的严格TinyML约束,提出一种迭代式硬件感知神经架构搜索框架,用于TACO数据集的垃圾检测。该方法构建类Once-for-All的ResDets超网络,并通过交替优化主干与颈部/头部结构的进化搜索,结合种群传递机制与准确率预测器,降低搜索成本并提升稳定性。所获系列化模型称为TrashDets。在包含纸张、塑料、瓶子、易拉罐、烟头共五类的TACO子集上,最强版本TrashDet-l达到19.5 mAP50,参数量30.5M,相比先前模型精度提升最高3.6 mAP50,且参数更少。整个系列参数范围1.2M至30.5M,mAP50为11.4至19.5,满足多样化的部署预算需求。在MAX78002微控制器上,基于TrashNet数据集的两个定制变体TrashDet-ResNet与TrashDet-MBNet共同超越ai87-fpndetector基线:前者实现7525~μJ每推理能耗、26.7毫秒延迟与37.45帧/秒;后者提升mAP50达10.2%;整体能耗降低最多88%,延迟减少最多78%,平均功耗下降最多53%。

原文摘要 · Abstract (English)

This paper addresses trash detection on the TACO dataset under strict TinyML constraints using an iterative hardware-aware neural architecture search framework targeting edge and IoT devices. The proposed method constructs a Once-for-All-style ResDets supernet and performs iterative evolutionary search that alternates between backbone and neck/head optimization, supported by a population passthrough mechanism and an accuracy predictor to reduce search cost and improve stability. This framework yields a family of deployment-ready detectors, termed TrashDets. On a five-class TACO subset (paper, plastic, bottle, can, cigarette), the strongest variant, TrashDet-l, achieves 19.5 mAP50 with 30.5M parameters, improving accuracy by up to 3.6 mAP50 over prior detectors while using substantially fewer parameters. The TrashDet family spans 1.2M to 30.5M parameters with mAP50 values between 11.4 and 19.5, providing scalable detector options for diverse TinyML deployment budgets on resource-constrained hardware. On the MAX78002 microcontroller with the TrashNet dataset, two specialized variants, TrashDet-ResNet and TrashDet-MBNet, jointly dominate the ai87-fpndetector baseline, with TrashDet-ResNet achieving 7525~$μ$J energy per inference at 26.7 ms latency and 37.45 FPS, and TrashDet-MBNet improving mAP50 by 10.2%; together they reduce energy consumption by up to 88%, latency by up to 78%, and average power by up to 53% compared to existing TinyML detectors.

垃圾检测TinyML架构搜索边缘计算

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