SAMIHS: Adaptation of Segment Anything Model for Intracranial Hemorrhage Segmentation

  • Yinuo Wang
  • , Kai Chen
  • , Weimin Yuan
  • , Zhouping Tang
  • , Cai Meng*
  • , Xiangzhi Bai*
  • *Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

Segment Anything Model (SAM), a vision foundation model trained on large-scale annotations, has recently continued raising awareness within medical image segmentation. Despite the impressive capabilities of SAM on natural scenes, it struggles with performance decline when confronted with medical images, especially those involving blurry boundaries and highly irregular regions of low contrast. In this paper, a SAM-based parameter-efficient fine-tuning method, called SAMIHS, is proposed for intracranial hemorrhage segmentation, which is a crucial and challenging step in stroke diagnosis and surgical planning. Distinguished from previous SAM and SAM-based methods, SAMIHS incorporates parameter-refactoring adapters into SAM's image encoder and considers the efficient and flexible utilization of adapters' parameters. Additionally, we employ a combo loss that combines the binary cross-entropy loss and a boundary-sensitive loss to enhance SAMIHS's ability to recognize the boundary regions. Our experimental results on two public datasets demonstrate the effectiveness of our proposed method. Code is available at https://github.com/mileswyn/SAMIHS.

Original languageEnglish
Title of host publicationIEEE International Symposium on Biomedical Imaging, ISBI 2024 - Conference Proceedings
PublisherIEEE Computer Society
ISBN (Electronic)9798350313338
DOIs
StatePublished - 2024
Event21st IEEE International Symposium on Biomedical Imaging, ISBI 2024 - Athens, Greece
Duration: 27 May 202430 May 2024

Publication series

NameProceedings - International Symposium on Biomedical Imaging
ISSN (Print)1945-7928
ISSN (Electronic)1945-8452

Conference

Conference21st IEEE International Symposium on Biomedical Imaging, ISBI 2024
Country/TerritoryGreece
CityAthens
Period27/05/2430/05/24

Keywords

  • CT
  • Foundation models
  • Intracranial hemorrhage segmentation
  • Medical image segmentation

Fingerprint

Dive into the research topics of 'SAMIHS: Adaptation of Segment Anything Model for Intracranial Hemorrhage Segmentation'. Together they form a unique fingerprint.

Cite this