An interpretable ensemble trees method with joint analysis of static and dynamic features for myocardial infarction detection

Abstract

Abstract Objective. In recent years, artificial intelligence-based electrocardiogram (ECG) methods have been massively applied to myocardial infarction (MI). However, the joint analysis of static and dynamic features to achieve accurate and interpretable MI detection has not been comprehensively addressed. Approach. This paper proposes a simplified ensemble tree method with a joint analysis of static and dynamic features to solve this issue for MI detection. Initially, the dynamic features are extracted by modeling the intrinsic dynamics of ECG via dynamic learning in addition to extracting classical static features. Secondly, a two-stage feature selection strategy is designed to identify a few significant features, which substitute the original variables that are employed in constructing the ensemble tree. This approach enhances the discriminative ability by selecting significant static and dynamic features. Subsequently, this paper presents an interpretable classification method named StackTree by introducing a stacked ensemble scheme to modify the ensemble tree simplification algorithm. The representative rules of the raw ensemble trees are selected as the intermediate training data that is used to retrain a decision tree with performance close to that of the source ensemble model. Using this scheme, the significant precision and interpretability of MI detection are thus comprehensively addressed. Main results. The effectiveness of our method in detecting MI is evaluated using the Physikalisch-Technische Bundesanstalt (PTB) and clinical database. The findings suggest that our algorithm outperforms the traditional methods based on a single type of feature. Additionally, it is comparable to the conventional random forest, achieving 97.1% accuracy under the inter-patient framework on the PTB database. Furthermore, feature subsets trained on PTB are validated using the clinical database, resulting in an accuracy of 84.5%. The chosen important features demonstrate that both static and dynamic information have crucial roles in MI detection. Crucially, the proposed method provides clear internal workings in an easy-to-understand visual manner.