Department of Computer Science and Engineering, BRAC University, Dhaka, Bangladesh.
Department of Embedded Systems Engineering, Incheon National University, Incheon, Republic of Korea.
Ultrasonics. 2023 Jul;132:107017. doi: 10.1016/j.ultras.2023.107017. Epub 2023 Apr 22.
Ultrasound imaging is a valuable tool for assessing the development of the fetal during pregnancy. However, interpreting ultrasound images manually can be time-consuming and subject to variability. Automated image categorization using machine learning algorithms can streamline the interpretation process by identifying stages of fetal development present in ultrasound images. In particular, deep learning architectures have shown promise in medical image analysis, enabling accurate automated diagnosis. The objective of this research is to identify fetal planes from ultrasound images with higher precision. To achieve this, we trained several convolutional neural network (CNN) architectures on a dataset of 12400 images. Our study focuses on the impact of enhanced image quality by adopting Histogram Equalization and Fuzzy Logic-based contrast enhancement on fetal plane detection using the Evidential Dempster-Shafer Based CNN Architecture, PReLU-Net, SqueezeNET, and Swin Transformer. The results of each classifier were noteworthy, with PreLUNet achieving an accuracy of 91.03%, SqueezeNET reaching 91.03% accuracy, Swin Transformer reaching an accuracy of 88.90%, and the Evidential classifier achieving an accuracy of 83.54%. We evaluated the results in terms of both training and testing accuracies. Additionally, we used LIME and GradCam to examine the decision-making process of the classifiers, providing explainability for their outputs. Our findings demonstrate the potential for automated image categorization in large-scale retrospective assessments of fetal development using ultrasound imaging.
超声成像是评估胎儿在妊娠期间发育情况的一种有价值的工具。然而,手动解释超声图像既费时又容易出现变化。使用机器学习算法进行自动图像分类可以通过识别超声图像中存在的胎儿发育阶段来简化解释过程。特别是,深度学习架构在医学图像分析中显示出了很大的潜力,能够实现准确的自动诊断。本研究的目的是提高从超声图像中识别胎儿平面的精度。为此,我们在一个包含 12400 张图像的数据集上训练了几种卷积神经网络 (CNN) 架构。我们的研究侧重于通过采用直方图均衡化和基于模糊逻辑的对比度增强来提高图像质量,然后使用基于证据的 Dempster-Shafer 卷积神经网络架构、PReLU-Net、SqueezeNET 和 Swin Transformer 来检测胎儿平面,从而探讨图像质量增强对胎儿平面检测的影响。每个分类器的结果都很显著,其中 PreLUNet 的准确率达到 91.03%,SqueezeNET 的准确率达到 91.03%,Swin Transformer 的准确率达到 88.90%,而基于证据的分类器的准确率达到 83.54%。我们从训练和测试精度两个方面评估了结果。此外,我们还使用 LIME 和 GradCam 来检查分类器的决策过程,为其输出提供可解释性。我们的研究结果表明,使用超声成像进行大规模回顾性胎儿发育评估,自动图像分类具有很大的潜力。