Congenital Heart Disease recognition using Deep Learning/Transformer models
Aidar Amangeldi, Vladislav Yarovenko, Angsar Taigonyrov
TL;DR
Addressing the challenge of non-invasive CHD detection and false negatives in newborn screening, the paper investigates a dual-modality deep-learning framework that leverages audio (heart sounds) and chest X-ray images. It evaluates multiple audio representations (STFT, Mel, GAF) with CNN backbones and applies preprocessing and transformer-based image classifiers on DICOM-derived X-ray data, followed by late-fusion strategies to combine predictions. The strongest results are achieved with STFT + ResNet-50V2 for audio and ResNet-18 for DICOM-XRAY, achieving 71% and 80.72% accuracy, respectively; preprocessing steps boosted DICOM performance by 6-7%. The findings highlight the potential of combining acoustic and imaging modalities for CHD screening, while also underscoring the trade-offs between model complexity, data requirements, and practical deployability in resource-limited settings.
Abstract
Congenital Heart Disease (CHD) remains a leading cause of infant morbidity and mortality, yet non-invasive screening methods often yield false negatives. Deep learning models, with their ability to automatically extract features, can assist doctors in detecting CHD more effectively. In this work, we investigate the use of dual-modality (sound and image) deep learning methods for CHD diagnosis. We achieve 73.9% accuracy on the ZCHSound dataset and 80.72% accuracy on the DICOM Chest X-ray dataset.
