Shap-Mix: Shapley Value Guided Mixing for Long-Tailed Skeleton Based Action Recognition
Jiahang Zhang, Lilang Lin, Jiaying Liu
TL;DR
Shap-Mix addresses the challenge of long-tailed skeleton-based action recognition by introducing a two-tier augmentation framework that combines spatial-temporal skeleton mixing (ST-Mix) with Shapley-value-guided saliency to preserve salient minority-class motion patterns. It maintains online saliency estimation via EMA and employs a tail-aware data synthesis distribution to improve decision boundaries for tail classes, all within end-to-end training and a balanced-softmax objective. Across NTU 60/120 and Kinetics Skeleton 400, the method achieves strong improvements on long-tailed distributions while remaining competitive on balanced data, with ablations confirming the effectiveness of both the ST-Mix design and Shapley-guided guidance. This work provides a practical, backbone-agnostic augmentation approach for robust skeleton-based action recognition in real-world, imbalanced settings, with code publicly available for reproducibility.
Abstract
In real-world scenarios, human actions often fall into a long-tailed distribution. It makes the existing skeleton-based action recognition works, which are mostly designed based on balanced datasets, suffer from a sharp performance degradation. Recently, many efforts have been madeto image/video long-tailed learning. However, directly applying them to skeleton data can be sub-optimal due to the lack of consideration of the crucial spatial-temporal motion patterns, especially for some modality-specific methodologies such as data augmentation. To this end, considering the crucial role of the body parts in the spatially concentrated human actions, we attend to the mixing augmentations and propose a novel method, Shap-Mix, which improves long-tailed learning by mining representative motion patterns for tail categories. Specifically, we first develop an effective spatial-temporal mixing strategy for the skeleton to boost representation quality. Then, the employed saliency guidance method is presented, consisting of the saliency estimation based on Shapley value and a tail-aware mixing policy. It preserves the salient motion parts of minority classes in mixed data, explicitly establishing the relationships between crucial body structure cues and high-level semantics. Extensive experiments on three large-scale skeleton datasets show our remarkable performance improvement under both long-tailed and balanced settings. Our project is publicly available at: https://jhang2020.github.io/Projects/Shap-Mix/Shap-Mix.html.
