Atrial fibrillation classification using step-by-step machine learning

被引:13
|
作者
Goodfellow, Sebastian D. [1 ]
Goodwin, Andrew [1 ]
Greer, Robert [1 ]
Laussen, Peter C. [1 ]
Mazwi, Mjaye [1 ]
Eytan, Danny [1 ,2 ]
机构
[1] Hosp Sick Children, Toronto, ON, Canada
[2] Rambam Med Ctr, Haifa, Israel
来源
关键词
machine learning; signal processing; ECG Waveforms; physionet challenge;
D O I
10.1088/2057-1976/aabef4
中图分类号
R8 [特种医学]; R445 [影像诊断学];
学科分类号
1002 ; 100207 ; 1009 ;
摘要
This paper presents a detailed overview of our submission to the 2017 Physionet Challenge where competitors were asked to build a model to classify a single lead ECG waveform as either normal sinus rhythm, atrial fibrillation, other rhythm, or noisy. A step-by-step machine learning pipeline was assembled, which included signal conditioning, R-peak detection and filtering, and feature extraction. Asuite of over 300 features, falling into one of three main feature groups; template features, RRI features, and full waveform features, were extracted from each waveform and an XGBoost, tree-based, gradient boosting classifier was used as the machine learning algorithm. The model produced a cross-validation F-1 score of 0.8245, a hidden sub-test score of 0.82, and a hidden test score of 0.8125. The score breakdown for each class (normal sinus rhythm, atrial fibrillation, other rhythm, and noisy) was as follows: F-1,F- NRS = 0.9024, F-1,F- AF = 0.8156, F-1,F- OR = 0.7194, F-1,F- Noise =. 0.5705.
引用
收藏
页数:16
相关论文
共 50 条
  • [31] Step-by-step migrations
    Holmes, TJ
    REVIEW OF ECONOMIC DYNAMICS, 2004, 7 (01) : 52 - 68
  • [32] STEP-BY-STEP REHABILITATION
    PAVORSKY, B
    MENTAL HOSPITALS, 1960, 11 (03): : 21 - &
  • [33] Step-by-step SMT
    Löw, Norbert
    Muehlbauer, Andreas
    SMT Surface Mount Technology Magazine, 2001, 15 (09):
  • [34] TVH - STEP-BY-STEP
    Occhino, J. A.
    Hokenstad, E. D.
    Trabuco, E. C.
    Green, I. C.
    INTERNATIONAL UROGYNECOLOGY JOURNAL, 2019, 30 : S205 - S206
  • [35] Biomimicry step-by-step
    Rowland, Regina
    BIOINSPIRED BIOMIMETIC AND NANOBIOMATERIALS, 2017, 6 (02) : 102 - 112
  • [36] STEP-BY-STEP RESUMES
    BOE, E
    MICROWAVES & RF, 1985, 24 (10) : 129 - +
  • [37] Drawing Step-by-Step
    Lombardo, Daniel
    LIBRARY JOURNAL, 2010, 135 (05) : 104 - 104
  • [38] SINEWAVES - STEP-BY-STEP
    HICKMAN, I
    ELECTRONICS WORLD & WIRELESS WORLD, 1995, (1708): : 215 - &
  • [39] STEP-BY-STEP BUILDING
    HALL, N
    BODOGAARD, T
    APPROPRIATE TECHNOLOGY, 1987, 13 (04) : 24 - 25
  • [40] STEP-BY-STEP TRAINING
    KELLEHER, J
    PERSONAL COMPUTING, 1986, 10 (12): : 111 - &