CAFENet: Class-Agnostic Few-Shot Edge Detection Network

Abstract

We tackle a novel few-shot learning challenge, which we call few-shot semantic edge detection, aiming to localize crisp boundaries of novel categories using only a few labeled samples. We also present a Class-Agnostic Few-shot Edge detection Network (CAFENet) based on meta-learning strategy. CAFENet employs a semantic segmentation module in small-scale to compensate for lack of semantic information in edge labels. The predicted segmentation mask is used to generate an attention map to highlight the target object region, and make the decoder module concentrate on that region. We also propose a new regularization method based on multi-split matching. In meta-training, the metric-learning problem with high-dimensional vectors are divided into small subproblems with low-dimensional sub-vectors. Since there is no existing dataset for few-shot semantic edge detection, we construct two new datasets, FSE-1000 and SBD-$5^i$, and evaluate the performance of the proposed CAFENet on them. Extensive simulation results confirm the performance merits of the techniques adopted in CAFENet.

15 pages, 6 figures

References in corpus (13)

• Adam: A Method for Stochastic Optimization
• Decoupled Weight Decay Regularization
• Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networks
• Prototypical Networks for Few-shot Learning
• Understanding Neural Networks Through Deep Visualization
• Richer Convolutional Features for Edge Detection
• Learning to Propagate Labels: Transductive Propagation Network for Few-shot Learning
• FSS-1000: A 1000-Class Dataset for Few-Shot Segmentation
• Cross Attention Network for Few-shot Classification
• SG-One: Similarity Guidance Network for One-Shot Semantic Segmentation
• Pixel-wise Deep Learning for Contour Detection
• Semantic Edge Detection with Diverse Deep Supervision
• Dynamic Feature Fusion for Semantic Edge Detection