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 기반 검출기가 도메인 시프트 환경에서의 성능을 기준선으로 설정하였다. 다양한 도메인 적응 시나리오에서 수행된 실험을 통해 제안한 방법의 효과성을 입증하였다.

도메인 적응 탐지 트랜스포머의 전이성 향상 | 최신 연구 논문 | HyperAI초신경