Achieving enhanced diagnostic precision in endometrial lesion analysis through a data enhancement framework

Yi Luo,Meiyi Yang,Xiaoying Liu,Liufeng Qin,Zhengjun Yu,Yunxia Gao,Xia Xu,Guofen Zha,Xuehua Zhu,Gang Chen,Xue Wang,Lulu Cao,Yuwang Zhou,Yun Fang
DOI: https://doi.org/10.3389/fonc.2024.1440881
IF: 4.7
2024-10-16
Frontiers in Oncology
Abstract:Objective: The aim of this study was to enhance the precision of categorization of endometrial lesions in ultrasound images via a data enhancement framework based on deep learning (DL), through addressing diagnostic accuracy challenges, contributing to future research. Materials and methods: Ultrasound image datasets from 734 patients across six hospitals were collected. A data enhancement framework, including image features cleaning and soften label, was devised and validated across multiple DL models, including ResNet50, DenseNet169, DenseNet201, and ViT-B. A hybrid model, integrating convolutional neural network and transformer architectures for optimal performance, to predict lesion types was developed. Results: Implementation of our novel strategies resulted in a substantial enhancement in model accuracy. The ensemble model achieved accuracy and macro-area under the receiver operating characteristic curve values of 0.809 of 0.911, respectively, underscoring the potential for use of DL in endometrial lesion ultrasound image classification. Conclusion: We successfully developed a data enhancement framework to accurately classify endometrial lesions in ultrasound images. Integration of anomaly detection, data cleaning, and soften label strategies enhanced the comprehension of lesion image features by the model, thereby boosting its classification capacity. Our research offers valuable insights for future studies and lays the foundation for creation of more precise diagnostic tools.
oncology
What problem does this paper attempt to address?