Learning active contour models based on self-attention for breast ultrasound image segmentation

作     者：Zhao, Yu Shen, Xiaoyan Chen, Jiadong Qian, Wei Sang, Liang Ma, He 

作者机构：Northeastern Univ Coll Med & Biol Informat Engn Shenyang 110819 Liaoning Peoples R China Dongguan Univ Technol Sch Life & Hlth Technol Dongguan 523808 Guangdong Peoples R China China Med Univ Hosp 1 Dept Ultrasound Shenyang 110002 Liaoning Peoples R China Northeastern Univ Key Lab Intelligent Comp Med Image Minist Educ Shenyang 110819 Liaoning Peoples R China 

出 版 物：《BIOMEDICAL SIGNAL PROCESSING AND CONTROL》 (生物医学信号处理与控制)

年 卷 期：2024年第89卷

核心收录：

学科分类：0831[工学-生物医学工程（可授工学、理学、医学学位）] 10[医学] 

基　　金：Natural Science Founda-tion of Liaoning Province  China [2022-YGJC-52] 

主　　题：Breast ultrasound image segmentation Transformer Loss function Average radial derivative 

摘      要：Computer-aided diagnosis (CAD) systems based on ultrasound have been developed and widely promoted in breast cancer screening. Due to the characteristics of low contrast and speckle noises, breast ultrasound image segmentation, one of the crucial steps of CAD systems, has always been challenging. Recently, the emerging Transformer-based medical segmentation methods, which have a better ability to model long dependencies than convolutional neural networks (CNNs), have shown significant value for medical image segmentation. However, due to the limited data with the high-quality label, Transformer performs weakly on breast ultrasound image segmentation without pretraining. Thus, we propose the Attention-Gate Medical Transformer (AGMT) for small breast ultrasound datasets, which introduces the attention-gate (AG) module to suppress background information and the average radial derivative increment (Delta ARD) loss function to enhance shape information. We evaluate the AGMT on both a private dataset A and a public dataset B. On dataset A, the AGMT outperforms MT on the metrics of true positive ratio, jaccard index (JI) and dice similarity coefficient (DSC) by 6.4%, 2.3% and 1.9%, respectively. Meanwhile, when compared with UNet, the AGMT improves JI and DSC by 5.3% and 4.9%, respectively. The results show performance has significantly improved compared with mainstream models. In addition, we also conduct ablation experiments on the AG module and Delta ARD, which prove their effectiveness.