Hausdorff Distance Matching with Adaptive Query Denoising for Rotated Detection Transformer
This paper proposes a Rotated Detection Transformer that improves oriented object detection by introducing a Hausdorff distance-based cost for more accurate bipartite matching and an adaptive query denoising method to overcome the limitations of static denoising, achieving significant performance gains on multiple benchmarks.
Original paper licensed under CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/). This is an AI-generated explanation of the paper below. It is not written or endorsed by the authors. For technical accuracy, refer to the original paper. Read full disclaimer
Imagine you are trying to teach a robot to recognize objects in a huge, messy aerial image of a city. Some objects, like cars, are easy to find because they are usually straight and box-shaped. Other objects, such as ships in a harbor or airplanes on a runway, are often tilted at strange angles. To find these, the robot must draw a tilted box around them.
This work introduces a new robot brain (called RHINO) that finds these tilted objects significantly better than previous models. The authors identified two main reasons why older robots struggled and fixed them with two clever tricks.
The Problem: The "Square" Confusion and the "Noisy" Teacher
1. The "Square" Confusion (The Hausdorff Distance Correction)
Imagine a game where you must match a series of red stickers (the robot's guesses) with blue stickers (the actual objects).
- The old way: The old robot used a simple ruler to measure the distance between the center of the red sticker and that of the blue one. Because tilted objects can look like squares from certain angles, the ruler got confused. It sometimes said: "Hey, this red sticker is far away, but it has the same angle as the blue one, so it's a hit!" This caused the robot to draw two boxes for the same object: one good one and one poor one with low confidence.
- The new way (RHINO): The authors realized that instead of looking only at the center, they should consider the entire shape of the box. They used a mathematical tool called Hausdorff distance. Imagine this as measuring the distance between the edges of the shapes, not just between their centers. It is like checking whether the corners of the red sticker actually match the corners of the blue one. This prevented the robot from being confused by square-like shapes and eliminated these duplicate, useless boxes.
2. The "Noisy" Teacher (Adaptive Query Denoising)
To teach the robot faster, the old method used a technique called "Query Denoising." Imagine a teacher trying to teach a student by giving them a smeared, distorted version of the correct answer and asking them to clean it up.
- The problem: At the beginning of training, the robot is very poor, so the teacher's smeared notes are actually helpful. But as soon as the robot becomes smarter and makes perfect predictions, the teacher continues to give it the same smeared, distorted notes. The robot gets confused: "Wait, I know the answer is perfect, but the teacher is telling me to fix this smeared version. Should I follow the teacher or my own brain?" This actually slowed down the robot's learning in later stages.
- The new way (RHINO): The authors made the teacher adaptive. They added a "filter" that checks the robot's current skill level.
- If the robot is a beginner, the filter lets the smeared notes through.
- If the robot is an expert and its own predictions are already better than the smeared notes, the filter throws away the smeared notes. It tells the robot: "You no longer need to fix this junk; trust your own perfect answer." This keeps training focused and efficient.
The Results
The authors tested this new robot (RHINO) on several famous aerial image datasets (such as DOTA and DIOR-R).
- The score: Compared to the previous best models using the same basic hardware (a ResNet-50 backbone), RHINO achieved significantly higher values. It improved accuracy by about 4 to 5 points on a scale where higher values are better.
- The comparison: It beat other top models, even those using much more powerful and complex computer brains (backbones).
In Brief
The work says: "We built a better detector for tilted objects by fixing two things:
- We changed how the robot measures distances so it stops drawing duplicate boxes for square-like objects.
- We made the training process smarter so it stops listening to 'noisy' examples once the robot has already learned the correct answer."
The result is a model that finds tilted objects (such as ships and airplanes) more accurately and with fewer errors than before.
Drowning in papers in your field?
Get daily digests of the most novel papers matching your research keywords — with technical summaries, in your language.