Sign In

No More Sliding Window: Efficient 3D Medical Image Segmentation with Differentiable Top-k Patch Sampling

Created by
  • Haebom
Category
Empty

저자

Young Seok Jeon, Hongfei Yang, Huazhu Fu, Mengling Feng

개요

본 논문은 3D 의료 영상 분할에서 슬라이딩 윈도우(SW) 기반 추론의 속도 저하 문제를 해결하기 위해 No-More-Sliding-Window (NMSW) 프레임워크를 제안합니다. NMSW는 차별 가능한 Top-k 모듈을 사용하여 관련성이 높은 패치만 선택적으로 샘플링하여 불필요한 계산을 최소화합니다. 패치 수준 예측이 부족한 경우, 전역 예측을 활용하여 결과를 개선합니다. 다양한 3D 분할 백본과 과제에 대한 평가 결과, NMSW는 SW 추론과 비교하여 경쟁력 있는 정확도를 달성하면서 계산 복잡도를 91% 감소시키고, H100 GPU에서는 9.1배, Xeon Gold CPU에서는 11.1배 빠른 추론 속도를 보였습니다. 모델 독립적인 특성으로 기존의 효율적인 분할 백본과 통합하여 효율성을 더욱 높일 수 있습니다.

시사점, 한계점

시사점:
3D 의료 영상 분할에서 슬라이딩 윈도우 기반 추론의 속도 저하 문제를 효과적으로 해결.
NMSW 프레임워크를 통해 계산 복잡도와 추론 시간을 획기적으로 단축.
모델 독립적인 구조로 다양한 3D 분할 백본에 적용 가능.
경쟁력 있는 정확도 유지하면서 효율성을 크게 향상.
한계점:
Top-k 모듈의 성능은 선택된 패치의 관련성에 의존적일 수 있음.
전역 예측을 활용하는 방식의 성능은 데이터셋 특성에 따라 달라질 수 있음.
제한된 데이터셋과 백본 네트워크를 사용하여 평가되었으므로 일반화 성능에 대한 추가 연구 필요.
👍