RP-SAM2: Refining Point Prompts for Stable Surgical Instrument Segmentation
Created by
Haebom
저자
Nuren Zhaksylyk, Ibrahim Almakky, Jay Paranjape, S. Swaroop Vedula, Shameema Sikder, Vishal M. Patel, Mohammad Yaqub
개요
백내장 수술에서 수술 기구 분할은 기술 평가 및 워크플로 최적화에 필수적이나, 부족한 주석 데이터로 인해 완전 자동화 모델 개발이 어렵습니다. 본 논문에서는 SAM2와 같은 프롬프트 기반 방법의 한계점인 포인트 프롬프트 위치에 대한 민감성을 해결하기 위해, 새로운 shift block과 복합 손실 함수를 통합한 RP-SAM2를 제시합니다. RP-SAM2는 정확한 포인트 배치에 대한 의존성을 줄이면서 견고한 분할 성능을 유지합니다. Cataract1k 데이터셋 실험 결과, RP-SAM2는 SAM2에 비해 mDSC 2% 향상, mHD95 21.36% 감소 및 단일 포인트 프롬프트 결과의 분산 감소를 보였습니다. 또한, CaDIS 데이터셋에서 SAM2의 마스크 디코더 미세 조정을 위한 RP-SAM2의 의사 마스크가 SAM2의 의사 마스크보다 우수한 성능을 보였습니다. 이는 데이터 제약이 있는 의료 환경에서 반자동 수술 기구 분할을 위한 실용적이고 안정적이며 신뢰할 수 있는 솔루션으로 RP-SAM2를 제시합니다. 코드는 https://github.com/BioMedIA-MBZUAI/RP-SAM2 에서 확인할 수 있습니다.