A deep learning strategy for automatic congestive heart failure detection using novel bottleneck attention module

Jibin Wang,Xingtian Guo
DOI: https://doi.org/10.1007/s10489-024-05364-7
IF: 5.3
2024-06-22
Applied Intelligence
Abstract:Congestive heart failure (CHF) is a sort of common cardiovascular disease. Artificial vision inspection on electrocardiogram (ECG) has turned into the mainstream diagnosis strategy for CHF detection, however it is strenuous and challenging. Motivated from bottleneck attention module (BAM), we construct an efficient architecture based on naive bidirectional gate recurrent unit (BGRU) for automatic CHF detection called BAM-BGRU. BAM-BGRU can refine the attention maps from both pathways: Channel attention and Spatial attention to enhance feature representation ability. To evaluate the effectiveness, we leverage naive GRU, BGRU and BAM as control groups with BIDMC congestive heart failure database (BCHFD) and congestive heart failure RR intervals database (CHFRID). The results achieve an obvious performance improvement than control groups and existing algorithms with an accuracy of 99.1% and 98.8%. Besides, we implement visualization analysis for multilevel derivative gradient flows of ECG episodes to strengthen model interpretability. To our knowledge, we first reformulate naive BGRU framework for consistent performance improvements showing promising potentials in diverse CNN backbones.
computer science, artificial intelligence
What problem does this paper attempt to address?