Searching for Effective Neural Network Architectures for Heart Murmur Detection from Phonocardiogram

Hao WEN1 and Jingsu Kang2
1Beihang University, 2Tianjin Medical University


Abstract

Aim: The George B. Moody PhysioNet Challenge 2022 raised problems of heart murmur detection and related abnormal cardiac function identification from phonocardiograms (PCGs). This work describes the novel approaches developed by our team, Revenger, to solve these problems.

Methods: PCGs were resampled to 1000 Hz, then filtered with a Butterworth band-pass filter of order 3, cutoff frequencies 25 - 400 Hz, and z-score normalized. We used the multi-task learning (MTL) method via hard parameter sharing to train one neural network model for all the Challenge tasks. We performed neural architecture searching (NAS) among a set of network backbones, including multi-branch convolutional neural networks (CNNs), SE-ResNets, TResNets, simplified wav2vec2, etc.

Based on a stratified splitting of the subjects, 20% of the public data was left out as a validation set for model selection. The AdamW optimizer was adopted, along with the OneCycle scheduler, to optimize the model weights.

Results: Our murmur detection classifier received a weighted accuracy score of 0.689 (ranked 79th out of 303 entries) and a Challenge cost score of 9471.652 (ranked 21th out of 303 entries) on the hidden validation set.

Conclusion: We provided a practical solution to the problems of detecting heart murmurs and providing clinical diagnosis suggestions from PCGs.