Diffusion-Augmented Spatiotemporal Graph Convolution for Clinical Gait and Motor Function Assessment

Read full text
Authors: Eran Bamani, Joao Buzzatto, Hermano Igo Krebs
Cited by: 3
References: 50

Abstract

Accurate assessment of gross motor function in children with cerebral palsy (CP) is essential for clinical decision-making, yet current practice is limited by data scarcity, severe class imbalance, and patient heterogeneity. Recent skeleton-based deep learning approaches, such as spatio-temporal graph convolutional networks (STGCN), enable automatic GMFCS prediction from monocular video but are limited in generalizability and fairness. In this work, we propose a unified generative-diagnostic pipeline that integrates a Conditional Skeleton Diffusion Model (CSDM) with a Biomechanically-Aware Spatio-Temporal Graph Convolutional Network (BA-STGCN). The CSDM generates clinically plausible 2D skeleton gait sequences conditioned on Gross Motor Function Classification System (GMFCS) level, Gait Deviation Index (GDI), and anthropometric covariates, guided by an anatomically structured covariance model to preserve biomechanical fidelity and clinical distributions. These synthetic sequences, combined with real patient data, are used to train the BA-STGCN, which incorporates a symmetry-based loss and a multi-task head for joint GMFCS classification and continuous GDI regression. Extensive evaluation on a pediatric clinical gait dataset demonstrates that our approach achieves 85.7% GMFCS classification accuracy with balanced precision and recall, reduces mean absolute error in GDI prediction to 4.6, and markedly improves recognition of severe phenotypes. These findings highlight that conditional skeleton diffusion, coupled with biomechanically informed graph learning, provides a scalable, interpretable, and privacy-preserving pathway for automated clinical gait assessment in CP.

Fields of Study & Subjects

Cerebral Palsy and Movement Disorders Balance Gait and Falls Prevention Stroke Rehabilitation and Recovery

Publication details

Journal / source
IEEE Transactions on Neural Systems and Rehabilitation Engineering
Publisher
Institute of Electrical and Electronics Engineers (IEEE)
Volume
34
Pages
1638–1649
Publication date
1 January 2026
Language
EN