Segment Anything Model (SAM) has achieved significant success in natural image segmentation. However, it struggles with ultrasound images, especially when dealing with individual organs with small samples. To address these challenges, we propose SAMTNU, an adaptive Segment Anything Model for Thyroid and Nodule Ultrasound Image Segmentation. The SAMTNU consists of two primary branches: The SAM branch and the Convolutional Neural Network (CNN) branch. Specifically, we fine-tune the SAM branch’s image encoder using adapters and a Multi-head Attention Low-Rank Adaption module (MHA-LoRA) while keeping other parameters frozen, reducing training costs. To extract more comprehensive features, we propose a Multi-Scale Attention Feature Extraction module (MSAFE) in the CNN branch. Experiments on three datasets show that SAMTNU outperforms existing promising methods. For example, SAMTNU achieves Dice \(\uparrow \) of 83.87% and HD \(\downarrow \) 23.98 on TG3K, surpassing the SAMUS by 7.49% and 2.56, respectively.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

SAMTNU: Adaptive Segment Anything Model for Thyroid and Nodule Ultrasound Image Segmentation

  • Dong Chen,
  • Xiaonan Li,
  • Haibin Ma,
  • Yunrong Zhang,
  • Lei Li

摘要

Segment Anything Model (SAM) has achieved significant success in natural image segmentation. However, it struggles with ultrasound images, especially when dealing with individual organs with small samples. To address these challenges, we propose SAMTNU, an adaptive Segment Anything Model for Thyroid and Nodule Ultrasound Image Segmentation. The SAMTNU consists of two primary branches: The SAM branch and the Convolutional Neural Network (CNN) branch. Specifically, we fine-tune the SAM branch’s image encoder using adapters and a Multi-head Attention Low-Rank Adaption module (MHA-LoRA) while keeping other parameters frozen, reducing training costs. To extract more comprehensive features, we propose a Multi-Scale Attention Feature Extraction module (MSAFE) in the CNN branch. Experiments on three datasets show that SAMTNU outperforms existing promising methods. For example, SAMTNU achieves Dice \(\uparrow \) of 83.87% and HD \(\downarrow \) 23.98 on TG3K, surpassing the SAMUS by 7.49% and 2.56, respectively.