arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2311.08190eess.IVcs.CVcs.LG

SAMIHS:基于Segment Anything Model的颅内出血分割自适应方法

SAMIHS: Adaptation of Segment Anything Model for Intracranial Hemorrhage Segmentation

Beihang University(北京航空航天大学)

Yinuo Wang, Kai Chen, Weimin Yuan, Cai Meng, XiangZhi Bai

AI总结:

针对SAM在医学图像分割中性能下降的问题,本文提出参数高效微调方法SAMIHS,通过参数重构适配器和组合损失函数,在颅内出血分割任务上取得有效结果。

摘要:

Segment Anything Model (SAM)是一种在大型标注数据上训练的视觉基础模型,近年来在医学图像分割领域持续引起关注。尽管SAM在自然场景中表现出令人印象深刻的能力,但在面对医学图像时,尤其是涉及模糊边界和高度不规则的低对比度区域时,其性能会下降。本文提出了一种基于SAM的参数高效微调方法,称为SAMIHS,用于颅内出血分割,这是中风诊断和手术规划中关键且具有挑战性的步骤。与之前的SAM和基于SAM的方法不同,SAMIHS将参数重构适配器集成到SAM的图像编码器中,并考虑了适配器参数的高效灵活利用。此外,我们采用了一种组合损失函数,将二元交叉熵损失和边界敏感损失相结合,以增强SAMIHS识别边界区域的能力。我们在两个公共数据集上的实验结果证明了所提方法的有效性。代码可在https://github.com/mileswyn/SAMIHS获取。

英文摘要:

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 binary cross-entropy loss and 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 .

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑