arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2310.06322cs.LGcs.AI

使用深度学习模型预测三种冻结步态事件

Predicting Three Types of Freezing of Gait Events Using Deep Learning Models

University of Toronto(多伦多大学);King Mongkut’s University of Technology Thonburi(国王科技大学(吞武里))

Wen Tao Mo, Jonathan H. Chan

AI总结:

本研究利用Transformer编码器与双向LSTM构建深度学习模型,预测三种冻结步态事件,最佳测试得分0.427,位列Kaggle竞赛前五,并指出过拟合可通过伪标签和简化架构改进。

摘要:

冻结步态是帕金森病的一种症状,它会间歇性地使患者在行走时无法迈步或转身。尽管医学专家已经发现了冻结步态的各种诱因和缓解措施,但其根本原因和预测模型至今仍在探索中。目前,利用机器学习的冻结步态预测模型基于时间序列数据在预测冻结步态方面达到了较高的灵敏度和特异性;然而,这些模型缺乏对冻结步态事件类型的说明。我们开发了多种深度学习模型,采用Transformer编码器架构加上双向LSTM层以及不同的特征集,来预测三种不同类型的冻结步态事件。表现最佳的模型在测试数据上取得了0.427的分数,这将在由迈克尔·J·福克斯基金会主办的Kaggle冻结步态预测竞赛中排名前五。然而,我们也认识到训练数据中存在过拟合问题,这可以通过对额外数据进行伪标签以及简化模型架构来潜在改进。

英文摘要:

Freezing of gait is a Parkinson's Disease symptom that episodically inflicts a patient with the inability to step or turn while walking. While medical experts have discovered various triggers and alleviating actions for freezing of gait, the underlying causes and prediction models are still being explored today. Current freezing of gait prediction models that utilize machine learning achieve high sensitivity and specificity in freezing of gait predictions based on time-series data; however, these models lack specifications on the type of freezing of gait events. We develop various deep learning models using the transformer encoder architecture plus Bidirectional LSTM layers and different feature sets to predict the three different types of freezing of gait events. The best performing model achieves a score of 0.427 on testing data, which would rank top 5 in Kaggle's Freezing of Gait prediction competition, hosted by THE MICHAEL J. FOX FOUNDATION. However, we also recognize overfitting in training data that could be potentially improved through pseudo labelling on additional data and model architecture simplification.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑