Local/Global explainability empowered expert-involved frameworks for essential tremor action recognition

Lei Zhang,Yanjin Zhu,Qin Ni,Xiaochen Zheng,Zhenyu Gao,Qing Zhao
DOI: https://doi.org/10.1016/j.bspc.2024.106457
IF: 5.1
2024-05-26
Biomedical Signal Processing and Control
Abstract:The interpretability of machine learning (ML) and neural networks has been attached great importance in medical-related applications. Considering random forest (RF) algorithm with self-explanatory function then integrating the concept of human–machine hybrid system, this paper proposes a expert-involved explainable artificial intelligence (XAI) framework for essential tremor (ET) action recognition, which assists to reveal the black-box model globally and locally, and to provide decision-making suggestions for medical practitioners. In addition, the human empirical judgment of clinical diagnosis is integrated into the action recognition process, which contributes to the model performance improvement and forms a dynamic and feedback loop. This paper collecting sensing data of six actions from 25 participants (20 ET patients and 5 healthy subjects) via smart phone monitoring system. Standardization, feature extraction and data segmentation are used to select the key features. The RF is used to train and test datasets with three different dimensions, and the six-dimensional dataset is selected for interpretable analysis, in which the expert rating data improve the recognition performance. Then, the feature importance, permutation importance, and Shapley additive exPlanations (SHAP) and local interpretable model-agnostic explanations (LIME) methods are applied from abstract to concrete for model interpretability. The model-agnostic explainable approach are relatively flexible and universal, which can visualize the features that have positive or negative impacts on the action recognition process respectively, and obtain the specific abnormal value for further diagnosis assistance. The proposed XAI framework is expected to realize the analysis of a single abnormal sample and provide more acceptable and reliable empirical suggestions for early diagnosis and healthcare.
engineering, biomedical
What problem does this paper attempt to address?