基于SAM2实现CT影像上骨骼肌的少样本分割

Few-Shot Segmentation of Skeletal Muscle in CT Images Based on SAM2

摘要: 肌少症对患者的身体健康有诸多不利影响，而肌量的测量是诊断肌少症的重要手段之一。然而，现有的手动测量方法效率较低，限制了其在临床中的广泛应用。文章旨在提出一种适用于肌少症临床诊断和医学研究的自动分割方法，该方法能够在少量训练数据的情况下实现高精度的骨骼肌分割。我们的方法首先使用少量数据对YOLOv10和SAM2模型进行微调。然后，通过YOLOv10获取每个样本的矩形框坐标，并将其作为SAM2的提示词进行骨骼肌分割。实验结果显示，我们的方法在分割任务中的DSC评分达到了0.9025，优于传统U-Net架构的0.8763和使用固定矩形区域作为提示词的0.8887。此外，在测试数据集上的验证结果表明，85%的分割结果在临床应用的可接受范围内。综上所述，文章提出的方法在骨骼肌的少样本自动分割任务中表现出较高的精度和可靠性，具有潜在的临床应用价值。

Abstract: Sarcopenia has numerous adverse effects on patients’ health, and muscle mass measurement is one of the important methods for diagnosing sarcopenia. However, the current manual measurement methods are inefficient, limiting their widespread clinical application. This study aims to propose an automatic segmentation method suitable for both clinical diagnosis and medical research of sarcopenia, which can achieve high-precision skeletal muscle segmentation with a small amount of training data. Our method first fine-tunes the YOLOv10 and SAM2 models using a small amount of data. Then, the rectangular box coordinates of each sample are obtained through YOLOv10 and used as prompts for SAM2 to perform skeletal muscle segmentation. Experimental results show that our method achieves a DSC score of 0.9025 in the segmentation task, outperforming the traditional U-Net architecture’s score of 0.8763 and the fixed rectangular region prompt’s score of 0.8887. Additionally, validation on the test dataset indicates that 85% of the segmentation results fall within the clinically acceptable range. In conclusion, the method proposed in the article demonstrates high accuracy and reliability in the task of few-shot automatic segmentation of skeletal muscle, showing potential clinical application value.