STAR: an STGCN ARchitecture for Skeleton-Based Human Action Recognition
Weiwei Wu,Fengbin Tu,Mengqi Niu,Zhiheng Yue,Leibo Liu,Shaojun Wei,Xiangyu Li,Yang Hu,Shouyi Yin
DOI: https://doi.org/10.1109/tcsi.2023.3254610
2023-01-01
IEEE Transactions on Circuits and Systems I Regular Papers
Abstract:Skeleton-based human action cognition (HAR) has drawn increasing attention recently. As an emerging approach for skeleton-based HAR tasks, Spatial-Temporal Graph Convolution Network (STGCN) achieves remarkable performance by fully exploiting the skeleton topology information via graph convolution. Unfortunately, existing GCN accelerators lose efficiency when processing STGCN models due to two limitations. (1) At the dataflow level, the hardware parallelism of GCN accelerators cannot match the computation parallelism of STGCN models, leading to computing resource under-utilization. (2) At the computation level, GCN accelerators fail to exploit the inherent temporal redundancy in STGCN models. To overcome the limitations, this paper proposes STAR, an STGCN architecture for skeleton-based human action recognition. STAR is designed based on the characteristics of different computation phases in STGCN. For limitation (1), a spatial-temporal dimension consistent (STDC) dataflow is proposed to fully exploit the data reuse opportunities in all the different dimensions of STGCN. For limitation (2), we propose a node-wise exponent sharing scheme and a temporal-structured redundancy elimination mechanism, to exploit the inherent temporal redundancy specially introduced by STGCN. To further address the under-utilization induced by redundancy elimination, we design a dynamic data scheduler to manage the feature data storage and schedule the features and weights for valid computation in real time. STAR achieves $4.48\times $ , $5.98\times $ , $2.54\times $ , and $103.88\times $ energy savings on average over the HyGCN, AWB-GCN, TPU, and Jetson TX2 GPU.