We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.CV

Change to browse by:

cs

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Computer Vision and Pattern Recognition

Title: Integrally Migrating Pre-trained Transformer Encoder-decoders for Visual Object Detection

Abstract: Modern object detectors have taken the advantages of backbone networks pre-trained on large scale datasets. Except for the backbone networks, however, other components such as the detector head and the feature pyramid network (FPN) remain trained from scratch, which hinders fully tapping the potential of representation models. In this study, we propose to integrally migrate pre-trained transformer encoder-decoders (imTED) to a detector, constructing a feature extraction path which is ``fully pre-trained" so that detectors' generalization capacity is maximized. The essential differences between imTED with the baseline detector are twofold: (1) migrating the pre-trained transformer decoder to the detector head while removing the randomly initialized FPN from the feature extraction path; and (2) defining a multi-scale feature modulator (MFM) to enhance scale adaptability. Such designs not only reduce randomly initialized parameters significantly but also unify detector training with representation learning intendedly. Experiments on the MS COCO object detection dataset show that imTED consistently outperforms its counterparts by $\sim$2.4 AP. Without bells and whistles, imTED improves the state-of-the-art of few-shot object detection by up to 7.6 AP. Code is available at this https URL
Comments: 6 figures, 6 tables
Subjects: Computer Vision and Pattern Recognition (cs.CV)
Cite as: arXiv:2205.09613 [cs.CV]
  (or arXiv:2205.09613v2 [cs.CV] for this version)

Submission history

From: Feng Liu [view email]
[v1] Thu, 19 May 2022 15:11:20 GMT (1290kb,D)
[v2] Fri, 2 Dec 2022 14:57:45 GMT (246kb,D)

Link back to: arXiv, form interface, contact.