通过时空双重掩码提升骨骼动作自监督识别性能
Spatial Hierarchy and Temporal Attention Guided Cross Masking for Self-supervised Skeleton-based Action Recognition
- 从空间与时间双维度设计掩码策略,增强模型泛化能力
- 在三个数据集上均实现领先效果,NTU-120准确率达94.3%
- 适合关注自监督骨骼动作识别的研究者和开发者
在自监督骨骼动作识别中,掩码重建范式因能有效提升模型精度和鲁棒性而受到关注。然而,以往方法多依赖单一掩码准则,导致模型过拟合特定特征并忽略其他有效信息。本文提出一种层级与注意力引导的交叉掩码框架(HA-CM),从空间和时间两个层面对骨骼序列实施掩码。在空间图中,利用双曲空间保持关节间的区分度,并有效保留高维骨骼的层级结构,以关节层级作为掩码依据;在时间流中,用关节的全局注意力替代传统距离度量,解决高维空间中距离收敛问题及缺乏全局视角的缺陷。此外,将基于交叉掩码框架的交叉对比损失引入损失函数,强化模型对实例级特征的学习。HA-CM在三个公开大规模数据集(NTU-60、NTU-120、PKU-MMD)上展现出高效性与普适性。代码已开源。
原文摘要 · Abstract (English)
In self-supervised skeleton-based action recognition, the mask reconstruction paradigm is gaining interest in enhancing model refinement and robustness through effective masking. However, previous works primarily relied on a single masking criterion, resulting in the model overfitting specific features and overlooking other effective information. In this paper, we introduce a hierarchy and attention guided cross-masking framework (HA-CM) that applies masking to skeleton sequences from both spatial and temporal perspectives. Specifically, in spatial graphs, we utilize hyperbolic space to maintain joint distinctions and effectively preserve the hierarchical structure of high-dimensional skeletons, employing joint hierarchy as the masking criterion. In temporal flows, we substitute traditional distance metrics with the global attention of joints for masking, addressing the convergence of distances in high-dimensional space and the lack of a global perspective. Additionally, we incorporate cross-contrast loss based on the cross-masking framework into the loss function to enhance the model's learning of instance-level features. HA-CM shows efficiency and universality on three public large-scale datasets, NTU-60, NTU-120, and PKU-MMD. The source code of our HA-CM is available at https://github.com/YinxPeng/HA-CM-main.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。