Enhancing Activity Recognition After Stroke: Generative Adversarial Networks for Kinematic Data Augmentation
This work addresses the problem of data scarcity for clinicians monitoring stroke survivors, offering a domain-specific incremental improvement in activity recognition.
The study tackled the limited generalizability of machine learning models for stroke rehabilitation monitoring by using Conditional Generative Adversarial Networks (cGANs) to generate synthetic kinematic data, resulting in an increase in task classification accuracy from 66.1% to 80.0%.
The generalizability of machine learning (ML) models for wearable monitoring in stroke rehabilitation is often constrained by the limited scale and heterogeneity of available data. Data augmentation addresses this challenge by adding computationally derived data to real data to enrich the variability represented in the training set. Traditional augmentation methods, such as rotation, permutation, and time-warping, have shown some benefits in improving classifier performance, but often fail to produce realistic training examples. This study employs Conditional Generative Adversarial Networks (cGANs) to create synthetic kinematic data from a publicly available dataset, closely mimicking the experimentally measured reaching movements of stroke survivors. This approach not only captures the complex temporal dynamics and common movement patterns after stroke, but also significantly enhances the training dataset. By training deep learning models on both synthetic and experimental data, we enhanced task classification accuracy: models incorporating synthetic data attained an overall accuracy of 80.0%, significantly higher than the 66.1% seen in models trained solely with real data. These improvements allow for more precise task classification, offering clinicians the potential to monitor patient progress more accurately and tailor rehabilitation interventions more effectively.