Joint-Embedding Prediction of Masked Point Tubes for Self-Supervised Learning on 4D Point Cloud Videos
Organizations: Department of Computer Science and Information Engineering, National Taiwan University
Abstract
Self-supervised representation learning for 4D point cloud videos is challenging because annotations are costly and reconstruction-based pretraining can overemphasize low-level geometric details. We propose a JEPA-style framework that learns from unlabeled spatiotemporal point clouds through latent point-tube prediction. Instead of reconstructing raw coordinates, the model masks spatiotemporal regions and predicts their target representations from visible context representations in feature space. To stabilize latent prediction, we incorporate Sketched Isotropic Gaussian Regularization, which encourages non-collapsed embeddings without relying on explicit reconstruction targets. This formulation aims to capture both spatial structure and temporal dynamics while keeping the pretraining objective aligned with downstream semantic recognition. Experiments on action and gesture recognition benchmarks show that the learned representations improve downstream fine-tuning, limited-label learning, and cross-dataset transfer. These results suggest that JEPA-style latent prediction is a promising alternative to reconstruction-centered pretraining for 4D point cloud videos.
Figures & tables
| Method | Acc. |
|---|---|
| Supervised Learning Only | |
| MeteorNet ( Liu et al. 2019 ) | 88.50 |
| PSTNet ( Fan et al. 2021b ) | 91.20 |
| PSTNet++ ( Fan et al. 2022 ) | 92.68 |
| Kinet ( Zhong et al. 2022 ) | 93.27 |
| PPTr ( Wen et al. 2022 ) | 92.33 |
| Method | Full | 50% Labels |
|---|---|---|
| Supervised Learning Only | ||
| 3DV-Motion ( Wang et al. 2020 ) | 84.5 | – |
| 3DV-PointNet++ ( Wang et al. 2020 ) | 88.8 | – |
| PSTNet ( Fan et al. 2021b ) | 90.5 | – |
| PSTNet++ ( Fan et al. 2022 ) | 91.4 | – |
| Kinet ( Zhong et al. 2022 ) | 92.3 | – |
| Initialization | 1-shot | 3-shot | 5-shot |
|---|---|---|---|
| Scratch | |||
| Pretrained | |||
| Gain |
| Method | Epochs | Acc. |
| Supervised Learning Only | ||
| PointLSTM ( Min et al. 2020 ) | – | 87.6 |
| PointLSTM-PSS ( Min et al. 2020 ) | – | 93.1 |
| Kinet ( Zhong et al. 2022 ) | – | 95.2 |
| P4Transformer ( Fan et al. 2021a ) | 30 | 87.5 |
| P4Transformer ( Fan et al. 2021a ) | 50 | 91.2 |
| Variant | Acc. |
|---|---|
| Full model | 94.08 |
| w/o SIGReg | 89.90 |
| Stop-gradient targets | 79.44 |
| w/o target-position query | 93.03 |
| EMA target encoder | 88.50 |
| Setting | Value | Target Latent Std. | Acc. |
|---|---|---|---|
| Mask ratio | 0.65 | – | 88.85 |
| Mask ratio | 0.75 | – | 94.08 |
| Mask ratio | 0.85 | – | 91.99 |
| 0.001 | 0.870 | 86.76 | |
| 0.01 | 0.984 | 94.08 | |
| 0.1 | 0.992 | 89.55 |
Appendix figures & tables6 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset | Classes | Train/Test | Protocol |
|---|---|---|---|
| MSRAction-3D | 20 | 270 / 297 | S1–S5 / S6–S10 |
| NTU RGB+D | 60 | 40,320 / 16,560 | Official |
| SHREC’17 | 28 | 1,960 / 840 | Official |
| Protocol | Epochs | Batch | LR | WD | Frames | Pts. | Clip Str. | Frame Str. | Sp. Str. | Nbrs. | SIGReg | |
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| MSR pretraining | 200 | 96 | 3e-4 | 5e-2 | 24 | 1024 | 1 | 1 | 0.3 | 32 | 32 | 0.01 |
| NTU pretraining | 100 | 56 | 3e-4 | 5e-2 | 24 | 1024 | 2 | 2 | 0.1 | 32 | 32 | 0.03 |
| MSR full fine-tuning | 50 | 48 | 5e-4 | 1e-4 | 24 | 2048 | 1 | 1 | 0.7 | 32 | 32 | – |
| NTU full/semi fine-tuning | 20 | 48 | 5e-4 | 5e-2 | 24 | 2048 | 2 | 2 | 0.1 | 32 | 32 | – |
| SHREC transfer fine-tuning | 50 | 24 | 1e-3 | 1e-4 | 16 | 256 | 1 | 1 | 0.3 | 16 | 9 | – |
| MSR few-shot fine-tuning | 50 | 48 | 5e-4 | 1e-4 | 24 | 2048 | 1 | 1 | 0.7 | 32 | 32 | – |
| Setting | MSR | NTU | SHREC | Few-shot |
| Enc. dim | 384 | 384 | 384 | 384 |
| Enc. heads | 8 | 8 | 8 | 8 |
| Enc. depth | 5 | 10 | 10 | 10 |
| Enc. MLP | 4 | 4 | 4 | 4 |
| Pred. dim | 192 | 192 | – | – |
| Pred. heads | 6 | 6 | – | – |
| Variant | Pred. Loss | Target Std. | Acc. |
|---|---|---|---|
| Full model ( , ) | 0.001219 | 0.984 | 94.08 |
| w/o SIGReg | 0.000000 | 0.008 | 89.90 |
| Stop-gradient targets | 0.095069 | 0.993 | 79.44 |
| w/o target-position query | 0.002563 | 0.942 | 93.03 |
| EMA target encoder | 0.000452 | 0.388 | 88.50 |
| Setting | Value | Pred. Loss | Target Std. | Acc. |
|---|---|---|---|---|
| Mask ratio | 0.65 | 0.001050 | 0.984 | 88.85 |
| Mask ratio | 0.75 | 0.001219 | 0.984 | 94.08 |
| Mask ratio | 0.85 | 0.001401 | 0.985 | 91.99 |
| 0.001 | 0.000179 | 0.870 | 86.76 | |
| 0.01 | 0.001219 | 0.984 | 94.08 | |
| 0.1 | 0.010965 | 0.992 | 89.55 |
| 1-shot | 3-shot | 5-shot | ||||
| Seed | Scratch | Pretrained | Scratch | Pretrained | Scratch | Pretrained |
| 0 | 35.889 | 33.798 | 62.718 | 62.718 | 79.094 | 77.003 |
| 1 | 19.861 | 27.526 | 24.739 | 65.505 | 77.700 | 72.474 |
| 2 | 30.662 | 38.328 | 57.840 | 60.279 | 84.669 | 83.275 |
| 3 | 6.969 | 36.585 | 54.355 | 58.537 | 80.139 | 81.882 |
| 4 | 13.589 | 37.631 | 40.767 | 64.460 | 82.927 | 81.185 |