atlas

Research map / Label-efficient and robust learning / Recognition and long-tail

MaskCLR: Attention-Guided Contrastive Learning for Robust Action Representation Learning

CVPR 2024 · Label-efficient and robust learning › Recognition and long-tail · approach: Correspondence and causal representation learning · 20 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes MaskCLR, which trains transformer skeleton action recognizers with attention-guided probabilistic joint masking plus sample- and class-level contrastive losses to improve accuracy, robustness, and cross-pose-estimator generalization.

Problem

Transformer-based skeleton action recognition relies on a small set of highly attended joints, so it degrades under small skeleton perturbations and when the pose estimator changes between training and testing.

Approach

Mask the most-attended joints (sampled from a sharpened softmax over attention scores via Gumbel top-K) and feed them to a masked pathway; align standard and masked representations at the sample level and align class-averaged representations with a class-contrastive loss that also targets confusing TP/FN/FP samples.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
add MaskCLR to DSTFormerDSTFormer · NTU60-XSubTop-1 accuracy: 92.8 -> 93.9Helped
add MaskCLR to DSTFormerDSTFormer · NTU60-XViewTop-1 accuracy: 97.1 -> 97.3Helped
add MaskCLR to DSTFormerDSTFormer · NTU120-XSubTop-1 accuracy: 84.8 -> 87.4Helped
add MaskCLR to DSTFormerDSTFormer · NTU120-XSetTop-1 accuracy: 86.4 -> 89.5Helped
add MaskCLR to DSTFormerDSTFormer · Kinetics400Top-1 accuracy: 38.8 -> 44.7Helped
add MaskCLR to DSTFormerDSTFormer · NTU120-XSub, HRNet (MQ) to ViTPose (HQ)Top-1 accuracy: 51.7 -> 78.3Helped
replace AGPM with 15% random masking (ablation)MaskCLR · NTU60-XSubTop-1 accuracy: 93.9 -> 92.0Hurt
add AGPM at masking ratio 0.7 instead of no masking (ablation)MaskCLR · NTU60-XSubTop-1 accuracy: 88.7 -> 91.7Helped
add sample contrastive loss to 15% random masking (ablation)MaskCLR · NTU60-XSubTop-1 accuracy: 89.7 -> 91.1Helped
add class contrastive loss to 15% random masking (ablation)MaskCLR · NTU60-XSubTop-1 accuracy: 89.7 -> 91.9Helped
add both sample- and class-level contrastive losses to 15% random masking (ablation)MaskCLR · NTU60-XSubTop-1 accuracy: 89.7 -> 92.0Helped
increase random masking from 15% to 30% (ablation)MaskCLR · NTU60-XSubTop-1 accuracy: 89.7 -> 89.6Hurt

Surprising findings

Limitations

Experimental setup

Datasets: NTU RGB+D 60 NTU RGB+D 120 Kinetics400

Benchmarks: NTU60-XSub NTU60-XView NTU120-XSub NTU120-XSet Kinetics400

Metrics: Top-1 accuracy

Baselines: MotionBERT PoseConv3D ST-GCN++ MS-G3D AAGCN CTR-GCN MAMP SkeletonMAE AimCLR CrosSCLR ActCLR FG-STFormer

Builds on: MotionBERT DSTFormer STTFormer SegFormer contrastive penalty Gumbel max trick

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy