arXiv:2412.18230cs.CV2024-12被引 3

轻量级工具箱让边缘设备高效部署目标检测,解决精度与速度难题。

Efficient Detection Framework Adaptation for Edge Computing: A Plug-and-play Neural Network Toolbox Enabling Edge Deployment

  • 用可插拔模块动态适配边缘环境,提升模型灵活性。
  • 在真实场景中实现毫秒级响应,准确率超越6个顶尖方法。
  • 专为安全帽带检测设计数据集,适合工业质检等实时应用。

边缘计算已成为时敏场景下部署深度学习目标检测的关键范式。然而,现有方法面临三大挑战:1)难以平衡检测精度与轻量化模型;2)通用部署设计适应性差;3)缺乏真实世界验证。为此,我们提出边缘检测工具箱(ED-TOOLBOX),采用可泛化、即插即用的组件,适配目标检测模型于边缘环境。具体地,引入轻量级重参数化动态卷积网络(Rep-DConvNet),通过加权多形态卷积分支提升检测性能;设计稀疏交叉注意力(SC-A)网络,结合局部映射自注意力机制,实现自适应特征融合;在YOLO框架中集成高效检测头,加速边缘模型优化。为验证实用性,我们发现安全帽佩戴中忽略绑带状态这一关键安全问题,构建了盔带检测数据集(HBDD)。基于ED-TOOLBOX优化的模型成功解决该任务。大量实验表明,其在视觉监控仿真中优于六种先进方法,实现毫秒级响应与高精度,证明了该工具箱在边缘目标检测中的优越性。

原文摘要 · Abstract (English)

Edge computing has emerged as a key paradigm for deploying deep learning-based object detection in time-sensitive scenarios. However, existing edge detection methods face challenges: 1) difficulty balancing detection precision with lightweight models, 2) limited adaptability of generalized deployment designs, and 3) insufficient real-world validation. To address these issues, we propose the Edge Detection Toolbox (ED-TOOLBOX), which utilizes generalizable plug-and-play components to adapt object detection models for edge environments. Specifically, we introduce a lightweight Reparameterized Dynamic Convolutional Network (Rep-DConvNet) featuring weighted multi-shape convolutional branches to enhance detection performance. Additionally, we design a Sparse Cross-Attention (SC-A) network with a localized-mapping-assisted self-attention mechanism, enabling a well-crafted joint module for adaptive feature transfer. For real-world applications, we incorporate an Efficient Head into the YOLO framework to accelerate edge model optimization. To demonstrate practical impact, we identify a gap in helmet detection -- overlooking band fastening, a critical safety factor -- and create the Helmet Band Detection Dataset (HBDD). Using ED-TOOLBOX-optimized models, we address this real-world task. Extensive experiments validate the effectiveness of ED-TOOLBOX, with edge detection models outperforming six state-of-the-art methods in visual surveillance simulations, achieving real-time and accurate performance. These results highlight ED-TOOLBOX as a superior solution for edge object detection.

边缘计算目标检测轻量化工具箱

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