MedCore: MedSAM을 위한 경계 보존형 의료 코어 프루닝 (Boundary-Preserving Medical Core Pruning)

요약

본 논문은 MedSAM과 같은 의료 세그멘테이션 모델을 위한 구조적 프루닝 프레임워크인 MedCore를 제안합니다. MedCore는 SAM에서 MedSAM으로의 적응 과정에서 중요해진 구조와 높은 경계 레버리지를 가진 구조 두 가지를 보존하는 것이 핵심입니다. 이 방법론은 '경계 레버리지 원리'를 도입하여 압축이 경계 지표에 미치는 영향을 분석하고, 용종 세그멘테이션 벤치마크에서 파라미터와 FLOPs를 대폭 줄이면서도 높은 Dice 및 Boundary F1 점수를 유지함을 입증했습니다.

핵심 포인트

• MedCore는 MedSAM을 위한 구조적 프루닝 프레임워크로, 모델 경량화와 성능 유지를 동시에 목표합니다.
• 프루닝은 SAM→MedSAM 적응 과정에서 중요해진 구조와 높은 경계 레버리지를 가진 구조를 보존하는 것이 핵심입니다.
• 경계 레버리지 원리를 통해 압축으로 인한 경계 변위가 로짓 공간 기울기에 의해 제어됨을 설명합니다.
• 용종 세그멘테이션 벤치마크에서 MedCore는 파라미터와 FLOPs를 크게 줄이면서도 높은 Dice 및 Boundary F1 점수를 달성했습니다.