arXiv:2603.27661cs.CV2026-03

通过动态剪枝非边缘特征,显著降低边缘检测计算量。

Amped: Adaptive Multi-stage Non-edge Pruning for Edge Detection

  • 识别高置信度非边缘片段并早期剔除,减少计算开销。
  • 最多降低40%计算量,仅损失0.4%精度(ODS F-measure)。
  • 适合需要高效部署的边缘检测实际应用。

边缘检测是图像分析的基础任务,支撑众多高层视觉应用。尽管基于Transformer的模型通过捕捉长程依赖显著提升了边缘质量,但其计算开销较大。提高像素级精度需更高输入分辨率,进一步增加计算成本,限制实际部署。本文提出自适应多阶段非边缘剪枝框架Amped,利用最新Transformer边缘检测器的强大表征能力,尽早识别并移除高置信度非边缘令牌,大幅降低计算量,实现高精度与低延迟的平衡。此外,为简化现有边缘检测网络结构并便于实际集成,提出轻量级高性能Transformer模型Streamline Edge Detector (SED)。在现有检测器和SED上应用该剪枝策略,可实现高达40%的GFLOPs减少,且仅带来0.4%的ODS F-measure下降。尽管结构简单,SED仍达到86.5%的当前最优ODS F-measure。代码将开源。

原文摘要 · Abstract (English)

Edge detection is a fundamental image analysis task that underpins numerous high-level vision applications. Recent advances in Transformer architectures have significantly improved edge quality by capturing long-range dependencies, but this often comes with computational overhead. Achieving higher pixel-level accuracy requires increased input resolution, further escalating computational cost and limiting practical deployment. Building on the strong representational capacity of recent Transformer-based edge detectors, we propose an Adaptive Multi-stage non-edge Pruning framework for Edge Detection(Amped). Amped identifies high-confidence non-edge tokens and removes them as early as possible to substantially reduce computation, thus retaining high accuracy while cutting GFLOPs and accelerating inference with minimal performance loss. Moreover, to mitigate the structural complexity of existing edge detection networks and facilitate their integration into real-world systems, we introduce a simple yet high-performance Transformer-based model, termed Streamline Edge Detector(SED). Applied to both existing detectors and our SED, the proposed pruning strategy provides a favorable balance between accuracy and efficiency-reducing GFLOPs by up to 40% with only a 0.4% drop in ODS F-measure. In addition, despite its simplicity, SED achieves a state-of-the-art ODS F-measure of 86.5%. The code will be released.

边缘检测Transformer剪枝效率优化

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