이 연구는 의료 영상 분할에서 널리 사용되는 nnU-Net 모델의 불확실성 추정 방법을 제안한다. 기존의 베이지안 신경망 및 몬테카를로 드롭아웃 방법과 달리, 제안하는 방법은 변분 추론 아키텍처를 요구하지 않으며 nnU-Net의 원래 구조를 그대로 유지한다. 이를 통해 nnU-Net의 우수한 성능과 사용의 편의성을 보존하면서도 불확실성 추정 능력을 향상시킨다.
제안 방법은 SGD 최적화 과정에서 가중치 공간을 탐색하는 특성을 활용한다. 적절한 시점에 가중치 체크포인트를 저장하고, 이를 활용하여 가중치 공간의 다중 모드를 포착하는 다중 모달 베이지안 추론을 수행한다. 이를 통해 기존 방법들보다 향상된 불확실성 추정 성능을 달성한다.
실험 결과, 제안 방법은 심장 MRI 데이터에서 기존 방법들을 능가하는 분할 정확도와 보정 성능을 보였다. 특히 어려운 입력 데이터에 대해 보다 강건한 불확실성 추정 능력을 보였다. 이를 통해 제안 방법이 의료 영상 분할 응용에서 정확도와 품질 관리 측면에서 nnU-Net을 강화할 수 있음을 보였다.
翻譯成其他語言
從原文內容
arxiv.org
深入探究