11일 전
도메인 적응 탐지 트랜스포머의 전이성 향상
Kaixiong Gong, Shuang Li, Shugang Li, Rui Zhang, Chi Harold Liu, Qiang Chen

초록
DETR 기반 검출기들은 도메인 내 시나리오에서 두드러진 성능을 보이지만, 도메인 시프트 설정에서의 특성에 대해서는 여전히 탐색이 부족한 상태이다. 본 논문은 두 가지 발견을 바탕으로 도메인 시프트 환경에서 간단하면서도 효과적인 기준선을 구축하는 것을 목표로 한다. 첫째, 백본과 디코더 출력 특징에서의 도메인 시프트를 완화하는 것이 유리한 결과를 얻는 데 효과적임을 확인하였다. 둘째, 두 부분 모두에서 고급 도메인 정렬 기법을 적용할 경우 성능이 더욱 향상됨을 발견하였다. 이를 바탕으로 백본과 검출기의 출력에 대한 종합적인 도메인 정렬을 달성하기 위해, 객체 인지 정렬(Object-Aware Alignment, OAA) 모듈과 최적 운송 기반 정렬(Optimal Transport-based Alignment, OTA) 모듈을 제안한다. OAA 모듈은 백본 출력에서 의사 레이블( pseudo-labels)을 통해 식별된 전경 영역을 정렬함으로써 도메인 불변 특징을 생성한다. OTA 모듈은 슬라이스된 워셰르슈타인 거리(sliced Wasserstein distance)를 활용하여 디코더 출력에서 위치 정보의 유지와 도메인 간 차이의 최소화를 동시에 달성한다. 본 연구에서 제안한 발견과 정렬 모듈을 적응 방법에 구현함으로써, DETR 기반 검출기가 도메인 시프트 환경에서의 성능을 기준선으로 설정하였다. 다양한 도메인 적응 시나리오에서 수행된 실험을 통해 제안한 방법의 효과성을 입증하였다.