Gradually Vanishing Bridge Based on Multi-Kernel Maximum Mean Discrepancy for Breast Ultrasound Image Classification

• Published in: Journal of advanced computational intelligence and intelligent informatics Vol. 28; no. 4; pp. 835 - 844
• Main Authors: Xu, Bo, Tan, Cuier, Wu, Ying, Li, Faming
• Format: Journal Article
• Language: English
• Published: Tokyo Fuji Technology Press Co. Ltd 01.07.2024
• Subjects: Adaptation; Breast; Datasets; Feature extraction; Generative adversarial networks; Image classification; Image enhancement; Ultrasonic imaging
• ISSN: 1343-0130; 1883-8014
• DOI: 10.20965/jaciii.2024.p0835

This study seeks to enhance the classification performance of breast ultrasound images, addressing the challenges of difficult and costly collection of breast ultrasound datasets as well as the discrepancies in feature distribution of the collected datasets. Performance is enhanced by using a mix of generative adversarial networks (GAN) and domain adaptive networks. First, an adaptive layer is first added to the basic model of the gradually vanishing bridge (GVB), to better match the feature distributions of the source and target domains of the dataset. The multi-kernel maximum mean discrepancy (MK-MMD), which is the most efficient of existing adaptive approaches, is implemented in the fully connected layer of the original model’s feature extraction network. Finally, through the process of fine-tuning, the model that has the highest level of overall performance is determined. In experiments, the proposed method surpassed the conventional unsupervised domain adaptation (DDC) and adversarial domain adaptation (MK_DAAN, GVB) in performance, achieving 85.11% accuracy, 97.48% recall, and 0.92 F1-score.