Source-linked AI summary

LSTD: A Low-Shot Transfer Detector for Object Detection

Hao Chen, Yali Wang, Guoyou Wang, Yu Qiao

arXiv:1803.01529v1cs.CV

TL;DR

Object detection performance depends on large fully annotated datasets, but target tasks may have only a few annotated images. LSTD transfers a pretrained source-domain detector and applies TK and BD regularization during target fine-tuning; it outperforms other state-of-the-art approaches in challenging low-shot experiments.

  • Problem

    Fully annotated target detection data may be limited, while general transfer strategies face task differences and detector overfitting in low-shot settings.

    Source: §1 · p. 1
  • Method

    LSTD combines SSD and Faster RCNN designs with source-domain pretraining and TK and BD regularization during target-domain fine-tuning.

    Source: §1 · pp. 1–2, §3 · p. 5
  • Results

    LSTD outperforms other state-of-the-art approaches on a number of challenging low-shot detection experiments.

    Source: §1 · p. 2, §5 · p. 7
  • Takeaways & Limitations

    LSTD is presented as a preferable deep detector for low-shot detection scenarios.

    Source: §5 · p. 7

    Weakly or semi-supervised detectors may have limited performance because their training images lack sufficient supervision.

    Source: §1 · p. 1

Abstract

from arXiv · show

Recent advances in object detection are mainly driven by deep learning with large-scale detection benchmarks. However, the fully-annotated training set is often limited for a target detection task, which may deteriorate the performance of deep detectors. To address this challenge, we propose a novel low-shot transfer detector (LSTD) in this paper, where we leverage rich source-domain knowledge to construct an effective target-domain detector with very few training examples. The main contributions are described as follows. First, we design a flexible deep architecture of LSTD to alleviate transfer difficulties in low-shot detection. This architecture can integrate the advantages of both SSD and Faster RCNN in a unified deep framework. Second, we introduce a novel regularized transfer learning framework for low-shot detection, where the transfer knowledge (TK) and background depression (BD) regularizations are proposed to leverage object knowledge respectively from source and target domains, in order to further enhance fine-tuning with a few target images. Finally, we examine our LSTD on a number of challenging low-shot detection experiments, where LSTD outperforms other state-of-the-art approaches. The results demonstrate that LSTD is a preferable deep detector for low-shot scenarios.

1 Introduction

LSTD addresses low-shot object detection by transferring source-domain detector knowledge to scarce target data. It combines SSD and Faster RCNN designs with TK and BD regularization to support target-domain fine-tuning.

  • Fully annotated target training sets may be limited, restricting the power of deep object detectors.Source: §1 · p. 1
  • Transfer learning remains challenging because small target sets cannot readily eliminate detection–classification task differences and detectors overfit more readily than classifiers.Source: §1 · p. 1
  • LSTD integrates SSD-style multi-convolutional bounding-box regression with Faster RCNN-style coarse-to-fine object classification.Source: §1 · p. 1, Figure 1 · p. 2
  • TK transfers source-object-label knowledge to target proposals, while BD uses target bounding-box knowledge as extra feature-map supervision to suppress background disturbances.Source: §1 · p. 2
  • LSTD outperforms other state-of-the-art approaches across challenging low-shot detection experiments.Source: §1 · p. 2

2 Related Works

The related work contrasts standard deep detectors and existing low-shot methods with LSTD, which combines detector architectures and transfers object knowledge from source and target domains.

Object Detection

  • Faster RCNN uses region proposals followed by Fast RCNN, whereas SSD is a one-stage detector with multi-layer bounding-box regression for varied object sizes.Source: §2 · p. 2
  • Directly applying Faster RCNN or SSD with few examples is problematic because their architectures lack low-shot considerations.Source: §2 · p. 2

Low-shot Learning

  • Existing low-shot learning approaches mainly address classification rather than object detection.Source: §2 · p. 2

LSTD

  • LSTD combines Faster RCNN and SSD advantages and uses regularization to integrate object knowledge from source and target domains.Source: §2 · pp. 2–3

3 Low-Shot Transfer Detector (LSTD)

LSTD combines SSD-style bounding-box regression with Faster RCNN-style coarse-to-fine classification, then transfers source and target object knowledge through TK and BD regularization for low-shot detection.

Basic Deep Architecture of LSTD

  • Pretrained SSD regression parameters can initialize the target domain, avoiding random re-initialization of bounding-box regression with few target images.This reduces fine-tuning burdens when target data are scarce.Source: §3.1 · p. 3
  • LSTD combines SSD-style multi-layer bounding-box regression with Faster RCNN-style coarse-to-fine object classification in a unified architecture.The design separates regression and classification across relatively separate places to decompose low-shot learning difficulties.Source: §3.1 · pp. 3–4

Regularized Transfer Learning for LSTD

  • LSTD fine-tunes a source-domain detector on the target domain with a regularization combining background-depression and transfer-knowledge terms.The total regularization uses coefficients λBD and λTK for LBD and LTK.Source: §3.2 · p. 4
  • BD masks middle-level feature cubes using target ground-truth boxes and penalizes background activation, helping suppress background regions and focus on target objects.Figure 3 reports reduced background disturbances on feature heatmaps.Source: §3.2 · p. 4, Figure 3 · p. 4
  • TK supplies source-object knowledge for target proposals despite different categories, using source-class soften-labels to transfer relations such as color or shape similarity.Figure 4 illustrates relevant source categories for Cow/Aeroplane proposals through Bear/Kite soften-labels.Source: §3.2 · p. 4, Figure 4 · p. 4

1. Pre-training on Large-scale Source Domain

Source-domain LSTD is trained using a large-scale detection benchmark.

Pre-training on Large-scale Source Domain

  • Source-domain LSTD is trained with a large-scale detection benchmark.Source: §1 · p. 5

2. Initialization for Small-scale Target Domain

Target-domain LSTD reuses the source detector except for the object classifier, which is reinitialized because source and target object categories differ.

Initialization for Small-scale Target Domain

  • The last layer of the (K + 1)-object classifier is randomly initialized because source and target object categories differ.All other parts of target-domain LSTD are initialized from source-domain LSTD.Source: §2 · p. 5

3. Fine-tuning for Small-scale Target Domain

LSTD fine-tunes the source-pretrained detector on a small target set with BD and TK regularizations, integrating source-domain knowledge into target-domain learning.

Fine-tuning for Small-scale Target Domain

  • Target-domain LSTD is fine-tuned with BD and TK regularizations using the small training set.Source: §3 · p. 5
  • Target-domain proposals are passed through the source-domain LSTD ROI pooling and object classifier to generate source-domain knowledge vectors.Source: §3 · p. 5
  • The source classifier converts proposal activations into soften-labels using softmax with temperature τ > 1, retaining richer label-relation information.Source: §3 · p. 5
  • A source-object soften classifier is added to target-domain LSTD so each target proposal predicts source-object categories as an auxiliary task.Source: §3 · p. 5
  • Cross-entropy between source-domain knowledge and target predictions forms the TK regularization, integrating source knowledge into target training.Source: §3 · p. 5

4 Experiment

The experiments evaluate LSTD across transfer tasks with scarce target-domain annotations, comparing its architecture, regularizations, robustness, and state-of-the-art performance. LSTD improves low-shot detection over closely related and weakly/semi-supervised alternatives under the reported settings.

Experimental setup

  • LSTD is evaluated on three transfer tasks using large-scale source training and 1/2/5/10/30 target training images per object class.The datasets include COCO, ImageNet2015, VOC2007, and VOC2010.Source: §4 · p. 5

Basic Deep Structure of LSTD

  • In Task 1, LSTD achieves competitive source-domain mAP and significantly outperforms SSD and Faster RCNN for one-shot target detection.The comparison uses models fine-tuned from source-domain pretrained models.Source: §4.1 · p. 6

Regularized Transfer Learning for LSTD

  • TK and BD regularizations significantly improve LSTD fine-tuning over the baseline, especially when the target training set is scarce, such as in one-shot detection.BD remains generally robust when implemented on different convolutional layers.Source: Table 3 · p. 6, §4.1 · p. 7

Comparison with the State-of-the-art

  • LSTD outperforms SSD and Faster RCNN in Task 1 as target-shot counts change, and surpasses weakly- or semi-supervised detectors beyond two shots in Tasks 2 and 3.The weakly- and semi-supervised results are taken from their original papers on the same test set.Source: Figure 5 · p. 6, §4.2 · p. 7
  • LSTD outperforms competing detectors on Task 3 with 0.4% of training data and is competitive with fully supervised detectors using 11% of the training set.At 11%, the reported mAP values are LSTD:69.7, SSD:68.0, and Faster RCNN:69.9.Source: §4.2 · p. 7

Visualization

  • Visualization shows that one-shot LSTD can localize but sometimes misclassifies objects, while five-shot training largely clarifies these errors.For 2-shot LSTD on VOC 2007, Animal errors are reported as 23% localization, 71% similarity, and 4% background.Source: §4.3 · p. 7, Figure 6 · p. 7

5 Conclusion

The conclusion presents LSTD as a low-shot transfer detector for object detection with few training images. It combines a flexible architecture with TK and BD regularization and reports superior performance across challenging experiments.

  • LSTD addresses object detection with a few training images through a flexible deep architecture designed to reduce low-shot transfer difficulties.Source: §5 · p. 7
  • The architecture incorporates strengths of SSD and Faster RCNN within a unified deep framework.Source: §5 · p. 7
  • TK and BD terms form a regularized transfer-learning framework intended to generalize fine-tuning with few target images.Source: §5 · p. 7
  • LSTD outperforms other state-of-the-art approaches across challenging low-shot detection experiments.Source: §5 · p. 7
Loading 1803.01529v1…