[edit]
A Lightweight Deep Residual Network for Rehabilitation Activity Recognition in Heterogeneous Pediatric Populations
Proceedings of the Fourth Swiss AI Days, PMLR 309:13-26, 2026.
Abstract
Human Activity Recognition based on wearable Inertial Measurement Units (IMUs) has emerged as a promising technology for the automated quantification of rehabilitation dosage and rehabilitation activity identification. However, existing solutions rely on multi-sensor configurations limiting clinical usability or fail to generalize to populations with heterogeneous motor functions. This study developed a lightweight residual network with self-attention mechanisms for classifying different phases of the rehabilitation activities (Rest, Balance, Walk) using data from a single IMU placed at the lower back, and collected from a pediatric cohort including 10 neurotypical children (mean age: years, 5 females) and 8 patients with neuromotor disorders as a consequence of cerebral palsy or acquired brain injury (mean age: years, 4 females). A preliminary ablation study across different IMU channel combinations revealed that combining accelerometer, gyroscope, and magnetometer signals allowed the model to achieve the best performance, with the magnetometer providing a key contribution for better discriminating between low-dynamic activities (Rest and Balance). Based on the optimal channel configuration identified in the ablation study, a Leave-One-Subject-Out cross-validation framework proved the model generalization abilities across heterogeneous motor functional domains, achieving an average macro F1-score of 0.81. These results confirm that the proposed framework provides an ecological and reliable tool for the objective recognition and quantification of rehabilitation activity in a clinical context.