End-To-End Model for Detecting Murmurs from Phonocardiograms Using Asymmetric Loss

Hussain Alasmawi and Mohammad Yaqub
Mohamed bin Zayed University of Artificial Intelligence


Abstract

Aims: This study aims to detect murmurs from Phonocardiograms using a deep learning approach via a convolutional neural network (CNN) model. Murmurs are abnormal waves caused by turbulent blood flow in the heart and vascular structures. These abnormal waves are closely associated with certain diseases.

Method: The training dataset consists of 942 patients who have 3163 record- ings from the George B. Moody PhysioNet Challenge 2022. In order to detect murmurs, we have combined the recordings belonging to the same patient, and then generated a Mel-Spectrogram. It is then plugged into our custom 4 lay- ers of 2D CNN model to determine whether the murmurs are present, absent, or unknown. because of the high class imbalance (695 with absent murmurs, 179 with present murmurs, and 68 with unknown murmurs), we have used an asymmetric loss function to reduce the model’s bias towards the majority class.

Results: To test our proposed method, we used 5 fold cross validation and computed the AUROC, AUPRC, F1-score and challenge score outlined by the PhysioNet Challenge as the main benchmark. The challenge score is based on the efficiency and performance of the algorithm where the lower score the better. As can be seen in the below table, the asymmetric cross entropy has the similar results in terms of AUROC, AUPRC, and F1-score but there is a huge gap in the challenge score, meaning that this new loss improves the model performance.

Conclusion: In the challenge score, asymmetric loss functions show to improve on the model performance for the challenge score. Furth