STARS: Self-supervised Tuning for 3D Action Recognition in Skeleton Sequences

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Mehraban, Soroush, Rajabi, Mohammad Javad, Iaboni, Andrea, Taati, Babak
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915604852637696
author Mehraban, Soroush
Rajabi, Mohammad Javad
Iaboni, Andrea
Taati, Babak
author_facet Mehraban, Soroush
Rajabi, Mohammad Javad
Iaboni, Andrea
Taati, Babak
contents Self-supervised pretraining methods with masked prediction demonstrate remarkable within-dataset performance in skeleton-based action recognition. However, we show that, unlike contrastive learning approaches, they do not produce well-separated clusters. Additionally, these methods struggle with generalization in few-shot settings. To address these issues, we propose Self-supervised Tuning for 3D Action Recognition in Skeleton sequences (STARS). Specifically, STARS first uses a masked prediction stage using an encoder-decoder architecture. It then employs nearest-neighbor contrastive learning to partially tune the weights of the encoder, enhancing the formation of semantic clusters for different actions. By tuning the encoder for a few epochs, and without using hand-crafted data augmentations, STARS achieves state-of-the-art self-supervised results in various benchmarks, including NTU-60, NTU-120, and PKU-MMD. In addition, STARS exhibits significantly better results than masked prediction models in few-shot settings, where the model has not seen the actions throughout pretraining. Project page: https://soroushmehraban.github.io/stars/
format Preprint
id arxiv_https___arxiv_org_abs_2407_10935
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle STARS: Self-supervised Tuning for 3D Action Recognition in Skeleton Sequences
Mehraban, Soroush
Rajabi, Mohammad Javad
Iaboni, Andrea
Taati, Babak
Computer Vision and Pattern Recognition
Self-supervised pretraining methods with masked prediction demonstrate remarkable within-dataset performance in skeleton-based action recognition. However, we show that, unlike contrastive learning approaches, they do not produce well-separated clusters. Additionally, these methods struggle with generalization in few-shot settings. To address these issues, we propose Self-supervised Tuning for 3D Action Recognition in Skeleton sequences (STARS). Specifically, STARS first uses a masked prediction stage using an encoder-decoder architecture. It then employs nearest-neighbor contrastive learning to partially tune the weights of the encoder, enhancing the formation of semantic clusters for different actions. By tuning the encoder for a few epochs, and without using hand-crafted data augmentations, STARS achieves state-of-the-art self-supervised results in various benchmarks, including NTU-60, NTU-120, and PKU-MMD. In addition, STARS exhibits significantly better results than masked prediction models in few-shot settings, where the model has not seen the actions throughout pretraining. Project page: https://soroushmehraban.github.io/stars/
title STARS: Self-supervised Tuning for 3D Action Recognition in Skeleton Sequences
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2407.10935