Saved in:
Bibliographic Details
Main Authors: Lee, Hakjin, Song, MinKi, Koo, Jamyoung, Seo, Junghoon
Format: Preprint
Published: 2023
Subjects:
Online Access:https://arxiv.org/abs/2305.07598
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917459476348928
author Lee, Hakjin
Song, MinKi
Koo, Jamyoung
Seo, Junghoon
author_facet Lee, Hakjin
Song, MinKi
Koo, Jamyoung
Seo, Junghoon
contents Detection Transformers (DETR) have recently set new benchmarks in object detection. However, their performance in detecting rotated objects lags behind established oriented object detectors. Our analysis identifies a key observation: the boundary discontinuity and square-like problem in bipartite matching poses an issue with assigning appropriate ground truths to predictions, leading to duplicate low-confidence predictions. To address this, we introduce a Hausdorff distance-based cost for bipartite matching, which more accurately quantifies the discrepancy between predictions and ground truths. Additionally, we find that a static denoising approach impedes the training of rotated DETR, especially as the quality of the detector's predictions begins to exceed that of the noised ground truths. To overcome this, we propose an adaptive query denoising method that employs bipartite matching to selectively eliminate noised queries that detract from model improvement. When compared to models adopting a ResNet-50 backbone, our proposed model yields remarkable improvements, achieving $\textbf{+4.18}$ AP$_{50}$, $\textbf{+4.59}$ AP$_{50}$, and $\textbf{+4.99}$ AP$_{50}$ on DOTA-v2.0, DOTA-v1.5, and DIOR-R, respectively.
format Preprint
id arxiv_https___arxiv_org_abs_2305_07598
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Hausdorff Distance Matching with Adaptive Query Denoising for Rotated Detection Transformer
Lee, Hakjin
Song, MinKi
Koo, Jamyoung
Seo, Junghoon
Computer Vision and Pattern Recognition
Machine Learning
Detection Transformers (DETR) have recently set new benchmarks in object detection. However, their performance in detecting rotated objects lags behind established oriented object detectors. Our analysis identifies a key observation: the boundary discontinuity and square-like problem in bipartite matching poses an issue with assigning appropriate ground truths to predictions, leading to duplicate low-confidence predictions. To address this, we introduce a Hausdorff distance-based cost for bipartite matching, which more accurately quantifies the discrepancy between predictions and ground truths. Additionally, we find that a static denoising approach impedes the training of rotated DETR, especially as the quality of the detector's predictions begins to exceed that of the noised ground truths. To overcome this, we propose an adaptive query denoising method that employs bipartite matching to selectively eliminate noised queries that detract from model improvement. When compared to models adopting a ResNet-50 backbone, our proposed model yields remarkable improvements, achieving $\textbf{+4.18}$ AP$_{50}$, $\textbf{+4.59}$ AP$_{50}$, and $\textbf{+4.99}$ AP$_{50}$ on DOTA-v2.0, DOTA-v1.5, and DIOR-R, respectively.
title Hausdorff Distance Matching with Adaptive Query Denoising for Rotated Detection Transformer
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2305.07598