Application of Dense Neural Networks for Detection of Atrial Fibrillation and Ranking of Augmented ECG Feature Set
<p>General DenseNet architecture optimized for rhythm classification {N,AF,O,X} with 137 input ECG features, batch normalization layer and configurable dense layers (depth, size): {number dense layers} = [1–4], number nodes@N{1–3} = [0–1024] (red dots). W{1–4} stand for all trainable dense layer weight matrices.</p> "> Figure 2
<p>Illustration of the process for computation of <math display="inline"><semantics> <mrow> <mi>C</mi> <mi>a</mi> <mi>s</mi> <mi>e</mi> <mo> </mo> <mi>W</mi> <mi>e</mi> <mi>i</mi> <mi>g</mi> <mi>h</mi> <mi>t</mi> <mi>s</mi> <mo> </mo> <mi>I</mi> <mi>m</mi> <mi>p</mi> <mi>o</mi> <mi>r</mi> <mi>t</mi> <mi>a</mi> <mi>n</mi> <mi>c</mi> <mi>e</mi> </mrow> </semantics></math> and <math display="inline"><semantics> <mrow> <mi>G</mi> <mi>l</mi> <mi>o</mi> <mi>b</mi> <mi>a</mi> <mi>l</mi> <mo> </mo> <mi>W</mi> <mi>e</mi> <mi>i</mi> <mi>g</mi> <mi>h</mi> <mi>t</mi> <mi>s</mi> <mo> </mo> <mi>I</mi> <mi>m</mi> <mi>p</mi> <mi>o</mi> <mi>r</mi> <mi>t</mi> <mi>a</mi> <mi>n</mi> <mi>c</mi> <mi>e</mi> </mrow> </semantics></math> based on an example neurons weights matrix <math display="inline"><semantics> <mrow> <mi>Wi</mi> <mo>∈</mo> <msup> <mo>ℝ</mo> <mrow> <mfenced close="]" open="["> <mrow> <mn>1</mn> <mo>×</mo> <mn>40</mn> </mrow> </mfenced> </mrow> </msup> </mrow> </semantics></math> and neurons activation state matrix <math display="inline"><semantics> <mrow> <mrow> <mi mathvariant="sans-serif">α</mi> <mi mathvariant="normal">i</mi> </mrow> <mfenced> <mrow> <msub> <mi mathvariant="normal">X</mi> <mi mathvariant="normal">k</mi> </msub> </mrow> </mfenced> <mo>∈</mo> <msup> <mo>ℝ</mo> <mrow> <mfenced close="]" open="["> <mrow> <mn>11</mn> <mo>×</mo> <mn>40</mn> </mrow> </mfenced> </mrow> </msup> </mrow> </semantics></math>.</p> "> Figure 3
<p>Performance of all DenseNet architectures in the grid search, grouped to the number of dense layers: (<b>a</b>) F1 (Total) score observations in respect to the number of trainable parameters; (<b>b</b>) Statistical distributions of F1 (Total) score, showing the median value (line), interquartile range 25–75% (boxes), non-outlier range (whiskers) and individual observations (markers).</p> "> Figure 4
<p>Performance of all NN architectures in the grid search with two (DenseNet-2), three (DenseNet-3) and four (DenseNet-4) dense layers in function of the number of nodes of the last hidden layer.</p> "> Figure 5
<p>Influence of rhythm class weights W (Normal, AF and Other rhythms) on DenseNet-3@128-32-4 performance. The contour plots of F1-score measured within W grid [0; 0.5] justify the choice of the weights as the average values in the mid-area of maximal performance (+mark): W(Normal) = 0.23, W(AF) = 0.25, W(Other) = 0.3.</p> "> Figure 6
<p>DenseNet-3 performance dependency on batch size (<b>a</b>) and learning rate (<b>b</b>).</p> "> Figure 7
<p><span class="html-italic">Global Weights Importance</span> presented for 137 input features and four rhythm classes. The latter are used to construct stacked barplots, which present the mean absolute values computed by 20 × 5-fold independent runs of DenseNet-3@128-32-4 model. The values are reported without normalization.</p> "> Figure 8
<p>Performance evaluation in case of feature reduction based on <span class="html-italic">Global Weights Importance</span> rank for: (<b>a</b>) Different architectures of DenseNet; (<b>b</b>) DenseNet-3@128-32-4 performance for different rhythms.</p> "> Figure 9
<p>Feature rank given to 137 input features in descending order of their <span class="html-italic">Relative Feature Importance</span> to detect specific rhythm class. The dots and whiskers represent the mean absolute values and standard deviations of the <span class="html-italic">Relative Feature Importance</span> computed by 20 × 5-fold independent runs of DenseNet-3@128-32-4 model.</p> "> Figure 10
<p>Visual representation of signed <span class="html-italic">Relative Feature Importance</span> for the top-20 features ranked in <a href="#sensors-21-06848-t006" class="html-table">Table 6</a> to maximally contribute to the detection of each of the four rhythm classes. The bars and whiskers represent the mean values and standard deviations of the <span class="html-italic">Relative Feature Importance</span> computed by 20 × 5-fold independent runs of DenseNet-3@128-32-4 model. Signed values of <span class="html-italic">Relative Feature Importance</span> are normalized in the range [−1; 1]. They comprehensively indicate the value of the feature that would contribute the positive detection of specific class, i.e., positive sign specifies high feature value and negative sign specifies low feature value.</p> "> Figure 11
<p>ECG example (file A00060) of correctly classified Normal rhythm. The single-lead ECG strip is representative to sinus rhythm corrupted with baseline and high-frequency artifacts due to poor electrode contact or/and patient’s movements. No discernible repeating P-waves are present, and RR intervals seem ‘irregularly irregular’. The barplots represent the <span class="html-italic">Case Feature Importance</span> of the top-10 ranked features, which maximally contribute to the output classification probability (<span class="html-italic">p</span>) of four rhythms. The ECG strip is definitively detected as Normal rhythm (P = 0.955) based on HRV features, P-wave presence and QRS-width by DenseNet-3@128-32-4 model.</p> "> Figure 12
<p>ECG example (file A01233) of erroneously classified Normal rhythm as AF. The single-lead ECG strip is representative to peak artifacts and significant variation of RR intervals in some parts of the record. Visual diagnosis of other dysrhythmias such as atrial flutter and multifocal atrial tachycardia cannot also be safely excluded. The barplots represent the <span class="html-italic">Case Feature Importance</span> of the top-10 ranked features, which maximally contribute to the output classification probability (P) of four rhythms. The ECG strip is definitively detected as AF (<span class="html-italic">p</span> = 0.847) based on HRV and atrial activity estimation, although Other rhythm (P = 0.138) is also weakly indicated by DenseNet-3@128-32-4 model.</p> "> Figure 13
<p>ECG example (file A00005) of correctly classified AF. The single-lead ECG strip is representative of non-sinus, narrow-complex rhythm with high-rate ventricular response and seemingly regular RR intervals in parts. This example presents a certain problem for visual differentiation between high-rate AF, atrial flutter (or alternation of AF and flutter) and AV-nodal re-entrant tachycardia. The barplots represent the <span class="html-italic">Case Feature Importance</span> of the top-10 ranked features, which maximally contribute to the output classification probability (P) of four rhythms. The ECG strip is detected as AF with very high probability (P = 0.784) based on HRV and atrial activity estimation, although Other rhythm (P = 0.216) based on six HRV features is also weakly indicated by DenseNet-3@128-32-4 model.</p> "> Figure 14
<p>ECG example (file A00015) of erroneously classified AF as Noise. The single-lead ECG strip is representative to significant deflections of the isoelectric line of ‘artifact-type’; not obvious RR irregularity; not clearly seen f-waves of AF. The visual ECG interpretation cannot safely differentiate between sinus rhythm (±supra-ventricular extrasystoles) and AF. The barplots represent the <span class="html-italic">Case Feature Importance</span> of the top-10 ranked features, which maximally contribute to the output classification probability (<span class="html-italic">p</span>) of four rhythms. The ECG strip is detected as Noise (P = 0.513) due to three top-ranked features representative to large QRS morphology variation. DenseNet-3@128-32-4 model decides with almost equal uncertainty about Other rhythm (P = 0.228), AF (P = 0.133) and Normal rhythm (P = 0.127).</p> "> Figure 15
<p>ECG example (file A00114) of correctly classified Other rhythm. The single-lead ECG strip is representative to sinus rhythm with alternating ventricular tri-, quadri- and bi-geminy and a widened baseline QRS as in fixed bundle-branch block. Some of the baseline wide QRS are changed additionally due to artifacts, and they might mislead ventricular ectopy. The barplots represent the <span class="html-italic">Case Feature Importance</span> of the top-10 ranked features, which maximally contributed to the output classification probability (P) in four rhythms. The ECG strip is definitively detected as Other rhythm (P = 0.974) by DenseNet-3@128-32-4 model based on HRV and ECG complexity features.</p> "> Figure 16
<p>ECG example (file A03727) of erroneously classified Other rhythm as AF. The single-lead ECG strip is representative to sinus rhythm with frequent supra-ventricular extrasystoles and varying RR intervals (due to post-extrasystole compensatory pause); and high-frequency artifacts of the isoelectric line, mimicking f-waves of AF. The barplots represent the <span class="html-italic">Case Feature Importance</span> of the top-10 ranked features, which maximally contributed to the output classification probability (P) in four rhythms. The ECG strip is detected as Other rhythm (P = 0.471) and AF (P = 0.526) with almost equal uncertainty by DenseNet-3@128-32-4 model based on HRV and atrial activity features.</p> ">
Abstract
:1. Introduction
- Support vector machine (SVM) classifiers are input with 47 features from the statistical and morphological rhythm representation [24]; 33 features expressive to the signal power, spectrum, entropy, RR intervals and P-waves [25]; and 61 features from the time-frequency ECG representation, both average and variability of RR intervals, and the average beat morphology [26];
- Linear and quadratic discriminant classifiers are input with a set of 122 RR-interval features from their time domain, frequency domain and distribution (histogram) representations [27] and 44 features measured by heart rate variability (HRV) analysis, average beat morphology analysis, and atrial activity analysis focused on the P-wave amplitude in the average beat and f-waves amplitude in TQ intervals [28];
- Decision tree classifiers are input with 30 multi-level features, including measures of AF, morphology, RR intervals and similarity between beats [29]; morphological coefficients and HRV features calculated by ECG waveform fitting with a piecewise linear spline [30]; 400 hand-crafted features, reflecting the complex physiology of cardiac arrhythmias visible in single-channel ECG [31]; and 74 features of the R-peak amplitude, RR-interval statistics, PQRST statistics, ECG signal irregularity, entropy, noise content and four additional sparse coding features [32];
- A multi-layer binary classification architecture is input with subsets of 77, 66 and 19 features selected from 188 dimensional feature pool containing time, frequency, morphological and statistical domain ECG features [33];
- A multi-stage classifier, combining SVM, decision tree and threshold-based rules is quantifying both atrial and ventricular activity, estimated by local features (beat classification) and global features (rhythm, signal quality, similarity) [34].
- Advanced multi-stage classifiers, combining decision trees and neural networks (NNs) include: a tree gradient boosting model and a recurrent long, short-term memory (LSTM) network as a global classifier that uses 42 summary ECG features of the full record and a sequence classifier that works on a beat-by-beat basis using individual features of each cardiac cycle [35]; a bagged tree ensemble with 43 input features based on QRS detection and PQRS morphology connected in parallel to a convolutional neural network (CNN) and a shallow NN for analysis of raw filtered ECG signals (8× envelograms, 1× band-pass) [36]; a nine-layer CNN for segmentation of P, QRS and T waves, inter-beat segments, noise and arrhythmic beats, additionally augmented by hand-crafted features, thus supplying a set of 181 features to eXtreme Gradient Boosting trees to classify the heart rhythm [37]; a densely connected CNN applied on time–frequency ECG spectrograms (9 s, 15 s) and subsequent refined classification by AdaBoost-abstain classifier of 437 features, including signal quality, frequency content, RR-interval, ECG-based reconstructed phase space and Poincaré plots [38]; and ENCASE as an ensemble of multiple gradient-boosting decision trees with 590 input features, including deep features extracted with a deep neural network (DNN) from raw ECG and engineering features (statistical, signal, morphological and unsupervised) [39];
- DNN classifiers, including a quadratic NN input with a set of 122 RR-interval features from their time domain, frequency domain and distribution (histogram) representations [27]; a 21-layer 1D convolutional recurrent neural network (RhythmNet) containing 16 CNN layers for raw ECG feature extraction followed by three recurrent layers for processing of ECG records with varying lengths [40]; and a combination of one 1D CNN layer and a sequence of three LSTM layers for raw ECG feature extraction and classification of arrhythmia [41].
2. Materials and Methods
2.1. ECG Database
2.2. ECG Features
- Noise detection—1 feature;
- HRV analysis—21 features;
- Average beat morphology analysis—25 features;
- Heartbeat classification and rhythm analysis—19 features;
- Principal component analysis of PQRST and TQ-intervals—5 features;
- P-wave analysis (time domain)—12 features;
- TQ-segment analysis (time domain)—10 features;
- TQ-segment analysis (frequency domain)—44 features.
2.3. Dense Neural Network Classifier
- Input layer with 137 nodes for fusion of all ECG features in Table 2. Raw feature measurements are retained in original scales and native units without applying initial control or preselection mechanisms.
- Batch normalization (BN) layer, employed as a renowned regularization technique that is known to accelerate training [56]. In our model, BN is applied for standardization of the input feature by removing the mean and scaling to unit variance for each mini-batch. BN transform layer computes two trainable parameters (γ,β) for each input feature x.
- A sequence of hidden dense (fully connected) layers for ECG feature fusion and multi-level abstraction of feature maps [57]. Considering a maximum number of three hidden dense layers, the output feature map (y3) of the input (x) can be computed as [58]:
- Output classification dense layer with number of nodes corresponding to the defined four target classes C = {N,AF,O,X} linked to SoftMax activation function (σ). It derives the probability that the input (x) transferred to the output of the third hidden dense layer above (y3) belongs to the jth class (j = 1–4):
2.4. Performance Evaluation and Training
- Global Challenge F1 score computed as an average value of the specific F1-scores for three rhythm classes , where the Noise class is skipped for averaging due to its small prevalence in 3.3% of the database [22]:
- Precision, Recall and Accuracy for all rhythm classes :
3. Results
3.1. Neural Network Optimization
3.1.1. Neural Architecture Search
- D = one to four dense layers, including the output dense layer.
- The number of neurons in a given hidden dense layer cannot be larger than the number of neurons in the previous hidden dense layer;
- The number of neurons in a given hidden dense layer can be any in the list {0, 1, 2, 4, 8, 16, 32, 64, 128, 512, 1024}.
- DenseNet-1@4 (Params = 552): F1-Total = 0.7708;
- DenseNet-2@256-4 (Params = 36,352): F1-Total = 0.7985;
- DenseNet-3@128-32-4 (Params = 21,888): F1-Total = 0.7997;
- DenseNet-4@1024-1024-16-4 (Params = 1,206,336): F1-Total = 0.7998.
3.1.2. Class Weights
3.1.3. Batch Size and Learning Rate
3.1.4. Optimal DenseNet Performance
3.2. Feature Importance
3.2.1. Global Weights Importance
3.2.2. Performance of Reduced Feature Sets Based on Global Weights Importance
- DenseNet-3@128-32-4: The validated architecture for 137 features. We consider that it could be redundant for less features; therefore, some reduced nets are also further defined in (2) and (3);
- DenseNet-3@-(nF x 2)-(nF/2)-4: The first hidden layer is two times the number of input features (nF), and the second hidden layer is four times shorter than the first one. This network can be considered as one two times redundant for input nF;
- DenseNet-3@(nF)-(nF/4)-4: The first hidden layer is equal to the number of input features (nF), and the second hidden layer is four times shorter than the first one. This network can be considered as one shrunk to input nF.
3.2.3. Relative Feature Importance
3.2.4. Case Feature Importance
4. Discussion
4.1. Feature Importance
- 1.
- AF classification is found to mostly rely on:
- Positive (‘corBeat(mean)’, ‘corBeat(50%)’, ‘corBeat(25%)’): Estimates high correlation of the morphology of the average beat vs. all beats, which keep normal ventricular depolarization;
- Positive (‘ratePP’), negative (‘MeanPPint’): Estimates high-rate oscillations of the atrial activity, which are discernible during rapid atrial fibrillation f-waves;
- Positive (‘Median_TQamp’): Estimates enhanced deflections of the iso-electric TQ intervals, which are discernible during high-amplitude atrial fibrillation f-waves;
- Negative (‘P-amp’, ‘maxc(P)’): Estimates low-amplitude P-waves in the average beat due to uncoordinated atrial depolarization during AF;
- Positive (‘PNN50’, ‘PNN50%’, ‘AF%’): Estimates enhanced HRV due to irregularly irregular (i.e., totally irregular) ventricular rate during AF.
- 2.
- Normal rhythm classification is found to mostly rely on:
- Positive (‘P-wave presence’), negative (‘maxc(QRS/P)’): Highlights well-discernible P-wave present in the average beat, considering heart rhythm controlled by the sinus node and each QRS preceded by a normal P-wave without conduction abnormalities;
- Negative (‘PNN50’, ‘dRRmean%’, ‘dRRmedian%’): Estimates low HRV due to regular ventricular rate, which naturally displays slight beat-to-beat variability controlled by the sympathetic and parasympathetic balance of the autonomic nervous system;
- Negative (‘Noise correction’): Noise not detected.
- 3.
- Other rhythm classification is found to mostly rely on:
- Positive (‘corBeat(50%)’, ‘corBeat(mean)’, ‘corBeat(25%)’): Estimates high-correlation of the morphology of the average beat vs. 25–50% proportion of all beats, representative to small variance of beat morphology of the sustained rhythm despite that it might include beats with abnormal ventricular depolarization or occasional ectopic beats;
- Negative (‘PNN50%’, ‘PNN50’): Estimates low HRV determined by the relatively regular ventricular rate of the sustained rhythm;
- Positive (‘SD1/SD2’) and negative (‘corRR’): Estimates enhanced HRV non-linearity of successive RR-intervals associated with shortened pre-ectopic and prolonged post-ectopic compensatory pause of occasional atrial and ventricular extrasystoles, which are commonly present in Other rhythms;
- Negative (‘MinRRVB’): Outlines very short minimal RR-intervals associated with tachyarrhythmias or short-coupled ventricular ectopic beats (e.g., R-on-T phenomenon);
- Negative (‘Noise correction’): Noise not detected;
- Negative (‘maxc(QRS/P)’): Finds low-slope QRS morphology associated with abnormal ventricular depolarization;
- Negative (‘MeanStdPCA(TQ)’): Estimates sustained atrial activity during TQ-intervals indicative to regular atrial depolarization preceding QRS;
- Positive (‘PQ-int’): Finds prolonged PQ-intervals, which are mainly due to slowing of conduction between the atria and ventricles, e.g., discernible in 1st degree AV blocks belonging to the group of Other rhythms.
- 4.
- Noise classification is found to mostly rely on:
- Negative (‘corBeat(mean)’, ‘corBeat(50%)’, ‘corBeat(25%)’): Estimates low correlation of the morphology of the average beat vs. all beats, due to arbitrary noise waveform contamination of most beats.
- Positive (‘Fragmentation’): Estimates QRS morphology alteration due to prominent noise impact;
- Negative (‘PNN50%’, ‘PNN50’, ‘CCM’): Estimates low HRV determined by the relatively regular ventricular rate of the sustained rhythm;
- Positive (‘Complexity_ECG’): Outlines high ECG complexity due to random noise impact on the ECG waveform;
- Negative (‘Median_TQamp’): Estimates low deflections of the isoelectric TQ intervals in normal or other baseline rhythm with regular atrial depolarization preceding QRS for the most part of signals in the noise group;
- Positive (‘MeanAmpP’, ‘P-wave presence’): Highlights well-discernible P-wave amplitude and P-wave presence in the average beat, considering a baseline rhythm with regular atrial depolarization preceding QRS.
- Average beat morphology analysis with a focus on the P-wave estimation by amplitude (‘P-wave presence’) and curvature (‘maxc(P), ‘maxc(QRS/P)’);
- Average beat morphology analysis with a focus on its correlation to all beat waveforms—‘corBeat(50%)’, ‘corBeat(mean)’, ‘corBeat(25%)’;
- HRV analysis with a focus on the RR–Tachogram (‘PNN50’, ‘PNN50%’), Poincaré Plot (‘SD1/SD2’, ‘corRR’) and dRR–Tachogram (‘dRRmedian%’, ‘dRRmean%’, ‘dRRmedian)’;
- Noise detection based on QRS detector failure—’Noise correction’;
- Average beat morphology analysis with a focus on the PQRST delineation—’T-amp’;
- TQ-segment analysis with a focus on the atrial activity estimation by amplitude (‘Median_TQamp’) and sustainability (‘MeanStd_PCA_TQ’);
- Heartbeat classification and rhythm analysis with a focus on the proportion of the detected normal beats (‘NBeats%’) and related RR intervals (‘MeanRRN’, ‘MinRRVB’).
4.2. Neural Network Optimization
- Number of dense layers (Figure 3): In the scanned range from one to four layers, several models with two, three and four dense layers and trainable parameters in the range [104; 106] are found to reach similar top performances with F1 (Total) > 0.798. Nevertheless, the one with maximal performance and minimal number of trainable parameters (21,888) is found with three dense layers;
- Number of neurons in hidden layers (Figure 4): In the scanned range from 0 to 1024 nodes per layer, the number of nodes in the last hidden layer, corresponding to the feature space into the output classification layer, has been found to be the one that most strongly affects the performance. It could be deduced that the minimal number of nodes in the last hidden layer should be ≥256 (DenseNet-2), ≥32 (DenseNet-3) and ≥16 (DenseNet-4) in order to reach maximal performance. However, redundant nodes in the last hidden layer are not improving but are more likely to decrease the performance (seen in DenseNet-3 and DenseNet-4).
- Class Weights (Figure 5): In the scanned range from 0 to 0.5, we found large deviation of performance F1 (Total) = 0.62–0.79, locating the optimal zone at a mean of: W(N) = 0.23, W(AF) = 0.25, W(O) = 0.3 and W(X) = 0.22. These optimal class weights are quite balanced compared to the prevalence of classes within the database (Table 1): 60% (N), 9% (AF), 28% (O) and 3% (X). This result suggests that careful optimization of the class weights is highly demandable in imbalanced datasets, considering that the common scenarios are using a default penalty proportional to the class prevalence.
- Batch Size (Figure 6a): In the scanned range from 32 to 1024, a noticeable performance maximum for a batch size of 256 is chosen as an optimal value.
- Learning Rate (Figure 6b): In the scanned range from 0.00001 to 0.1, the default learning rate of 0.001 is chosen as an optimal value within the top-performance zone.
4.3. Comparative Performance Evaluation
5. Conclusions
Author Contributions
Funding
Institutional Review Board Statement
Informed Consent Statement
Data Availability Statement
Conflicts of Interest
References
- Hindricks, G.; Potpara, T.; Dagres, N.; Arbelo, E.; Bax, J.; Blomstrom-Lundqvist, C.; Boriani, G.; Castella, M.; Dan, G.-A.; Dilaveris, P.E.; et al. 2020 ESC Guidelines for the diagnosis and management of atrial fibrillation developed in collaboration with the European Association of Cardio-Thoracic Surgery (EACTS). Eur. Heart J. 2020, 42, 373–498. [Google Scholar] [CrossRef]
- Tiver, K.; Quah, J.; Lahiri, A.; Ganesan, A.; McGavigan, A. Atrial fibrillation burden: An update—the need for a CHA2DS2-VASc-AFBurden score. EP Eur. 2021, 23, 665–673. [Google Scholar] [CrossRef]
- Naydenov, S.; Runev, N.; Manov, E. Are Three Weeks of Oral Anticoagulation Sufficient for Safe Cardioversion in Atrial Fibrillation? Medicina 2021, 57, 554. [Google Scholar] [CrossRef] [PubMed]
- Lip, G.Y.; Lane, D.A. Stroke prevention in atrial fibrillation: A systematic review. JAMA 2015, 313, 1950–1962. [Google Scholar] [CrossRef] [PubMed]
- Steinberg, J.S.; O’Connell, H.; Li, S.; Ziegler, P.D. Thirty-second gold standard definition of atrial fibrillation and its relationship with subsequent arrhythmia patterns: Analysis of a large prospective device database. Circ. Arrhythm Electrophysiol. 2018, 11, e006274. [Google Scholar] [CrossRef]
- Ballatore, A.; Matta, M.; Saglietto, A.; Desalvo, P.; Bocchino, P.P.; Gaita, F.; De Ferrari, G.M.; Anselmino, M. Subclinical and Asymptomatic Atrial Fibrillation: Current Evidence and Unsolved Questions in Clinical Practice. Medicina 2019, 55, 497. [Google Scholar] [CrossRef] [Green Version]
- Rosero, S.Z.; Kutyifa, V.; Olshansky, B.; Zareba, W. Ambulatory ECG Monitoring in Atrial Fibrillation Management. Prog. Cardiovasc. Dis. 2013, 56, 143–152. [Google Scholar] [CrossRef]
- Hickey, K.T.; Riga, T.C.; Mitha, S.A.; Reading, M.J. Detection and management of atrial fibrillation using remote monitoring. Nurse Pr. 2018, 43, 24–30. [Google Scholar] [CrossRef]
- Sattar, Y.; Chhabra, L. Electrocardiogram; StatPearls Publishing: Treasure Island, FL, USA, 2021. Available online: https://www.ncbi.nlm.nih.gov/books/NBK549803/ (accessed on 10 September 2021).
- Everss-Villalba, E.; Melgarejo-Meseguer, F.; Blanco-Velasco, M.; Gimeno-Blanes, F.; Sala-Pla, S.; Rojo-Álvarez, J.; García-Alberola, A. Noise maps for quantitative and clinical severity towards long-term ECG monitoring. Sensors 2017, 17, 2448. [Google Scholar] [CrossRef] [Green Version]
- Jekova, I.; Krasteva, V.; Leber, R.; Schmid, R.; Twerenbold, R.; Reichlin, T.; Müller, C.; Abächerli, R. A real-time quality monitoring system for optimal recording of 12-lead resting ECG. Biomed. Signal Process. Control. 2017, 34, 126–133. [Google Scholar] [CrossRef]
- Dotsinsky, I. Atrial wave detection algorithm for discovery of some rhythm abnormalities. Physiol. Meas. 2007, 28, 595–610. [Google Scholar] [CrossRef]
- Du, X.; Rao, N.; Qian, M.; Liu, D.; Li, J.; Feng, W.; Yin, L.; Chen, X. A novel method for real-time atrial fibrillation detection in electrocardiograms using multiple parameters. Ann. Noninvasive Electrocardiol. 2014, 19, 217–225. [Google Scholar] [CrossRef]
- Christov, I.; Bortolan, G.; Daskalov, I. Sequential analysis for automatic detection of atrial fibrillation and flutter. Comput. Cardiol. 2001, 28, 293–296. [Google Scholar]
- Ladavich, S.; Ghoraani, B. Rate-independent detection of atrial fibrillation by statistical modeling of atrial activity. Biomed. Signal Process. Control 2015, 18, 274–281. [Google Scholar] [CrossRef]
- Linker, D. Accurate, automated detection of atrial fibrillation in ambulatory recordings. Cardiovasc. Eng. Technol. 2016, 7, 182–189. [Google Scholar] [CrossRef] [PubMed]
- Petrėnas, A.; Marozas, V.; Sörnmo, L. Low-complexity detection of atrial fibrillation in continuous long-term monitoring. Comput. Biol. Med. 2015, 65, 184–191. [Google Scholar] [CrossRef] [PubMed]
- Lake, D.; Moorman, J. Accurate estimation of entropy in very short physiological time series: The problem of atrial fibrillation detection in implanted ventricular devices. Am. J. Physiol. Heart Circ. Physiol. 2011, 300, H319–H325. [Google Scholar] [CrossRef] [PubMed]
- Babaeizadeh, S.; Gregg, R.; Helfenbein, E.; Lindauer, J.; Zhou, S. Improvements in atrial fibrillation detection for real-time monitoring. J. Electrocardiol. 2009, 42, 522–526. [Google Scholar] [CrossRef]
- Petrėnas, A.; Sörnmo, L.; Lukoševicius, A.; Marozas, V. Detection of occult paroxysmal atrial fibrillation. Med. Biol. Eng. Comput. 2015, 53, 287–297. [Google Scholar] [CrossRef]
- Larburu, N.; Lopetegi, T.; Romero, I. Comparative study of algorithms for atrial fibrillation detection. Comput. Cardiol. 2011, 38, 265–268. [Google Scholar]
- Clifford, G.D.; Liu, C.; Moody, B.; Lehman, L.H.; Silva, I.; Li, Q.; Johnson, A.E.; Mark, R.G. AF Classification from a short single lead ECG recording: The PhysioNet/computing in cardiology challenge 2017. In Proceedings of the 2017 Computing in Cardiology (CinC), Rennes, France, 24–27 September 2017. [Google Scholar] [CrossRef]
- Liu, C.; Lehman, L.; Moody, B.; Li, Q.; Clifford, G. Focus on detection of arrhythmia and noise from cardiovascular data. Physiol. Meas. 2018. Available online: https://iopscience.iop.org/journal/0967-3334/page/Focus_on_detection_of_arrhythmia_and_noise_from_cardiovascular_data (accessed on 29 August 2021).
- Athif, M.; Yasawardene, P.; Daluwatte, C. Detecting atrial fibrillation from short single lead ECGs using statistical and morphological features. Physiol. Meas. 2018, 39, 064002. [Google Scholar] [CrossRef] [PubMed]
- Liu, N.; Sun, M.; Wang, L.; Zhou, W.; Dang, H.; Zhou, X. A support vector machine approach for AF classification from a short single lead ECG recording. Physiol. Meas. 2018, 39, 064004. [Google Scholar] [CrossRef] [PubMed]
- Gliner, V.; Yaniv, Y. An SVM approach for identifying atrial fibrillation. Physiol. Meas. 2018, 39, 094007. [Google Scholar] [CrossRef] [PubMed]
- Sadr, N.; Jayawardhana, M.; Pham, T.; Tang, R.; Balaei, A.; de Chazal, P. A low complexity algorithm for detection of atrial fibrillation using the ECG. Physiol. Meas. 2018, 39, 064003. [Google Scholar] [CrossRef]
- Christov, I.; Krasteva, V.; Simova, I.; Neycheva, T.; Schmid, R. Ranking of the most reliable beat morphology and heart rate variability features for the detection of atrial fibrillation in short single-lead ECG. Physiol. Meas. 2018, 39, 094005. [Google Scholar] [CrossRef]
- Shao, M.; Bin, G.; Wu, S.; Bin, G.; Huang, J.; Zhou, Z. Detection of atrial fibrillation from ECG recordings using decision tree ensemble with multi-level features. Physiol. Meas. 2018, 39, 094008. [Google Scholar] [CrossRef]
- Chen, Y.; Wang, X.; Jung, Y.; Abedi, V.; Zand, R.; Bikak, M.; Adibuzzaman, M. Classification of short single lead electrocardiograms (ECGs) for atrial fibrillation detection using piecewise linear spline and XGBoost. Physiol. Meas. 2018, 39, 104006. [Google Scholar] [CrossRef]
- Kropf, M.; Hayn, D.; Morris, D.; Radhakrishnan, A.K.; Belyavskiy, E.; Frydas, A.; Pieske-Kraigher, E.; Pieske, B.; Schreier, G. Cardiac anomaly detection based on time and frequency domain features using tree-based classifiers. Physiol. Meas. 2018, 39, 114001. [Google Scholar] [CrossRef]
- Rizwan, M.; Whitaker, B.; Anderson, D. AF detection from ECG recordings using feature selection, sparse coding, and ensemble learning. Physiol. Meas. 2018, 39, 124007. [Google Scholar] [CrossRef]
- Mukherjeez, A.; Choudhuryz, A.; Datta, S.; Puri, C.; Banerjee, R.; Singh, R.; Ukil, A.; Bandyopadhyay, S.; Pal, A.; Khandelwal, S. Detection of atrial fibrillation and other abnormal rhythms from ECG using a multi-layer classifier architecture. Physiol. Meas. 2019, 40, 054006. [Google Scholar] [CrossRef]
- Smisek, R.; Hejc, J.; Ronzhina, M.; Nemcova, A.; Marsanova, L.; Kolarova, J.; Smital, L.; Vitek, M. Multi-stage SVM approach for cardiac arrhythmias detection in short single-lead ECG recorded by a wearable device. Physiol. Meas. 2018, 39, 094003. [Google Scholar] [CrossRef]
- Teijeiro, T.; García, C.A.; Castro, D.; Félix, P. Abductive reasoning as a basis to reproduce expert criteria in ECG atrial fibrillation identification. Physiol. Meas. 2018, 39, 084006. [Google Scholar] [CrossRef] [Green Version]
- Plesinger, F.; Nejedly, P.; Viscor, I.; Halamek, J.; Jurak, P. Parallel use of a convolutional neural network and bagged tree ensemble for the classification of Holter ECG. Physiol. Meas. 2018, 39, 094002. [Google Scholar] [CrossRef]
- Sodmann, P.; Vollmer, M.; Nath, N.; Kaderali, L. A Convolutional neural network for ECG annotation as the basis for classification of cardiac rhythms. Physiol. Meas. 2018, 39, 104005. [Google Scholar] [CrossRef] [PubMed]
- Parvaneh, S.; Rubin, J.; Rahman, A.; Conroy, B.; Babaeizadeh, S. Analyzing single-lead short ECG recordings using dense convolutional neural networks and feature-based post processing to detect atrial fibrillation. Physiol. Meas. 2018, 39, 084003. [Google Scholar] [CrossRef] [PubMed]
- Hong, S.; Zhou, Y.; Wu, M.; Shang, J.; Wang, Q.; Li, H.; Xie, J. Combining deep neural networks and engineered features for cardiac arrhythmia detection from ECG recordings. Physiol. Meas. 2019, 40, 054009. [Google Scholar] [CrossRef] [PubMed]
- Xiong, Z.; Nash, M.; Cheng, E.; Fedorov, V.; Stiles, M.; Zhao, J. ECG signal classification for the detection of cardiac arrhythmias using a convolutional recurrent neural network. Physiol. Meas. 2018, 39, 094006. [Google Scholar] [CrossRef]
- Warrick, P.; Homsi, M. Ensembling convolutional and long short-term memory networks for electrocardiogram arrhythmia detection. Physiol. Meas. 2018, 39, 114002. [Google Scholar] [CrossRef]
- Jiang, M.; Gu, J.; Li, Y.; Wei, B.; Zhang, J.; Wang, Z.; Xia, L. HADLN: Hybrid attention-based deep learning network for automated arrhythmia classification. Front. Physiol. 2021, 12, 683025. [Google Scholar] [CrossRef]
- Lee, H.; Shin, M. Learning explainable time-morphology patterns for automatic arrhythmia classification from short single-lead ECGs. Sensors 2021, 21, 4331. [Google Scholar] [CrossRef]
- Tran, L.; Li, Y.; Nocera, L.; Shahabi, C.; Xiong, L. MultiFusionNet: Atrial fibrillation detection with deep neural networks. AMIA Jt. Summits Transl. Sci. Proc. 2020, 2020, 654–663. [Google Scholar] [PubMed]
- Hsieh, C.-H.; Li, Y.-S.; Hwang, B.-J.; Hsiao, C.-H. Detection of atrial fibrillation using 1D convolutional neural network. Sensors 2020, 20, 2136. [Google Scholar] [CrossRef] [Green Version]
- Marinucci, D.; Sbrollini, A.; Marcantoni, I.; Morettini, M.; Swenne, C.; Burattini, L. Artificial neural network for atrial fibrillation identification in portable devices. Sensors 2020, 20, 3570. [Google Scholar] [CrossRef] [PubMed]
- Mousavi, S.; Afghah, F.; Acharya, U.R. HAN-ECG: An interpretable atrial fibrillation detection model using hierarchical attention networks. Comput. Biol. Med. 2020, 127, 104057. [Google Scholar] [CrossRef]
- Andersen, R.; Peimankar, A.; Puthusserypady, S. A deep learning approach for real-time detection of atrial fibrillation. Expert Syst. Appl. 2019, 115, 465–473. [Google Scholar] [CrossRef]
- Xia, Y.; Wulan, N.; Wang, K.; Zhang, H. Detecting atrial fibrillation by deep convolutional neural networks. Comput. Biol. Med. 2018, 93, 84–92. [Google Scholar] [CrossRef]
- Rouhi, R.; Clause, M.; Oster, J.; Lauer, F. An interpretable hand-crafted feature-based model for atrial fibrillation detection. Front. Physiol. 2021, 12, 657304. [Google Scholar] [CrossRef]
- Goldberger, A.L.; Amaral, L.A.N.; Glass, L.; Hausdorff, J.M.; Ivanov, P.C.; Mark, R.G.; Mietus, J.E.; Moody, G.B.; Peng, C.-K.; Stanley, H.E. PhysioBank, PhysioToolkit, and PhysioNet: Components of a new research resource for complex physiologic signals. Circulation 2000, 101, e215–e220. [Google Scholar] [CrossRef] [Green Version]
- Christov, I.; Krasteva, V.; Simova, I.; Neycheva, T.; Schmid, R. Multi-parametric analysis for atrial fibrillation classification in ECG. In Proceedings of the 2017 Computing in Cardiology (CinC), Rennes, France, 24–27 September 2017; Volume 44. [Google Scholar] [CrossRef]
- Jekova, I.; Stoyanov, T.; Dotsinsky, I. Arrhythmia classification via time and frequency domain analyses of ventricular and atrial contractions. In Proceedings of the 2017 Computing in Cardiology (CinC), Rennes, France, 24–27 September 2017; Volume 44, pp. 1–4. [Google Scholar] [CrossRef]
- Heart rate variability. Standards of measurement, physiological interpretation, and clinical use. Task Force of the European Society of Cardiology and the North American Society of Pacing and Electrophysiology. Eur. Heart J. 1996, 17, 354–381.
- Jekova, I.; Vassilev, P.; Stoyanov, T.; Pencheva, T. InterCriteria Analysis: Application for ECG Data Analysis. Mathematics 2021, 9, 854. [Google Scholar] [CrossRef]
- Ioffe, S.; Szegedy, C. Batch normalization: Accelerating deep network training by reducing internal covariate shift. In Proceedings of the 32nd International Conference on Machine Learning (ICML 2015), Lille, France, 6–11 July 2015; Volume 1, pp. 448–456. [Google Scholar]
- Ramsundar, B.; Zadeh, R.B. Chapter 4. fully connected deep networks. In TensorFlow for Deep Learning: From Linear Regression to Reinforcement Learning, 1st ed.; Roumeliotis, R., Young, A., Eds.; O’Reilly Media, Inc.: Sebastopol, CA, USA, 2018; pp. 81–102. [Google Scholar]
- MathWorks, Neural Network Architectures. Available online: https://ww2.mathworks.cn/help/deeplearning/ug/neural-network-architectures.html (accessed on 29 August 2021).
- Olden, J.D.; Jackson, D.A. Illuminating the “black box”: Understanding variable contributions in artificial neural networks. Ecol. Model. 2002, 154, 135–150. [Google Scholar] [CrossRef]
- Olden, J.D.; Joy, M.K.; Death, R.G. An accurate comparison of methods for quantifying variable importance in artificial neural networks using simulated data. Ecol. Model. 2004, 178, 389–397. [Google Scholar] [CrossRef]
- Li, C.; Farkhoor, H.; Liu, R.; Yosinski, J. Measuring the intrinsic dimension of objective landscapes. arXiv 2018, arXiv:1804.08838. [Google Scholar]
- Krasteva, V.; Ménétré, S.; Didon, J.-P.; Jekova, I. Fully convolutional deep neural networks with optimized hyperparameters for detection of shockable and non-shockable rhythms. Sensors 2020, 20, 2875. [Google Scholar] [CrossRef] [PubMed]
- Jekova, I.; Krasteva, V. Optimization of end-to-end convolutional neural networks for analysis of out-of-hospital cardiac arrest rhythms during cardiopulmonary resuscitation. Sensors 2021, 21, 4105. [Google Scholar] [CrossRef]
- Kandel, I.; Castelli, M. The effect of batch size on the generalizability of the convolutional neural networks on a histopathology dataset. ICT Express 2020, 6, 312–315. [Google Scholar] [CrossRef]
- Vidushi; Agarwal, M. Performance analysis on the basis of learning rate. In Innovations in Computer Science and Engineering. Lecture Notes in Networks and Systems; Saini, H., Sayal, R., Buyya, R., Aliseri, G., Eds.; Springer: Singapore, 2020; Volume 203, pp. 41–46. [Google Scholar] [CrossRef]
- Teijeiro, T.; Félix, P. On the adoption of abductive reasoning for time series interpretation. Artif. Intell. 2018, 262, 163–188. [Google Scholar] [CrossRef] [Green Version]
Rhythm Class | Total | DB Splits for 5-Fold Cross Validation | Rhythm Class | ||||
---|---|---|---|---|---|---|---|
Labelling | DB1 | DB2 | DB3 | DB4 | DB5 | Distribution | |
Normal sinus rhythm (N) | 5076 | 1016 | 1015 | 1015 | 1015 | 1015 | 59.5% |
Atrial fibrillation (AF) | 758 | 152 | 152 | 152 | 151 | 151 | 8.9% |
Other arrhythmia (O) | 2415 | 483 | 483 | 483 | 483 | 483 | 28.3% |
Noise (X) | 279 | 56 | 56 | 56 | 56 | 55 | 3.3% |
Analysis Type | Feature Index | Feature Name | Description |
---|---|---|---|
Noise detection | 1 | Noise correction | Binary value for triggering noise correction technique in case of QRS detection failure. Method described in [28] |
HRV analysis (RR–Tachogram) | 2–3 | RRmean, RRmedian | Mean and median value of all RR intervals (ms) [28,54] |
4–5 | RRstd, RRmeand | Standard, mean deviation of all RR intervals (ms) [28,54] | |
6–7 | RRstd%, RRmeand% | Proportion of Rstd and RRmeand to RRmean (%) [28,54] | |
8 | RRrat | Ratio of mean-to-median value of RR intervals [28,54] | |
HRV analysis (dRR–Tachogram) | 9–11 | dRRmean, dRRstd, dRRmedian | Mean value, standard deviation and median deviation of all RR intervals first differences dRR (ms) [28,54] |
12–14 | dRRmean%, dRRstd%, dRRmedian% | Proportion of dRRmean, dRRstd, dRRmedian to RRmean (%) [28,54] | |
15–16 | PNN50, PNN50% | Proportion of dRR > 50 ms normalized to dRR total sum (min−1) and total number (%) [28,54] | |
17–18 | RMSSD, RMSSD% | Root mean square of successive RR interval differences (ms) and its proportion to RRmean (%) [28,54] | |
HRV analysis (RR–Histogram) | 19 | HRV TrIdx | HRV Triangular Index: Total number of RR intervals divided by the number of RR intervals in the modal bin (7.8125 ms) [28,54] |
HRV analysis (Poincaré Plot (RRn, RRn−1)) | 20 | SD1/SD2 | Ratio of minor to major semi-axis of the fitted ellipse [28] |
21 | CCM | Complex Correlation Measure quantifying the point-to-point (dynamic) variation of the Poincaré plot [28] | |
22 | corRR | Pearson’s correlation coefficient, representing the goodness of linear fitting of all points in the Poincaré plot [28] | |
Average beat morphology analysis (Amplitudes) | 23 | ECG_range | Min-to-max range amplitude of the ECG signal (mV) [28] |
24 | stdAmp | Standard deviation of the peak absolute amplitudes of all detected beats (mV) [28] | |
25 | stdAmp% | Normalized standard amplitude deviation (stdAmp/meanAmp, %), where meanAmp is the mean peak absolute amplitude of all detected beats [28] | |
26 | rejAmp% | Proportion of rejected beats for synthesis of the average beat with extreme peak absolute amplitudes outside the range (meanAmp ± stdAmp) [28] | |
Average beat morphology analysis (Cross-correlation) | 27 28 29 | corBeat(mean), corBeat(50%), corBeat(25%) | Mean value, 25th and 50th percentile of the maximal cross-correlation coefficient of all detected beats against the average beat [28] |
Average beat morphology analysis (Delineation of fiducial points) | 30 | P-wave presence | Binary test for the detected P-wave using criteria in [28] |
31 | QRS-amp | Amplitude difference (R-S) (mV) [28] | |
32–33 | T-amp, P-amp | Absolute amplitude differences from T-peak, P-peak to the isoelectric Q-point [28] | |
34–35 | J-shift, J-shift% | Absolute amplitude shift of J-point in respect to Q-point (mV) and its normalized value to QRSpp-amp (%) [28] | |
36–38 | QRS-width, PQ-int, TQ-int | Durations of intervals between fiducial points (ms): QRS-width = (J-Q), PQ-int = Q-Ppeak, TQ-int =Tend-Q [28] | |
39 | Fragmentation | Binary test for detected inversion of the slope to the left of the R-peak (R-fragm) that could not be physiologically accepted as Q point, satisfying both criteria: - short interval between R-fragm and R-peak (<80 ms) - small amplitude drop at the point of the slope inversion (R-peak − R-fragm < 30% QRS-amp) [28] | |
40 | Inverted QRS-T | Binary test for detected opposite signs of QRS-peak and T-peak [28] | |
41 | LBBB | Binary test for detected specific case of inverted QRS and T-wave, satisfying two additional criteria: (QRS-width > 140 ms) and (T-amp > 1/3 ∗ QRS-amp) [28] | |
Average beat morphology analysis (Curvatures) | 42–47 | maxc(QRS), maxc(P), maxc(T), maxc(QRS/P), maxc(QRS/T), maxc(T/P) | Maximal curvatures during P, QRS, T waves and their ratios [28] |
Heartbeat classification and rhythm analysis | 48–51 | MeanAmpVB, MinAmpVB, MaxAmpVB, StdAmpVB | Mean, minimal, maximal values and standard deviation of the amplitudes of all ventricular beats VB (mV) [55] |
52–55 | MeanAmpN, MinAmpN, MaxAmpN, StdAmpN | Mean, minimal, maximal values and standard deviation of the amplitudes of all detected normal beats (mV) [55] | |
56–59 | MeanRRVB, MinRRVB, MaxRRVB, StdRRVB | Mean, minimal, maximal values and standard deviation of RR intervals of all VB (ms) [55] | |
60–63 | MeanRRN, MinRRN, MaxRRN, StdRRN | Mean, minimal, maximal values and standard deviation of the RR intervals between normal beats (ms) [55] | |
64 | NBeats% | Proportion of the number of normal heartbeats to the total number of beats (%) [55] | |
65 | AF% | Probability the rhythm to be AF based on assessment of the RR-intervals irregularity (%) [55] | |
66 | Complexity_ECG | ECG signal complexity [55] | |
Principal component analysis of PQRST and TQ-intervals | 67–70 | MeanStd_PCA_PQRST, MinStd_PCA_PQRST, MaxStd_PCA_PQRST, RangeStd_PCA_PQRST | Mean, minimal, maximal values and range of the standard deviation between the amplitude of samples in all PQRST segments and the corresponding samples in the PQRST first PCA vector (mV) [55] |
71 | MeanStd_PCA_TQ | Mean deviation between the amplitudes of samples in all TQ segments and the corresponding samples in the TQ first PCA vector (mV) [55] | |
P-wave analysis (time domain) | 72–75 | MeanAmpP, MinAmpP, MaxAmpP, StdAmpP | Mean, minimal, maximal values and standard deviation of the P-waves amplitudes (mV) [55] |
76–79 | MeanPPint, MinPPint, MaxPPint, StdPPint | Mean, minimal, maximal values and standard deviation of the intervals between consecutive P-waves (ms) [55] | |
80–81 | MeanPcountRRint, StdPcountRRint | Mean value and standard deviation of the P-waves number in each RR interval [55] | |
82 | DoubleP% | Proportion of RR intervals with two or more detected P-waves (%) [55] | |
83 | ratePP | Rate of P-waves (min−1) [55] | |
TQ-segment analysis (time domain) | 84 | Complexity_TQ | TQ segments complexity [55] |
85–88 | MeanLeak_TQ, MinLeak_TQ, MaxLeak_TQ, StdLeak_TQ | Mean, minimal, maximal values and standard deviation of the leakage, calculated for consecutive TQ segments [55] | |
89–92 | MeanT_TQ, MinT_TQ, MaxT_TQ, StdT_TQ | Mean, minimal, maximal values and standard deviation of the period (T), measured for consecutive TQ segments (ms) [55] | |
93 | Median_TQamp | Median amplitude of atrial fibrillatory waves in the TQ segment (mV) [28] | |
TQ-segment analysis (frequency domain) | 94–97 | MeanDF, MinDF, MaxDF, StdDF | Mean, minimal, maximal values and standard deviation of the dominant frequency of the power spectrum computed over non-overlapping 4 s intervals (Hz) [55] |
98–101 | MeanRI, MinRI, MaxRI, StdRI | Mean, minimal, maximal values and standard deviation of the regularity index, which quantifies the sharpness of the dominant peak in the spectra [55] | |
102–105 | MeanFSNM, MinFSNM, MaxFSNM, StdFSNM | Mean, minimal, maximal values and standard deviation of the first spectral normalized moment [55] | |
106–121 | MeanSpecWidth_level, MinSpecWidth_level, MaxSpecWidth_level, StdSpecWidth_level level = {_02,_04,_06,_08} | Mean, minimal, maximal values and standard deviation of the spectral width at 4 different levels (0.2, 0.4, 0.6, 0.8) of the normalized maximum power in the range 3–15 Hz [55] | |
122–137 | MeanSpecArea_level, MinSpecArea_level, MaxSpecArea_level, StdSpecArea_level level = {_02,_04,_06,_08} | Mean, minimal, maximal values and standard deviation of the spectral area enclosed within 4 different levels (0.2, 0.4, 0.6, 0.8) of the normalized maximum power in the range 3–15 Hz [55] |
Rhythm Classes | Predicted Classification | Performance Metrics | ||||||
---|---|---|---|---|---|---|---|---|
N | AF | O | X | F1-Score | Prec. (%) | Recall (%) | Acc. (%) | |
Normal sinus rhythm (N) | 4603 | 14 | 433 | 26 | 0.884 | 86.2 | 90.7 | 85.8 |
Atrial fibrillation (AF) | 16 | 615 | 111 | 16 | 0.821 | 83.1 | 81.1 | 96.9 |
Other arrhythmia (O) | 657 | 101 | 1627 | 30 | 0.702 | 73.3 | 67.4 | 83.8 |
Noise (X) | 67 | 10 | 49 | 153 | 0.607 | 68.0 | 54.8 | 97.7 |
Feature | Global Weights Importance (GWI) * | ||||||
---|---|---|---|---|---|---|---|
Rank | Input Number | Name | All Rhythms cumGWI (Ratio to max(cumGWI)) | Normal Rhythm Signed GWI (% cumGWI) | AF Signed GWI (% cumGWI) | Other Rhythm Signed GWI (% cumGWI) | Noise Signed GWI (% cumGWI) |
1 | 30 | P-wave presence | 0.957 (1.00) | 0.709 (74%) | −0.074 (8%) | 0.133 (14%) | 0.041 (4%) |
2 | 28 | corBeat(50%) | 0.824 (0.86) | 0.193 (23%) | 0.217 (26%) | 0.364 (44%) | −0.050 (6%) |
3 | 15 | PNN50 | 0.811 (0.85) | −0.412 (51%) | 0.131 (16%) | −0.219 (27%) | −0.049 (6%) |
4 | 16 | PNN50% | 0.706 (0.74) | −0.288 (41%) | 0.124 (18%) | −0.230 (33%) | −0.064 (9%) |
5 | 45 | maxc(QRS/P) | 0.655 (0.68) | −0.372 (57%) | 0.043 (7%) | −0.201 (31%) | −0.038 (6%) |
6 | 20 | SD1/SD2 | 0.612 (0.64) | −0.202 (33%) | 0.081 (13%) | 0.301 (49%) | 0.029 (5%) |
7 | 1 | Noise correction | 0.611 (0.64) | −0.351 (57%) | 0.041 (7%) | −0.212 (35%) | 0.007 (1%) |
8 | 43 | maxc(P) | 0.609 (0.64) | −0.244 (40%) | −0.214 (35%) | −0.133 (22%) | 0.019 (3%) |
9 | 27 | corBeat(mean) | 0.566 (0.59) | −0.004 (1%) | 0.248 (44%) | 0.250 (44%) | −0.063 (11%) |
10 | 29 | corBeat(25%) | 0.536 (0.56) | 0.104 (19%) | 0.199 (37%) | 0.185 (35%) | −0.048 (9%) |
11 | 32 | T-amp | 0.533 (0.56) | 0.335 (63%) | −0.036 (7%) | 0.137 (26%) | 0.025 (5%) |
12 | 22 | corRR | 0.510 (0.53) | 0.205 (40%) | 0.052 (10%) | −0.228 (45%) | −0.025 (5%) |
13 | 93 | Median_TQamp | 0.505 (0.53) | −0.109 (22%) | 0.168 (33%) | −0.170 (34%) | −0.058 (12%) |
14 | 60 | MeanRRN | 0.472 (0.49) | −0.246 (52%) | −0.121 (26%) | 0.091 (19%) | 0.014 (3%) |
15 | 64 | NBeats% | 0.471 (0.49) | 0.295 (63%) | 0.117 (25%) | −0.035 (7%) | −0.023 (5%) |
16 | 57 | MinRRVB | 0.453 (0.47) | −0.074 (16%) | −0.055 (12%) | −0.294 (65%) | −0.030 (7%) |
17 | 14 | dRRmedian% | 0.436 (0.46) | −0.354 (81%) | 0.039 (9%) | 0.017 (4%) | 0.026 (6%) |
18 | 12 | dRRmean% | 0.436 (0.46) | −0.382 (88%) | 0.024 (6%) | 0.018 (4%) | 0.012 (3%) |
19 | 11 | dRRmedian | 0.426 (0.45) | −0.288 (67%) | 0.074 (17%) | 0.048 (11%) | 0.016 (4%) |
20 | 71 | MeanStd_PCA_TQ | 0.412 (0.43) | −0.155 (38%) | −0.020 (5%) | −0.200 (49%) | 0.037 (9%) |
Number of Input Features (nF) | 4 | 8 | 16 | 32 | 64 | 137 |
---|---|---|---|---|---|---|
DenseNet-3@128-32-4 Params | @(128-32-4) 4908 | @(128-32-4) 5432 | @(128-32-4) 6480 | @(128-32-4) 8576 | @(128-32-4) 12,768 | @(128-32-4) 22,331 |
DenseNet-3@(nFx2)-(nF/2)-4 Params | @(8-4-4) 104 | @(16-4-4) 252 | @(32-8-4) 888 | @(64-16-4) 3312 | @(128-32-4) 12,768 | @(256-64-4) 52,443 |
DenseNet-3@(nF)-(nF/4)-4 Params | @(4-4-4) 68 | @(8-4-4) 148 | @(16-4-4) 404 | @(32-8-4) 1448 | @(64-16-4) 5456 | @(128-32-4) 22,331 |
Rank | Normal Rhythm | AF | Other Rhythm | Noise | ||||
---|---|---|---|---|---|---|---|---|
1 | P-wave presence | 1.0 ± 0.00 | corBeat(mean) | 0.98 ± 0.04 | corBeat(50%) | 0.99 ± 0.02 | corBeat(mean) | 0.78 ± 0.17 |
2 | PNN50 | 0.59 ± 0.07 | corBeat(50%) | 0.87 ± 0.09 | SD1/SD2 | 0.83 ± 0.09 | PNN50% | 0.76 ± 0.20 |
3 | dRRmean% | 0.55 ± 0.07 | maxc(P) | 0.86 ± 0.12 | MinRRVB | 0.81 ± 0.08 | Fragmentation | 0.68 ± 0.25 |
4 | Maxc(QRS/P) | 0.53 ± 0.07 | corBeat(25%) | 0.79 ± 0.10 | corBeat(mean) | 0.68 ± 0.11 | Median_TQamp | 0.67 ± 0.25 |
5 | Noise correction | 0.51 ± 0.10 | ratePP | 0.73 ± 0.12 | PNN50% | 0.63 ± 0.11 | corBeat(50%) | 0.64 ± 0.20 |
6 | dRRmedian% | 0.50 ± 0.04 | Median_TQamp | 0.68 ± 0.13 | corRR | 0.63 ± 0.14 | MeanAmpP | 0.64 ± 0.19 |
7 | T-amp | 0.48 ± 0.09 | P-amp | 0.56 ± 0.12 | PNN50 | 0.60 ± 0.12 | corBeat(25%) | 0.61 ± 0.17 |
8 | dRRmean | 0.43 ± 0.06 | MeanPPint | 0.54 ± 0.11 | Noise correction | 0.58 ± 0.10 | PNN50 | 0.58 ± 0.21 |
9 | NBeats% | 0.42 ± 0.09 | PNN50 | 0.53 ± 0.10 | Maxc(QRS/P) | 0.55 ± 0.10 | CCM | 0.57 ± 0.28 |
10 | dRRmedian | 0.41 ± 0.05 | AF% | 0.52 ± 0.15 | MeanStdPCA(TQ) | 0.55 ± 0.07 | Complexity_ECG | 0.56 ± 0.27 |
11 | PNN50% | 0.41 ± 0.08 | PNN50% | 0.50 ± 0.08 | PQ-int | 0.51 ± 0.10 | P-wave presence | 0.50 ± 0.29 |
12 | MeanRRN | 0.35 ± 0.06 | MeanRRN | 0.49 ± 0.10 | corBeat(25%) | 0.51 ± 0.12 | Maxc(QRS/P) | 0.48 ± 0.21 |
13 | maxc(P) | 0.35 ± 0.05 | NBeats% | 0.47 ± 0.11 | Median_TQamp | 0.47 ± 0.08 | MinRRN | 0.47 ± 0.24 |
14 | StdLeak_TQ | 0.34 ± 0.08 | HRV TrIdx | 0.46 ± 0.10 | stdAmp | 0.44 ± 0.11 | Maxc(T) | 0.47 ± 0.19 |
15 | MinSpecWidth_08 | 0.32 ± 0.06 | RRmean | 0.45 ± 0.15 | Complexity_ECG | 0.42 ± 0.09 | MeanStd_PCA_TQ | 0.46 ± 0.21 |
16 | StdRRN | 0.31 ± 0.05 | CCM | 0.44 ± 0.12 | Inverted QRS-T | 0.40 ± 0.10 | RRmean | 0.43 ± 0.17 |
17 | J-shift | 0.31 ± 0.08 | QT-int | 0.43 ± 0.10 | P-amp | 0.39 ± 0.14 | QRS-width | 0.42 ± 0.20 |
18 | MeanDF | 0.31 ± 0.05 | Maxc(T) | 0.41 ± 0.08 | HRV TrIdx | 0.38 ± 0.12 | Maxc(QRS/T) | 0.41 ± 0.22 |
19 | corRR | 0.30 ± 0.08 | MeanAmpP | 0.40 ± 0.09 | MaxPPint | 0.38 ± 0.07 | dRRstd% | 0.41 ± 0.18 |
20 | RMSSD% | 0.29 ± 0.05 | MeanRRVB | 0.37 ± 0.08 | T-amp | 0.38 ± 0.11 | StdAmpN | 0.38 ± 0.15 |
Study | Method | F1 (N) | F1 (AF) | F1 (O) | F1 (Total) |
---|---|---|---|---|---|
This study | Dense neural network (DenseNet-3@128-32-4) | ||||
137 features | 0.884 | 0.821 | 0.702 | 0.802 * | |
64 features (selected by Global Weights Importance) | 0.883 | 0.825 | 0.705 | 0.804 * | |
32 features (selected by Global Weights Importance) | 0.880 | 0.821 | 0.693 | 0.798 * | |
Athif et al. (2018) [24] | SVM classifier, 47 features | 0.879 | 0.779 | 0.673 | 0.777 * |
Liu et al. (2018) [25] | SVM classifier, 33 features | 0.908 | 0.786 | 0.718 | 0.800 # |
Gliner et al. (2018) [26] | SVM classifier, 61 features | 0.89 | 0.80 | 0.73 | 0.81 * |
Sadr et al. (2018) [27] | Linear discriminant classifier, 122 features | 0.87 | 0.71 | 0.63 | 0.737 * |
Quadratic discriminant classifier, 122 features | 0.86 | 0.61 | 0.53 | 0.667 * | |
Quadratic neural network, 122 features | 0.87 | 0.74 | 0.66 | 0.757 * | |
Christov et al. (2018) [28] | Linear discriminant classifier, 44 features | 0.899 | 0.809 | 0.697 | 0.800 # |
Shao et al. (2018) [29] | Decision tree classifier, 30 features | 0.91 | 0.82 | 0.73 | 0.82 # |
Chen et al. (2018) [30] | Decision tree classifier, morphological coefficients and HRV features | 0.90 | 0.78 | 0.74 | 0.805 * |
Kropf et al. (2018) [31] | Decision tree classifier, 400 features | 0.91 | 0.84 | 0.75 | 0.83 # |
Rizwan et al. (2018) [32] | Decision tree classifier | ||||
74 features | 0.889 | 0.781 | 0.699 | 0.790 * | |
40 features selected with Minimal Redundancy | 0.889 | 0.791 | 0.702 | 0.794 * | |
40 Minimal Redundancy features + 4 sparse features | 0.890 | 0.785 | 0.710 | 0.795 * | |
Mukherjee et al. (2019) [33] | Decision tree two-layer binary classifier, 188 features | 0.91 | 0.79 | 0.77 | 0.823 * |
Smisek et al. (2018) [34] | Multi-stage classifier (SVM + decision tree + threshold-based rules), 260 features | 0.90 | 0.81 | 0.72 | 0.81 # |
Teijeiro et al. (2018) [35] | Multi-stage classifier (Tree gradient boosting model + LSTM), 42 features | 0.92 | 0.86 | 0.77 | 0.85 # |
Rouhi et al. (2021) [50] | Random Forest classifier | ||||
52 features (selected by Logistic Regression) | 0.896 | 0.742 | 0.719 | 0.786 * | |
56 features (selected by Permutation Testing) | 0.899 | 0.740 | 0.730 | 0.790 * | |
43 features (selected by Random Forest) | 0.898 | 0.751 | 0.728 | 0.792 * | |
28 features (selected by SHAP technique) | 0.900 | 0.768 | 0.733 | 0.800 * |
Publisher’s Note: MDPI stays neutral with regard to jurisdictional claims in published maps and institutional affiliations. |
© 2021 by the authors. Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license (https://creativecommons.org/licenses/by/4.0/).
Share and Cite
Krasteva, V.; Christov, I.; Naydenov, S.; Stoyanov, T.; Jekova, I. Application of Dense Neural Networks for Detection of Atrial Fibrillation and Ranking of Augmented ECG Feature Set. Sensors 2021, 21, 6848. https://doi.org/10.3390/s21206848
Krasteva V, Christov I, Naydenov S, Stoyanov T, Jekova I. Application of Dense Neural Networks for Detection of Atrial Fibrillation and Ranking of Augmented ECG Feature Set. Sensors. 2021; 21(20):6848. https://doi.org/10.3390/s21206848
Chicago/Turabian StyleKrasteva, Vessela, Ivaylo Christov, Stefan Naydenov, Todor Stoyanov, and Irena Jekova. 2021. "Application of Dense Neural Networks for Detection of Atrial Fibrillation and Ranking of Augmented ECG Feature Set" Sensors 21, no. 20: 6848. https://doi.org/10.3390/s21206848
APA StyleKrasteva, V., Christov, I., Naydenov, S., Stoyanov, T., & Jekova, I. (2021). Application of Dense Neural Networks for Detection of Atrial Fibrillation and Ranking of Augmented ECG Feature Set. Sensors, 21(20), 6848. https://doi.org/10.3390/s21206848