UOD: Universal One-shot Detection of Anatomical Landmarks
Created by
Haebom
저자
Heqin Zhu, Quan Quan, Qingsong Yao, Zaiyi Liu, S. Kevin Zhou
개요
본 논문은 다중 도메인 비표지 데이터 상황에서 도메인 편향 문제를 겪는 기존의 단일 샷 의료 랜드마크 검출 방법의 한계를 극복하기 위해, 도메인 적응형 단일 샷 랜드마크 검출 프레임워크인 UOD(Universal One-shot Detection)를 제안합니다. UOD는 도메인 특정 모듈과 도메인 공유 모듈의 조합으로 설계된 두 단계와 두 개의 범용 모델로 구성됩니다. 첫 번째 단계에서는 도메인 적응형 합성곱 모델을 자기 지도 학습하여 의사 랜드마크 레이블을 생성하고, 두 번째 단계에서는 도메인 적응형 트랜스포머를 설계하여 도메인 편향을 제거하고 다중 도메인 데이터에 대한 전역 컨텍스트를 구축합니다. 각 도메인에서 단 하나의 주석이 달린 샘플만 사용 가능하더라도, 도메인 공유 모듈은 UOD가 모든 단일 샷 샘플을 집계하여 더욱 강력하고 정확한 랜드마크를 검출하도록 돕습니다. 세 가지 널리 사용되는 공개 X-레이 데이터셋(두부, 손, 흉부)에서 UOD를 정성적 및 정량적으로 평가하여 각 도메인에서 최첨단 성능을 달성했습니다. 코드는 https://github.com/heqin-zhu/UOD_universal_oneshot_detection 에서 이용 가능합니다.