Boosting Deep Unsupervised Edge Detection via Segment Anything Model

Wenya Yang, Xiao Diao Chen, Wen Wu, Hongshuai Qin, Kangming Yan, Xiaoyang Mao, Haichuan Song

Research output: Contribution to journalArticlepeer-review

15 Scopus citations

Abstract

Segment anything model (SAM), a vision foundation network trained on a massive segmentation corpus, exhibits a superior boundary localization capability for nature images. This work aims to leverage such strengths to develop a deep unsupervised edge detection (UED) framework for alleviating the high reliance on dense labeling. However, applying vanilla SAM to edge detection fails to identify the salient edge cues but only the semantic boundary. This article introduces a lightweight adapter-tuning scheme to learn detailed edge information for filling the gap between boundary and edge, enabling a well-fitting even with limited training data. Moreover, considering the low-quality pseudo labels used in our UED framework, we propose two training strategies, adaptive progressive learning and gradient-guided pseudo label updating, to alleviate the impact of noisy labels from traditional UED methods. Extensive experiments demonstrate that our method achieves comparable results to state-of-the-art fully supervised edge detectors.

Original languageEnglish
Pages (from-to)8961-8971
Number of pages11
JournalIEEE Transactions on Industrial Informatics
Volume20
Issue number6
DOIs
StatePublished - 1 Jun 2024

Keywords

  • Edge detection
  • segment anything model (SAM)
  • self-learning
  • unsupervised learning

Fingerprint

Dive into the research topics of 'Boosting Deep Unsupervised Edge Detection via Segment Anything Model'. Together they form a unique fingerprint.

Cite this