深層学習ベースのエッジ検出器は、多くの場合、複数のアノテーターによって提供されるピクセル単位のラベルに大きく依存しています。既存の方法は、単純な投票プロセスを使用して複数の注釈を融合し、エッジの固有のあいまいさと注釈者のラベル付けバイアスを無視します。この論文では、不確実性を利用して多様な注釈の主観性と曖昧性を調査する、新しい不確実性認識エッジ検出器 (UAED) を提案します。具体的には、最初に決定論的ラベル空間を学習可能なガウス分布に変換します。その分散は、異なる注釈間のあいまいさの程度を測定します。次に、学習した分散を予測エッジ マップの推定不確実性と見なし、不確実性が高いピクセルは、エッジ検出のハード サンプルになる可能性があります。したがって、ネットワークが重要なピクセルに徐々に集中するのに役立つ、不確実性の高いピクセルからの学習を強調するために、適応的な重み付け損失を設計します。 UAED はさまざまなエンコーダー/デコーダー バックボーンと組み合わせることができ、広範な実験により、UAED が複数のエッジ検出ベンチマークで一貫して優れたパフォーマンスを達成することが実証されています。ソースコードは https://github.com/ZhouCX117/UAED で入手できます
Deep learning-based edge detectors heavily rely on pixel-wise labels which are often provided by multiple annotators. Existing methods fuse multiple annotations using a simple voting process, ignoring the inherent ambiguity of edges and labeling bias of annotators. In this paper, we propose a novel uncertainty-aware edge detector (UAED), which employs uncertainty to investigate the subjectivity and ambiguity of diverse annotations. Specifically, we first convert the deterministic label space into a learnable Gaussian distribution, whose variance measures the degree of ambiguity among different annotations. Then we regard the learned variance as the estimated uncertainty of the predicted edge maps, and pixels with higher uncertainty are likely to be hard samples for edge detection. Therefore we design an adaptive weighting loss to emphasize the learning from those pixels with high uncertainty, which helps the network to gradually concentrate on the important pixels. UAED can be combined with various encoder-decoder backbones, and the extensive experiments demonstrate that UAED achieves superior performance consistently across multiple edge detection benchmarks. The source code is available at https://github.com/ZhouCX117/UAED