Octree-based Video Representation
Organizations: Peking University
Abstract
Video models commonly use uniform grids even though visual complexity varies substantially across space and time. We introduce OctVideo, which approximates a video clip with an octree. This hierarchy recursively partitions a spatio-temporal volume into eight subvolumes, so that smooth regions remain coarse while detailed regions receive finer cells. Each leaf stores local RGB values and spatio-temporal gradients, supplemented by a lightweight learned residual. For reconstruction, a Conv1D VAE maps the serialized cells to a regular latent grid and selectively refines details during decoding. Our VAE achieves 36.12 dB PSNR with 38.2M parameters and 189.4 GFLOPs per clip on Kinetics-400 (K400). It also generalizes zero-shot to the high-resolution Densely Annotated VIdeo Segmentation (DAVIS) 2016 dataset with reconstruction quality comparable to the best evaluated models. On both datasets, it requires the fewest model FLOPs and achieves the fastest encoding and decoding among the evaluated models. OctVideo also supports video understanding, achieving competitive recognition performance with few input tokens when trained from scratch. By exploiting the redundancy already present in video signals and efficiently processing sparse structures, OctVideo provides an efficient representation for video.
Figures & tables
| Method | Params | GFLOPs | Mem. | MSE | MAE | PSNR | SSIM | LPIPS | Enc./Dec. | |
|---|---|---|---|---|---|---|---|---|---|---|
| Wan2.1 | 32 | 126.9M | 14,685.8 | 1.669 | 0.0108 | 35.72 | 0.958 | 0.0182 | 0.2005/0.3423 | |
| CogVideoX | 32 | 206M | 33,014.4 | 4.924 | 0.0148 | 32.30 | 0.939 | 0.0380 | 0.4403/0.8770 | |
| IV-VAE | 29 | 241.9M | 33,132.6 | 2.082 | 0.0096 | 36.84 | 0.966 | 0.0188 | 0.9006/0.8279 | |
| WF-VAE | 33 | 316M | 12,953.2 | 2.693 | 0.0100 | 36.29 | 0.962 | 0.0227 | 0.0700/0.2387 | |
| LeanVAE | 29 | 40M | 328.3 | 0.251 | 0.0111 | 35.34 | 0.955 | 0.0200 | 0.0114/0.0210 | |
| OctVideo | 32 | 38.2M | 189.4 | 0.803 | 0.0106 | 36.12 | 0.960 | 0.0220 | 0.0105/0.0091 |
| Method | PSNR | SSIM | LPIPS | rFVD | VAE s/clip | GFLOPs | |
|---|---|---|---|---|---|---|---|
| Wan2.1 \citeyearpar wan21 | 32 | 33.47 | 0.904 | 0.0713 | 23.0 | 466,892 | |
| HunyuanVideo \citeyearpar hunyuanvae | 29 | 32.73 | 0.901 | 0.079 | 27.1 | 1,320,531 | |
| LeanVAE \citeyearpar leanvae | 29 | 32.68 | 0.898 | 0.0768 | 18.3 | 0.904 | 12,668 |
| OctVideo | 32 | 32.76 | 0.891 | 0.094 | 22.1 | 0.540 | 7815 |
| Variant | Controlled change | Avg. tokens | GFLOPs | PSNR | SSIM | LPIPS |
|---|---|---|---|---|---|---|
| Dense Conv3D (small) | lightweight dense baseline | 8,192 | 239.7 | 32.30 | 0.928 | 0.0605 |
| Dense Conv3D (large) | scale dense capacity | 8,192 | 1,780.5 | 34.12 | 0.947 | 0.0399 |
| RGB-only tokens | remove explicit gradients | 3,225 | 189.4 | 32.70 | 0.932 | 0.0537 |
| Random matched tree | replace error-guided placement | 3,225 | 189.4 | 33.50 | 0.941 | 0.0422 |
| Full decoder expansion | remove selective refinement | 3,225 | 835.9 | 33.90 | 0.940 | 0.0403 |
| w/o latent residual | remove dense latent residual | 3,225 | 189.2 | 32.20 | 0.913 | 0.0535 |
| Input | Top-1 | Top-5 | Top-1 | Top-5 |
|---|---|---|---|---|
| Original video | 82.27 | 96.80 | – | – |
| Direct reconstruction | 79.51 | 94.40 |
| Method | Pre-train | Params | Input | Views | Tokens/view | Top-1 | Top-5 |
|---|---|---|---|---|---|---|---|
| Rev-MViT-B \citep mangalam2022reversible | none | 34.9M | 25,088 | 78.5 | 93.4 | ||
| MViTv1-B \citep fan2021multiscale | none | 36.6M | 50,176 | 80.2 | 94.4 | ||
| MViTv2-S \citep li2022mvitv2 | none | 34.5M | 25,088 | 81.0 | 94.6 | ||
| OctVideo | none | 31M | 4,200 | 80.1 | 94.4 | ||
| Video Swin-T \citep videoswinT | IN-1K | 28.2M | 50,176 | 78.8 | 93.6 | ||
| Hiera-B \citep ryali2023hiera | K400 MAE | 51M | 25,088 | 84.0 | – |
Appendix figures & tables8 assets
Supplementary material from the paper’s appendix.
Appendix
| Level | Active depths | Operation | Consumer | Avg. tokens |
|---|---|---|---|---|
| Sparse input | – | adaptive leaves | classifier | 4,200 |
| Level-5 state | – | pool | classifier | 3,700 |
| Level-4 state | – | pool | classifier | 800 |
| Method | Pretraining | Top-1 | Top-5 |
|---|---|---|---|
| C3D (3 nets) \citep tran2015learning | I380K+Sports-1M | 85.2 | – |
| I3D-RGB \citep Carreira_2017_CVPR | ImageNet+K400 | 95.6 | – |
| R(2+1)D-RGB \citep tran2018closer | K400 | 96.8 | – |
| TSM \citep lin2019tsm | K400 | 95.9 | 99.7 |
| VideoMAE V2-g \citep wang2023videomaev2 | UnlabeledHybrid+K710 | 99.6 | 100.0 |
| OctVideo (ours) | K400 | 96.43 | 99.52 |
| Component | Input output | Layers and hyperparameters | Activation / normalization |
|---|---|---|---|
| Analytic token projection | Linear | LayerNorm, GELU | |
| Per-leaf residual | Conv3D , 24 ch.; [DWConv3D , Conv3D ]; mean+max pool; Linear | SiLU; zero-init. output; start at 10% | |
| Morton sequence encoder | 8 blocks; DWConv1D ; dilations ; MLP ratio 4 | Pre-LayerNorm, GELU | |
| Copy/pool mapping | Linear ; copy coarse leaves / mean-pool 8 children | learned 3D position embedding | |
| Dense latent residual | Conv3D , 32 ch.; 1 residual [DWConv3D , Conv3D ]; Conv3D to 128 | GroupNorm, SiLU; zero-init. output; start at 20% | |
| Latent-grid block | 1 block; DWConv3D ; pointwise MLP ratio 1 | GroupNorm, GELU |
| Stage | Blocks | Token operation | Feature configuration |
|---|---|---|---|
| Input | – | Linear + depth embedding | width 512 |
| Stage 1 | 4 | patch attention, then pool | 8 heads, chunk 256, MLP |
| Stage 2 | 4 | patch attention, then pool | 8 heads, chunk 256, MLP |
| Stage 3 | 2 | patch attention | 8 heads, chunk 256, MLP |
| Readout | – | single-query attention + Linear | final FP32 LayerNorm |
Video Representation Learning
Video representation learning develops features that capture appearance, motion, and temporal structure for recognition, prediction, and control.
See all 143 papers