RefTok: Reference-Based Tokenization for Video Generation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Fan, Xiang, Sun, Xiaohang, Thakkar, Kushan, Liu, Zhu, Bhat, Vimal, Krishna, Ranjay, Hao, Xiang
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908432317022208
author Fan, Xiang
Sun, Xiaohang
Thakkar, Kushan
Liu, Zhu
Bhat, Vimal
Krishna, Ranjay
Hao, Xiang
author_facet Fan, Xiang
Sun, Xiaohang
Thakkar, Kushan
Liu, Zhu
Bhat, Vimal
Krishna, Ranjay
Hao, Xiang
contents Effectively handling temporal redundancy remains a key challenge in learning video models. Prevailing approaches often treat each set of frames independently, failing to effectively capture the temporal dependencies and redundancies inherent in videos. To address this limitation, we introduce RefTok, a novel reference-based tokenization method capable of capturing complex temporal dynamics and contextual information. Our method encodes and decodes sets of frames conditioned on an unquantized reference frame. When decoded, RefTok preserves the continuity of motion and the appearance of objects across frames. For example, RefTok retains facial details despite head motion, reconstructs text correctly, preserves small patterns, and maintains the legibility of handwriting from the context. Across 4 video datasets (K600, UCF-101, BAIR Robot Pushing, and DAVIS), RefTok significantly outperforms current state-of-the-art tokenizers (Cosmos and MAGVIT) and improves all evaluated metrics (PSNR, SSIM, LPIPS) by an average of 36.7% at the same or higher compression ratios. When a video generation model is trained using RefTok's latents on the BAIR Robot Pushing task, the generations not only outperform MAGVIT-B but the larger MAGVIT-L, which has 4x more parameters, across all generation metrics by an average of 27.9%.
format Preprint
id arxiv_https___arxiv_org_abs_2507_02862
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle RefTok: Reference-Based Tokenization for Video Generation
Fan, Xiang
Sun, Xiaohang
Thakkar, Kushan
Liu, Zhu
Bhat, Vimal
Krishna, Ranjay
Hao, Xiang
Computer Vision and Pattern Recognition
Effectively handling temporal redundancy remains a key challenge in learning video models. Prevailing approaches often treat each set of frames independently, failing to effectively capture the temporal dependencies and redundancies inherent in videos. To address this limitation, we introduce RefTok, a novel reference-based tokenization method capable of capturing complex temporal dynamics and contextual information. Our method encodes and decodes sets of frames conditioned on an unquantized reference frame. When decoded, RefTok preserves the continuity of motion and the appearance of objects across frames. For example, RefTok retains facial details despite head motion, reconstructs text correctly, preserves small patterns, and maintains the legibility of handwriting from the context. Across 4 video datasets (K600, UCF-101, BAIR Robot Pushing, and DAVIS), RefTok significantly outperforms current state-of-the-art tokenizers (Cosmos and MAGVIT) and improves all evaluated metrics (PSNR, SSIM, LPIPS) by an average of 36.7% at the same or higher compression ratios. When a video generation model is trained using RefTok's latents on the BAIR Robot Pushing task, the generations not only outperform MAGVIT-B but the larger MAGVIT-L, which has 4x more parameters, across all generation metrics by an average of 27.9%.
title RefTok: Reference-Based Tokenization for Video Generation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2507.02862