论文标题

解释人体步态分析中年龄分类的机器学习模型

Explaining machine learning models for age classification in human gait analysis

论文作者

Slijepcevic, Djordje, Horst, Fabian, Simak, Marvin, Lapuschkin, Sebastian, Raberger, Anna-Maria, Samek, Wojciech, Breiteneder, Christian, Schöllhorn, Wolfgang I., Zeppelzauer, Matthias, Horsak, Brian

论文摘要

机器学习(ML)模型已被证明有效地分类了步态分析数据,例如,年轻人与老年人的二元分类。然而,ML模型缺乏为人类的预测提供可理解的解释。这种“黑框”行为阻碍了对哪些输入具有模型预测的理解。我们研究了一种可解释的人工智能方法,即,层次相关性传播(LRP),以获取步态分析数据。研究问题是:ML模型使用哪些输入特征来对步行模式的年龄相关差异进行分类?我们在健康参与者的赤脚步行过程中使用了AIST步态数据库的一部分,该子集包含五个双边反作用力(GRF)录音。每个输入信号在串联之前将最小值归一化,并送入卷积神经网络(CNN)。参与者分为三个年龄段:年轻(20-39岁),中年(40-64岁)和年龄较大(65-79岁)的成年人。在分层的十倍交叉验证上平均分类精度和相关得分(使用LRP得出)。平均分类精度为60.1%,显然高于37.3%的零规则基线。混乱矩阵表明,CNN区分了年轻和老年人,但很难对中年成年人进行建模。

Machine learning (ML) models have proven effective in classifying gait analysis data, e.g., binary classification of young vs. older adults. ML models, however, lack in providing human understandable explanations for their predictions. This "black-box" behavior impedes the understanding of which input features the model predictions are based on. We investigated an Explainable Artificial Intelligence method, i.e., Layer-wise Relevance Propagation (LRP), for gait analysis data. The research question was: Which input features are used by ML models to classify age-related differences in walking patterns? We utilized a subset of the AIST Gait Database 2019 containing five bilateral ground reaction force (GRF) recordings per person during barefoot walking of healthy participants. Each input signal was min-max normalized before concatenation and fed into a Convolutional Neural Network (CNN). Participants were divided into three age groups: young (20-39 years), middle-aged (40-64 years), and older (65-79 years) adults. The classification accuracy and relevance scores (derived using LRP) were averaged over a stratified ten-fold cross-validation. The mean classification accuracy of 60.1% was clearly higher than the zero-rule baseline of 37.3%. The confusion matrix shows that the CNN distinguished younger and older adults well, but had difficulty modeling the middle-aged adults.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源