Test-Time Domain Generalization via Universe Learning: A Multi-Graph Matching Approach for Medical Image Segmentation
Created by
Haebom
Category
Empty
저자
Xingguo Lv, Xingbo Dong, Liwen Wang, Jiewen Yang, Lei Zhao, Bin Pu, Zhe Jin, Xuejun Li
개요
본 논문은 의료 영상 분할에서 도메인 변화로 인한 성능 저하 문제를 해결하기 위해, 사전 훈련된 모델의 성능 저하를 완화하는 도메인 일반화(DG)와 테스트 시간 적응(TTA)을 결합한 새로운 프레임워크를 제시한다. 기존 TTA 방법들이 의료 영상의 고유한 사전 지식을 고려하지 못하는 한계를 극복하기 위해, 형태 정보를 통합한 다중 그래프 매칭 기반 프레임워크를 제안한다. 학습 가능한 유니버스 임베딩을 도입하여 다중 소스 훈련 중 형태학적 사전 정보를 통합하고, 새로운 비지도 테스트 시간 도메인 적응 패러다임을 제시한다. 이를 통해 다중 매칭에서 순환 일관성을 보장하고, 미지의 데이터의 불변 사전 정보를 효과적으로 포착하여 도메인 변화의 영향을 크게 완화한다. 두 가지 의료 영상 분할 벤치마크에 대한 광범위한 실험을 통해, 제안된 방법이 다중 소스 및 단일 소스 도메인 일반화 작업 모두에서 최첨단 접근 방식을 능가함을 보여준다. 소스 코드는 https://github.com/Yore0/TTDG-MGM 에서 확인 가능하다.