Data Augmentation for a Deep Learning Framework for Ventricular Septal Defect Ultrasound Image Classification

摘要

Congenital heart diseases (CHD) can be detected through ultrasound imaging. Although ultrasound can be used for immediate diagnosis, doctors require considerable time to read dynamic clips; typically, physicians must continuously examine disease data from beating heart images. Most importantly, this type of diagnosis relies heavily on the expertise and experience of the diagnosing physician. This study established an ultrasound image classification with deep learning algorithms to overcome the challenges involved in CHD diagnosis. We detected the most common CHD, namely the first, second, and fourth types of ventricular septal defect (VSD). We improved the performance levels of well-known deep learning algorithms (InceptionV3, ResNet, and DenseNet). Because algorithm optimization and overfitting problems can influence the performance of deep learning algorithms, we studied some optimizer algorithms and early-stopping strategies. To enhance the solution quality, we used data augmentation methods for solving this classification problem. The selected approach was further compared with Google AutoML, which applies structure search for quality prediction. Our results revealed that the proposed deep learning algorithm was able to recognize most types of VSD. However, one type of VSD remains unconquered and warrants more advanced techniques.