@inproceedings{
anand2023one,
title={One shot localization and segmentation of medical images with Foundation Models},
author={Deepa Anand and Gurunath Reddy and Vanika Singhal and Dattesh D. Shanbhag and Shriram KS and Uday Patil and Chitresh Bhushan and Kavitha Manickam and Dawei Gui and Rakesh Mullick and Avinash Gopal and Parminder Bhatia and Taha Kass-Hout},
booktitle={R0-FoMo:Robustness of Few-shot and Zero-shot Learning in Large Foundation Models},
year={2023},
url={https://openreview.net/forum?id=IaNJC1IRds}
}
Recent advances in Vision Transformers (ViT) and Stable Diffusion (SD) models with their ability to capture rich semantic features of the image have been used for image correspondence tasks on natural images. In this paper, we examine the ability of a variety of pre-trained ViT (DINO, DINOv2, SAM, CLIP) and SD models, trained exclusively on natural images, for solving the correspondence problems on medical images. While many works have made a case for in-domain training, we show that the models trained on natural images can offer good performance on medical images across different modalities (CT,MR,Ultrasound) sourced from various manufacturers, over multiple anatomical regions (brain, thorax, abdomen, extremities), and on wide variety of tasks. Further, we leverage the correspondence with respect to a template image to prompt a Segment Anything (SAM) model to arrive at single shot segmentation, achieving dice range of 62%-90% across tasks, using just one image as reference. We also show that our single-shot method outperforms the recently proposed few-shot segmentation method - UniverSeg (Dice range 47%-80%) on most of the semantic segmentation tasks(six out of seven) across medical imaging modalities.
One shot localization and segmentation of medical images with Foundation Models
Deepa Anand Gurunath Reddy Vanika Singhal Dattesh D. Shanbhag Shriram KS Uday Patil Chitresh Bhushan Kavitha Manickam Dawei Gui Rakesh Mullick Avinash Gopal Parminder Bhatia Taha Kass-Hout