Detection Transformer with Stable Matching

Stable-DINO로 명칭

 

기존 DINO 보다 좋은 성능을 보여준다. DINO도 점점 샌드백 알고리즘이 되어가는 느낌…

 

기존 DINO에서 해당 case의 경우 multi optimization path가 발생한다.

DETR-like detector는 고전 detector와 다르게 query와 gt가 one-to-one 매칭 (하나의 gt에 여러개의 anchor box가 매칭되는 것과는 다름)

따라서 multi optimization path문제에 더 취약하다.

DETR-like detector에서 하나의 query가 매 epoch 마다 다른 gt에 matching 되는 문제는 loss 수렴을 방해

이를 해결하기위해 많은 방법들이 나왔는데 기본적으로 초기 query값을 좋은 애들로 만들어 보자는 노력들이 대부분
(conditional-DETR, DAB-DETR, DINO 등등)

epoch사이의 일관성뿐만 아니라 decoder layer사이의 gt switching 문제도 해결하려는 노력

근본적으로 matching cost를 변경해줘야 한다.

 

기존 DINO classification loss

 

matching에 사용되는 cost

 

여기서 제안하는 position-supervised loss

 

Cost

 

결과

 

Memory fusion

 

검출 결과