Graph Contrastive Learning for Skeleton-based Action Recognition
About
In the field of skeleton-based action recognition, current top-performing graph convolutional networks (GCNs) exploit intra-sequence context to construct adaptive graphs for feature aggregation. However, we argue that such context is still \textit{local} since the rich cross-sequence relations have not been explicitly investigated. In this paper, we propose a graph contrastive learning framework for skeleton-based action recognition (\textit{SkeletonGCL}) to explore the \textit{global} context across all sequences. In specific, SkeletonGCL associates graph learning across sequences by enforcing graphs to be class-discriminative, \emph{i.e.,} intra-class compact and inter-class dispersed, which improves the GCN capacity to distinguish various action patterns. Besides, two memory banks are designed to enrich cross-sequence context from two complementary levels, \emph{i.e.,} instance and semantic levels, enabling graph contrastive learning in multiple context scales. Consequently, SkeletonGCL establishes a new training paradigm, and it can be seamlessly incorporated into current GCNs. Without loss of generality, we combine SkeletonGCL with three GCNs (2S-ACGN, CTR-GCN, and InfoGCN), and achieve consistent improvements on NTU60, NTU120, and NW-UCLA benchmarks. The source code will be available at \url{https://github.com/OliverHxh/SkeletonGCL}.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Action Recognition | NTU RGB+D 120 (X-set) | Accuracy91.2 | 661 | |
| Action Recognition | NTU RGB+D 60 (Cross-View) | Accuracy97.1 | 575 | |
| Action Recognition | NTU RGB+D 60 (X-sub) | Accuracy93.1 | 467 | |
| Action Recognition | NTU RGB+D X-sub 120 | Accuracy89.8 | 377 | |
| Skeleton-based Action Recognition | NTU 60 (X-sub) | Accuracy93.1 | 220 | |
| Skeleton-based Action Recognition | NTU RGB+D 120 (X-set) | Top-1 Accuracy91.2 | 184 | |
| Action Recognition | NTU RGB+D X-View 60 | Accuracy97.1 | 172 | |
| Skeleton-based Action Recognition | NTU 120 (X-sub) | Accuracy89.8 | 139 | |
| Skeleton-based Action Recognition | NTU RGB+D 60 (X-View) | Top-1 Accuracy97.1 | 126 | |
| Action Recognition | NW-UCLA | Top-1 Acc96.8 | 67 |