Wavelet-Based Frequency Replacement and Edge Enhancement for Semi-Supervised Fetal Ultrasound Image Segmentation.
Authors
Affiliations (4)
Affiliations (4)
- School of Computer Science and Engineering, Wuhan Institute of Technology, Wuhan, China.
- Intelligent Interconnected Systems Laboratory of Anhui Province, Hefei University of Technology, Hefei, China.
- Department of Radiation Oncology, University of Texas Southwestern Medical Center, Dallas, Texas, USA.
- School of Computer and Control Engineering, Yantai University, Yantai, China.
Abstract
Ultrasound image segmentation remains a significant challenge due to inherent low contrast and blurred anatomical boundaries. Fully supervised deep learning approaches require extensive annotated datasets, which are costly and labor-intensive to acquire. This study aims to develop an effective semi-supervised segmentation framework for ultrasound images with limited annotations. We propose a novel semi-supervised segmentation framework tailored for ultrasound images, leveraging frequency component augmentation and edge mask enhancement to promote structural consistency between weakly and strongly augmented inputs. Specifically, discrete wavelet transform (DWT) is used to decompose ultrasound images into low-frequency and high-frequency sub-bands. A high-frequency component replacement strategy is introduced for strongly augmented images, and an edge mask enhancement module is designed to further emphasize anatomical boundaries. Experiments conducted on 3 public fetal ultrasound imaging segmentation datasets-PSFHS, HC18, and CCAUI-demonstrate that our method achieves average Dice similarity coefficients (DSC) of 0.81 and 0.91, respectively, using only 10 annotated images. This represents a 2-3% DSC improvement over existing semi-supervised methods such as FixMatch. Ablation studies confirm the effectiveness of both the high-frequency augmentation and edge enhancement components. The proposed framework offers a promising direction for ultrasound image segmentation in settings with limited annotations, effectively improving segmentation accuracy by combining frequency-domain augmentation and edge-aware enhancement. Code will be available at https://github.com/apple1986/WTEM-SemiSeg.