Heart Murmur Classification Using a Capsule Neural Network

The healthcare industry has made significant progress in the diagnosis of heart conditions due to the use of intelligent detection systems such as electrocardiograms, cardiac ultrasounds, and abnormal sound diagnostics that use artificial intelligence (AI) technology, such as convolutional neural ne...

Full description

Bibliographic Details
Main Authors: Yu-Ting Tsai, Yu-Hsuan Liu, Zi-Wei Zheng, Chih-Cheng Chen, Ming-Chih Lin
Format: Article
Language:English
Published: MDPI AG 2023-10-01
Series:Bioengineering
Subjects:
Online Access:https://www.mdpi.com/2306-5354/10/11/1237
_version_ 1797460132514234368
author Yu-Ting Tsai
Yu-Hsuan Liu
Zi-Wei Zheng
Chih-Cheng Chen
Ming-Chih Lin
author_facet Yu-Ting Tsai
Yu-Hsuan Liu
Zi-Wei Zheng
Chih-Cheng Chen
Ming-Chih Lin
author_sort Yu-Ting Tsai
collection DOAJ
description The healthcare industry has made significant progress in the diagnosis of heart conditions due to the use of intelligent detection systems such as electrocardiograms, cardiac ultrasounds, and abnormal sound diagnostics that use artificial intelligence (AI) technology, such as convolutional neural networks (CNNs). Over the past few decades, methods for automated segmentation and classification of heart sounds have been widely studied. In many cases, both experimental and clinical data require electrocardiography (ECG)-labeled phonocardiograms (PCGs) or several feature extraction techniques from the mel-scale frequency cepstral coefficient (MFCC) spectrum of heart sounds to achieve better identification results with AI methods. Without good feature extraction techniques, the CNN may face challenges in classifying the MFCC spectrum of heart sounds. To overcome these limitations, we propose a capsule neural network (CapsNet), which can utilize iterative dynamic routing methods to obtain good combinations for layers in the translational equivariance of MFCC spectrum features, thereby improving the prediction accuracy of heart murmur classification. The 2016 PhysioNet heart sound database was used for training and validating the prediction performance of CapsNet and other CNNs. Then, we collected our own dataset of clinical auscultation scenarios for fine-tuning hyperparameters and testing results. CapsNet demonstrated its feasibility by achieving validation accuracies of 90.29% and 91.67% on the test dataset.
first_indexed 2024-03-09T17:01:41Z
format Article
id doaj.art-849930bcfb32494cb49e964dc71b6d76
institution Directory Open Access Journal
issn 2306-5354
language English
last_indexed 2024-03-09T17:01:41Z
publishDate 2023-10-01
publisher MDPI AG
record_format Article
series Bioengineering
spelling doaj.art-849930bcfb32494cb49e964dc71b6d762023-11-24T14:29:34ZengMDPI AGBioengineering2306-53542023-10-011011123710.3390/bioengineering10111237Heart Murmur Classification Using a Capsule Neural NetworkYu-Ting Tsai0Yu-Hsuan Liu1Zi-Wei Zheng2Chih-Cheng Chen3Ming-Chih Lin4Master’s Program in Electro-Acoustics, Feng Chia University, Taichung 40724, TaiwanMaster’s Program in Electro-Acoustics, Feng Chia University, Taichung 40724, TaiwanHyper-Automation Laboratory, Feng Chia University, Taichung 40724, TaiwanHyper-Automation Laboratory, Feng Chia University, Taichung 40724, TaiwanDepartment of Post-Baccalaureate Medicine, College of Medicine, National Chung Hsing University, Taichung 402, TaiwanThe healthcare industry has made significant progress in the diagnosis of heart conditions due to the use of intelligent detection systems such as electrocardiograms, cardiac ultrasounds, and abnormal sound diagnostics that use artificial intelligence (AI) technology, such as convolutional neural networks (CNNs). Over the past few decades, methods for automated segmentation and classification of heart sounds have been widely studied. In many cases, both experimental and clinical data require electrocardiography (ECG)-labeled phonocardiograms (PCGs) or several feature extraction techniques from the mel-scale frequency cepstral coefficient (MFCC) spectrum of heart sounds to achieve better identification results with AI methods. Without good feature extraction techniques, the CNN may face challenges in classifying the MFCC spectrum of heart sounds. To overcome these limitations, we propose a capsule neural network (CapsNet), which can utilize iterative dynamic routing methods to obtain good combinations for layers in the translational equivariance of MFCC spectrum features, thereby improving the prediction accuracy of heart murmur classification. The 2016 PhysioNet heart sound database was used for training and validating the prediction performance of CapsNet and other CNNs. Then, we collected our own dataset of clinical auscultation scenarios for fine-tuning hyperparameters and testing results. CapsNet demonstrated its feasibility by achieving validation accuracies of 90.29% and 91.67% on the test dataset.https://www.mdpi.com/2306-5354/10/11/1237heart murmurauscultationheart sound diagnosiscardiac dysphonic diagnosiscapsule neural networkdeep learning in healthcare
spellingShingle Yu-Ting Tsai
Yu-Hsuan Liu
Zi-Wei Zheng
Chih-Cheng Chen
Ming-Chih Lin
Heart Murmur Classification Using a Capsule Neural Network
Bioengineering
heart murmur
auscultation
heart sound diagnosis
cardiac dysphonic diagnosis
capsule neural network
deep learning in healthcare
title Heart Murmur Classification Using a Capsule Neural Network
title_full Heart Murmur Classification Using a Capsule Neural Network
title_fullStr Heart Murmur Classification Using a Capsule Neural Network
title_full_unstemmed Heart Murmur Classification Using a Capsule Neural Network
title_short Heart Murmur Classification Using a Capsule Neural Network
title_sort heart murmur classification using a capsule neural network
topic heart murmur
auscultation
heart sound diagnosis
cardiac dysphonic diagnosis
capsule neural network
deep learning in healthcare
url https://www.mdpi.com/2306-5354/10/11/1237
work_keys_str_mv AT yutingtsai heartmurmurclassificationusingacapsuleneuralnetwork
AT yuhsuanliu heartmurmurclassificationusingacapsuleneuralnetwork
AT ziweizheng heartmurmurclassificationusingacapsuleneuralnetwork
AT chihchengchen heartmurmurclassificationusingacapsuleneuralnetwork
AT mingchihlin heartmurmurclassificationusingacapsuleneuralnetwork