Method for fetal ultrasound image classification using pseudo-labelling with PCA-KMeans and an attention-augmented MobileNet-LSTM model

一种基于伪标签、PCA-KMeans 和注意力增强型 MobileNet-LSTM 模型的胎儿超声图像分类方法

阅读:2

Abstract

Accurate classification of fetal ultrasound images is critical for early diagnosis, yet remains challenging due to limited labeled data and high inter-class variability. This study presents a robust deep learning framework that combines a MobileNet backbone with multi-head self-attention and LSTM layers to enhance feature learning and temporal context. To address data scarcity and imbalance, unsupervised clustering was employed using Principal Component Analysis (PCA) for dimensionality reduction and K-means (k=4) for pseudo-label generation. These pseudo-labeled clusters were then balanced using oversampling techniques. The proposed model was trained using transfer learning on the augmented dataset and achieved a test accuracy of approximately 98 % with a macro-F1 score of 0.98, indicating highly reliable classification performance.•Employed PCA (100 components) and K-means (k=4) for effective pseudo-labeling and class balancing.•Designed a hybrid deep learning architecture using MobileNet, multi-head attention, and LSTM.•Achieved ∼98 % test accuracy and 0.98 macro-F1 score, demonstrating strong model generalization.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。