[공지사항]을 빙자한 안부와 근황 
Show more

Daily Arxiv

This is a page that curates AI-related papers published worldwide.
All content here is summarized using Google Gemini and operated on a non-profit basis.
Copyright for each paper belongs to the authors and their institutions; please make sure to credit the source when sharing.

Test-time Adaptation for Foundation Medical Segmentation Model without Parametric Updates

Created by
  • Haebom

Author

Kecheng Chen, Xinyu Luo, Tiexin Qin, Jie Liu, Hui Liu, Victor Ho Fun Lee, Hong Yan, Haoliang Li

Outline

This paper aims to improve the performance of MedSAM, the most popular base model for medical image segmentation. MedSAM is vulnerable to perturbation by bounding box prompts and has poor performance for certain lesions with complex structures and appearances. Existing test-time adaptation (TTA) methods can solve these problems, but their effectiveness is limited due to the limitation of parameter updates, and their computational complexity is also high. In this paper, we theoretically analyze that image embedding can be directly improved under the MedSAM structure to achieve the same goal as parameter updates, and propose a novel TTA method that improves computational efficiency, segmentation performance, and avoids the forgetting problem. The proposed method combines the distributional approximate latent conditional random field loss and the entropy minimization loss to maximize the factorized conditional probability of the posterior prediction probability. Experimental results show that the Dice score is improved by about 3% on three datasets, while reducing the computational complexity by more than 7 times.

Takeaways, Limitations

Takeaways:
We present a novel TTA method that effectively addresses the performance degradation problem of MedSAM.
Improving computational efficiency and performance simultaneously through direct improvements to image embedding.
Achieve high performance without forgetting problems.
We demonstrate a 3% improvement in Dice score and a 7x reduction in computational complexity across three datasets.
Limitations:
The effectiveness of the proposed method may be limited to the MedSAM structure.
Generalization performance to other medical image segmentation models or different types of lesions requires further study.
Further validation of generalizability is needed due to limitations in the dataset used in the experiment.
👍