English

CP-AGCN: Pytorch-based Attention Informed Graph Convolutional Network for Identifying Infants at Risk of Cerebral Palsy

Computer Vision and Pattern Recognition 2022-09-08 v1 Machine Learning Image and Video Processing

Abstract

Early prediction is clinically considered one of the essential parts of cerebral palsy (CP) treatment. We propose to implement a low-cost and interpretable classification system for supporting CP prediction based on General Movement Assessment (GMA). We design a Pytorch-based attention-informed graph convolutional network to early identify infants at risk of CP from skeletal data extracted from RGB videos. We also design a frequency-binning module for learning the CP movements in the frequency domain while filtering noise. Our system only requires consumer-grade RGB videos for training to support interactive-time CP prediction by providing an interpretable CP classification result.

Keywords

Cite

@article{arxiv.2209.02824,
  title  = {CP-AGCN: Pytorch-based Attention Informed Graph Convolutional Network for Identifying Infants at Risk of Cerebral Palsy},
  author = {Haozheng Zhang and Edmond S. L. Ho and Hubert P. H. Shum},
  journal= {arXiv preprint arXiv:2209.02824},
  year   = {2022}
}