SAUGE: Taming SAM for Uncertainty-Aligned Multi-Granularity Edge Detection

TL;DR

This paper introduces SAUGE, a novel method that leverages the segment anything model (SAM) to explicitly model and align uncertainty in multi-granularity edge detection, improving generalization and flexibility.

Contribution

SAUGE utilizes SAM's intermediate features to explicitly model edge uncertainty at multiple granularities, with a lightweight module for adaptive fusion, enhancing cross-dataset performance.

Findings

Outperforms existing methods on BSDS500, Muticue, NYUDv2 datasets.

Demonstrates strong generalizability for cross-dataset edge detection.

Flexible in producing edges at any granularity level.

Abstract

Edge labels are typically at various granularity levels owing to the varying preferences of annotators, thus handling the subjectivity of per-pixel labels has been a focal point for edge detection. Previous methods often employ a simple voting strategy to diminish such label uncertainty or impose a strong assumption of labels with a pre-defined distribution, e.g., Gaussian. In this work, we unveil that the segment anything model (SAM) provides strong prior knowledge to model the uncertainty in edge labels. Our key insight is that the intermediate SAM features inherently correspond to object edges at various granularities, which reflects different edge options due to uncertainty. Therefore, we attempt to align uncertainty with granularity by regressing intermediate SAM features from different layers to object edges at multi-granularity levels. In doing so, the model can fully and explicitly explore diverse ``uncertainties'' in a data-driven fashion. Specifically, we inject a lightweight module (~ 1.5% additional parameters) into the frozen SAM to progressively fuse and adapt its intermediate features to estimate edges from coarse to fine. It is crucial to normalize the granularity level of human edge labels to match their innate uncertainty. For this, we simply perform linear blending to the real edge labels at hand to create pseudo labels with varying granularities. Consequently, our uncertainty-aligned edge detector can flexibly produce edges at any desired granularity (including an optimal one). Thanks to SAM, our model uniquely demonstrates strong generalizability for cross-dataset edge detection. Extensive experimental results on BSDS500, Muticue and NYUDv2 validate our model's superiority.