Integration of prototype generation and contrastive learning for liver tumor segmentation in ultrasound image
10.3969/j.issn.1005-202X.2025.10.008
- VernacularTitle:融合原型生成与对比学习的超声影像肝肿瘤分割
- Author:
Congrui ZHANG
1
;
Xukun ZHANG
;
Minghao HAN
;
Lihua ZHANG
;
Xiaoying WANG
Author Information
1. 上海理工大学健康科学与工程学院,上海 200093
- Publication Type:Journal Article
- Keywords:
ultrasound image;
liver tumor segmentation;
prototype generation;
contrastive learning;
Transformer
- From:
Chinese Journal of Medical Physics
2025;42(10):1321-1327
- CountryChina
- Language:Chinese
-
Abstract:
To tackle the challenges posed by low resolution,weak contrast,and positional variations in liver tumor ultrasound images,which affects the diagnostic efficiency and accuracy,a novel method based on prototype generation and contrastive learning is proposed for liver tumor segmentation in ultrasound images.The core of this method is a weighted mask attention Transformer structure which utilizes the probabilities of real categories in the predicted probability distribution to weight image feature vectors,generates class prototypes with category discrimination,and thereby effectively captures key features while enhancing spatial information representation.By combining contrastive loss with Dice cross-entropy loss,the model achieves significant improvements in both category discrimination capability and segmentation accuracy,and overcomes the limitations of traditional models related to insufficient spatial information and intra-class pixel distribution imbalance.Comprehensive evaluations of the proposed method are conducted on a collected dataset of 253 ultrasound images,and the experimental results reveal that the proposed method attains a mean intersection-over-union of 78.44%and a Dice similarity coefficient of 87.41%,validating its superiority in liver tumor segmentation from ultrasound image.
