OneVAE联合优化离散与连续表示,让视频自编码器训练更快、重建更准。
OneVAE: Joint Discrete and Continuous Optimization Helps Discrete Video VAE Train Better
- 用连续先验引导离散量化,加速收敛并提升性能。
- 多标记量化使PSNR提升近1 dB,压缩比不变。
- 首帧重建增强+联合优化,适合高压缩视频生成任务。
将视频编码为离散标记可与文本标记对齐,利于构建统一的多模态大模型,但相比连续表示带来显著时空压缩。以往离散视频自编码器存在训练不稳定、耗时长、重建质量差等问题。鉴于连续自编码器更易训练且表现更优,本文重新思考离散与连续表示的内在关联,发现FSQ能更好保留预训练连续自编码器先验。借助该先验,模型收敛速度比从零开始训练快数倍,且在收敛后表现更优。此外提出两项改进:其一,借鉴连续自编码器增大隐空间的做法,引入多标记量化机制,在不改变标记压缩比前提下实现近1 dB的PSNR提升;其二,强化首帧重建能力,使因果自编码器可在后续帧中利用该信息,显著提升4×16×16离散自编码器性能。进一步提出联合离散-连续优化方案,首次在单一网络中实现连续与离散表示的性能媲美。方法命名为OneVAE以体现二者统一性。
原文摘要 · Abstract (English)
Encoding videos into discrete tokens could align with text tokens to facilitate concise and unified multi-modal LLMs, yet introducing significant spatiotemporal compression compared to continuous video representation. Previous discrete video VAEs experienced unstable training, long training time, and degraded reconstruction quality. Given the easier training and superior performance of continuous VAEs, an intuitive idea is to enhance discrete video VAEs by leveraging continuous VAEs. After rethinking the intrinsic link between discrete and continuous representations, we found that FSQ could effectively preserve pre-trained continuous VAE priors compared to other quantization methods. By leveraging continuous VAE priors, it converges several times faster than training from scratch and achieves superior performance at convergence. Meanwhile, two structural improvements are proposed. First, inspired by how continuous VAEs enhance reconstruction via enlarged latent dimensions, we introduce a multi-token quantization mechanism, which achieves nearly a 1 dB improvement in PSNR without compromising the token compression ratio. Second, to tackle reconstruction challenges in high-compression video VAEs, we strengthen first-frame reconstruction, enabling the causal VAE to leverage this information in subsequent frames and markedly improving the performance of 4 x 16 x 16 discrete VAEs. Furthermore, we propose a joint discrete-continuous optimization scheme that unifies the two paradigms and, for the first time, achieves competitive performance on both continuous and discrete representations within a single network. We name our method OneVAE to reflect this connection.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。