Ultrasound thyroid nodule segmentation algorithm based on wavelet transform and CNN-Transformer
10.16016/j.2097-0927.202409145
- VernacularTitle:基于小波变换和CNN-Transformer的超声甲状腺结节分割算法研究
- Author:
Shuijing ZHENG
1
;
Jun YANG
;
Yujiao CAI
;
Jing WEN
Author Information
1. 重庆大学计算机学院模式识别实验室
- Keywords:
thyroid nodule segmentation;
wavelet transform;
ultrasound image diagnosis;
deep learning
- From:
Journal of Army Medical University
2025;47(14):1595-1601
- CountryChina
- Language:Chinese
-
Abstract:
Objective To develop an automatic segmentation network for thyroid nodules by integrating wavelet transform and CNN-Transformer in order to improve the efficiency and accuracy of ultrasound image segmentation.Methods A total of 1 371 sets of ultrasound images of thyroid nodules were collected from Department of Ultrasonography of Second Affiliated Hospital of Army Medical University between May 2023 and February 2024.After preprocessing and normalization,the data were divided into training,validation,and testing sets in a ratio of 8∶1∶1.Based on UNet,CNN and Swin-Transformer were used in parallel as the encoder,with a wavelet transform module inserted between the encoder and decoder to construct a thyroid nodule segmentation network.The performance of the segmentation model was evaluated on the collected internal dataset using accuracy,IoU,and Dice coefficient metrics.Results The finally verified 1 371 sets of ultrasonic thyroid nodules had an average Dice coefficient of 79.63%and an IoU of 67.30%.Compared with UNet,the segmentation accuracy was improved by 1.02%.The segmentation model obtained accurate location and smooth edges of thyroid nodules,and the segmentation was more consistent in thyroid nodule edge and morphology with those marked by doctors manually when compared with other segmentations.Compared with UNet,this segmentation method can learn the texture of nodules more fully and avoid the situation that nodules had been incorrectly divided into surrounding tissues.Conclusion Our developed segmentation model based on wavelet transform and CNN-Transformer demonstrates better segmentation accuracy in comparison to conventional UNet variants,such as UNet,Attention-UNet,and UNetv2,and medical segment anything models like SAM Med2D.This segmentation method enables accurate segmentation of ultrasound thyroid nodules,thereby enhancing clinical workflow efficiency through automated precise delineation.
