AVTok:用于整体音视频生成的1D统一Tokenization AVTok: 1D Unified Tokenization for Holistic Audio-Video Generation
提出统一音视频tokenizer,双流transformer实现1D联合编码
前置知识
Tokenization
Tokenization是将高维连续数据(如音频、视频)压缩成低维离散token序列的过程,类似于将文本分割成词或子词。在深度学习中,这通常通过编码器-量化器-解码器架构实现,编码器将输入映射到潜在空间,量化器将连续潜在表示离散化为有限码本中的索引,解码器则从token重建原始信号。有效的tokenization能大幅降低计算复杂度,同时保留语义信息。
本文的核心是音视频tokenization,理解这一概念是把握全文技术路线的基础。作者对比了1D和3D tokenization的区别,设计了专门的量化策略和码本结构。
Query-based Transformer
传统transformer通过self-attention机制处理序列,但query-based transformer引入了一组可学习的查询向量作为输入,而不是序列本身。这些查询通过cross-attention机制与输入特征交互,可以自适应地关注输入的不同部分,从而提取更全局、更语义化的表示。这种方法在VQ-GAN的变体中得到广泛应用,特别适合需要整体理解而非局部特征的任务。
AVTok的核心架构基于query-based transformer设计,使用1024个holistic query来捕获音视频的全局信息。理解这种设计原理对于理解作者如何实现整体tokenization至关重要。
Autoregressive Generation
自回归生成是指根据已生成的序列预测下一个token的过程,形式化为概率分布p(x_t | x_t)。在离散token空间中,这通常通过语言模型实现,如GPT系列。自回归模型的优点是能够利用因果依赖关系,生成时可以逐步细化,缺点是推理时需要串行生成,速度较慢。对于音视频生成,自回归模型需要token具有清晰的因果顺序。
AVTok专门为自回归生成设计了1D token表示和训练策略,包括AR prior model和两种token顺序。理解这一概念有助于理解作者的设计动机和下游任务设置。
Dual-stream Architecture
双流架构是指为不同模态或任务设计并行的处理路径,通常共享部分参数但保留模态特定的组件。在AVTok中,音频和视频流使用相同的编码器-解码器主体,但拥有独立的归一化层和可学习查询向量用于视频,用于音频。这种设计既能保持模态特定性,又能通过共享参数实现隐式的跨模态交互。
双流架构是AVTok与vanilla单流版本的关键区别,也是论文消融实验验证的重要组件。理解这一设计能帮助读者理解作者如何在效率和效果之间取得平衡。
VFAL Training Strategy
VFAL是一种分层训练策略,分三个阶段:Stage 1仅训练视频流,建立强视觉表示空间;Stage 2冻结视频和共享模块,仅训练音频特定模块;Stage 3微调解码器以实现统一音视频重构。这种渐进式训练解决了视觉信息丰富性主导导致音频学习被抑制的问题,类似于课程学习的思想。
VFAL是本文的关键创新之一,消融实验显示移除VFAL会导致rFVD从12.80上升到13.19,rFAD从5.93上升到9.38。理解这一策略有助于理解作者如何处理模态不平衡问题。
研究动机
现有音视频生成模型面临三个核心问题。首先是计算成本高昂,主流方法如JavisDiT和Ovi采用重型双分支架构,每个模态有独立的tokenization和生成模块,加上额外的跨模态交互模块,导致模型参数量巨大(Ovi达到17.3B生成器参数),训练和部署成本极高。其次是语义对齐问题,不同模态使用独立预训练tokenizer(如视频用LARP,音频用UniCodec)会在各自的嵌入空间中学习表示,如图2所示,这些空间之间存在显著的表示间隙,导致生成的音视频在语义层面不匹配。最后是架构复杂性,双分支设计需要精心设计跨模态融合机制,增加了模型设计和优化的难度。
本文的目标是本文的核心目标是设计一个统一的音视频tokenizer,能够将音频-视频对高效且有效地编码到单一的1D潜在表示中。具体而言,这个tokenizer应该满足四个要求:第一,使用统一的码本,避免模态间的表示间隙;第二,采用1D离散潜在表示,保持与音频原生1D表示的一致性,同时适应自回归生成;第三,能够实现高质量的重构,为下游生成任务提供良好的基础;第四,能够支持三种下游生成任务:Audio-to-Video(A2V)、Video-to-Audio(V2A)和Class-conditional Joint Audio-Video Generation(cJAVG),即音频到视频、视频到音频和类别条件联合音视频生成。
与已有工作不同的是,本文的独特切入角度是首次提出统一的音视频tokenization任务,这与现有方法形成鲜明对比。现有方法要么是单模态tokenizer(如视频领域的LARP、AdapTok,音频领域的UniCodec、SpectralCodec),要么是音视频生成模型但使用双tokenizer设计。本文的核心洞察是:如果能够在共享的嵌入空间中联合编码音频和视频,不仅能够避免表示间隙问题,还能消除维护昂贵双分支架构的必要性。这是一个新颖且具有挑战性的研究方向,因为需要同时解决三个子问题:选择合适的表示形式、设计有效的架构实现联合编码、以及开发合适的训练策略处理模态信息不平衡。
核心方法
AVTok的整体思路是借鉴1D视频tokenization的成功经验,将其扩展到音视频联合编码的统一框架。从直觉上讲,视频虽然是3D数据(时间乘以空间),但可以通过patchify转换为2D图像再展平为1D序列;音频的mel-spectrogram本质上可以看作灰度图像,因此可以采用类似的处理流程。这种方法使得两种模态能够在统一的1D离散潜在空间中融合。技术路线上,作者基于LARP的query-based tokenizer,将其从单模态扩展到双流架构:两个模态各自通过独立的forward pass处理,但共享编码器-解码器的核心参数,仅保留模态特定的归一化层和可学习查询。这种设计既能捕获模态特定信息,又能通过共享参数实现隐式跨模态交互。为了训练这个架构,作者设计了VFAL分层策略和表征对齐学习,逐步建立各模态的重构能力并增强跨模态对齐。
AVTok的核心创新点在于双流query-based transformer架构与分层训练策略的结合。与现有方法的本质区别在于:第一,架构层面,现有音视频生成模型使用双tokenizer设计(视频tokenizer加音频tokenizer),而AVTok使用单一tokenizer处理两种模态,通过模态特定的查询向量(视频n=1024个holistic query,音频q=128个holistic query)在共享架构中区分模态。第二,表示层面,现有视频tokenizer大多使用3D时空潜在表示,而AVTok选择1D离散表示,这与音频的1D波结构天然兼容,更利于自回归生成。第三,训练层面,VFAL策略首次提出渐进式建立模态能力的范式,不同于端到端联合训练。这种组合使得AVTok能够在统一空间中同时实现高质量的音视频重构,为下游生成任务提供坚实基础。
方法步骤详情
AVTok的完整处理流程包含四个步骤。第一步是数据预处理和patchify,对于视频输入V,使用降采样因子f_T=4, f_H=8, f_W=8将其转换为E_v的patch embedding,其中m是总token数。对于音频,使用22kHz采样率的原始波形,转换为mel-spectrogram A,然后使用f_M=16, f_L=16转换为E_a的patch embedding。第二步是编码和量化,对于每个模态i,使用模态特定的holistic query Q_i^L(视频Q_v^L,n=1024;音频Q_a^L,q=128)与patch embedding拼接后输入共享编码器,得到Z_i = E(Q_i^L || E_i),然后量化前j个query对应的部分得到离散token x_i = Q(Z_i^(1:j))。第三步是解码,使用模态特定的patch query Q_i^P(视频Q_v^P,m=1024;音频Q_a^P,p=120)与去量化后的hat Z_i = Q^(-1)(x_i)拼接,输入共享解码器得到hat E_i,最后重塑并重构得到hat V和hat A。第四步是训练,采用VFAL三阶段策略:Stage 1仅训练视频流,损失为L_vrec;Stage 2冻结视频和共享模块,仅训练音频流,损失为L_arec;Stage 3联合训练,总损失为L = lambda_1 L_vrec + lambda_2 L_arec + lambda_3 L_rep + lambda_4 L_prior,其中L_rep是使用音频-视觉基础模型M_F的表征对齐损失,L_prior是AR prior的next token prediction损失。
技术新颖性
AVTok的技术新颖性体现在四个方面。首先,这是首次提出统一的音视频tokenization任务和解决方案,填补了单模态tokenizer和双tokenizer音视频生成模型之间的空白。这个方向本身具有开创性,因为它重新思考了多模态表示学习的基本问题。其次,双流query-based架构是新颖的架构设计,不同于现有视频tokenizer(如LARP使用单流)和音频-视觉预训练模型(如AudioCLIP、VideoCLIP使用patch-wise局部约束),AVTok使用learned queries捕获holistic信息,同时通过双流设计保持模态特定性。第三,VFAL训练策略是新颖的训练范式,它认识到模态信息密度的不对称性(视觉信息远比听觉信息丰富),通过课程学习式的渐进训练避免一个模态主导另一个模态的学习。消融实验显示VFAL将rFAD从9.38降低到5.93,证明其有效性。最后,表征对齐损失L_rep利用预训练音频-视觉基础模型作为语义对齐的桥梁,这是一种巧妙的迁移学习应用,增强了AVTok的跨模态对齐能力。
实验结果
论文通过四个实验验证了AVTok的有效性。重构实验在TAVGBench和VGGSound数据集上进行,使用16帧128乘以128分辨率视频和80乘以384 mel-spectrogram音频。结果显示AVTok在视频重构上达到PSNR=25.62、rFVD=12.80、LPIPS=0.126,超越了LARP(PSNR=24.53、rFVD=14.24、LPIPS=0.137)等视频tokenizer;在音频重构上达到SI-SDR=23.09、rFAD=5.93、MR-STFT=1.523,与SpectralCodec(SI-SDR=29.30、rFAD=5.56、MR-STFT=1.514)等音频codec性能相当。这证明了统一tokenization在两个模态上都能达到competitive性能。下游生成实验在VGGSound上评估三种任务:A2V(音频到视频)上AVTok-A2V达到gFVD=150.26、DeSync=1.317、IB-Score=0.143,相比TempoTokens(gFVD=786.61、DeSync=1.359、IB-Score=0.132)在质量上有显著提升;V2A(视频到音频)上AVTok-V2A达到gFAD=49.47、DeSync=1.239、IB-Score=0.249,相比VinTAGe(gFAD=80.06、DeSync=1.294、IB-Score=0.044)和V-AURA(gFAD=126.92、DeSync=0.967、IB-Score=0.231)在音频质量和语义对齐上都有优势;cJAVG(类别条件联合生成)上AVTok-cJAVG达到gFVD=138.80、gFAD=56.58、DeSync=1.319、IB-Score=0.206,相比JavisDiT(gFVD=1040.28、gFAD=268.51、DeSync=1.330、IB-Score=0.195)和Ovi(gFVD=972.65、gFAD=129.02、DeSync=0.814、IB-Score=0.172)在质量上有巨大提升,同时保持了相当的对齐性。消融实验在Table 3中展示了各组件的贡献:相比vanilla单流设计(rFVD=14.87、rFAD=10.26),双流AVTok(rFVD=12.80、rFAD=5.93)有明显提升;移除VFAL后(rFVD=13.19、rFAD=9.38)性能下降;移除L_rep后(rFVD=12.90、rFAD=8.48)对齐性变差;移除L_prior后(rFVD=10.63、rFAD=3.47)重构质量最好但生成性能最差(gFVD=266.82、gFAD=67.84),验证了AR prior对于下游任务的重要性。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| Video Reconstruction | rFVD (Fréchet Video Distance, lower is better) | 12.80 | LARP (14.24) | 10.1% reduction |
| Video Reconstruction | PSNR (Peak Signal-to-Noise Ratio, higher is better) | 25.62 | LARP (24.53) | 4.4% improvement |
| Audio Reconstruction | rFAD (Fréchet Audio Distance, lower is better) | 5.93 | UniCodec (6.73) | 11.9% reduction |
| Audio-to-Video Generation | gFVD (generated FVD, lower is better) | 150.26 | TempoTokens (786.61) | 80.9% reduction |
| Video-to-Audio Generation | gFAD (generated FAD, lower is better) | 49.47 | VinTAGe (80.06) | 38.2% reduction |
| Class-conditional Joint AV Generation | gFVD (lower is better) | 138.80 | JavisDiT (1040.28) | 86.7% reduction |
| Class-conditional Joint AV Generation | gFAD (lower is better) | 56.58 | Ovi (129.02) | 56.1% reduction |
局限与改进
作者在论文中明确承认了一些局限性。首先是模态信息不平衡问题,尽管VFAL策略缓解了视觉信息丰富性对音频学习的压制,但这个问题并未完全解决,可能在更复杂的数据集中更加突出。其次是跨模态交互的隐式性,AVTok通过共享参数实现隐式融合,相比显式的cross-attention机制可能限制了对齐精度。第三是计算复杂度,虽然相比双tokenizer设计有所简化,但双流forward pass和AR prior训练仍然需要大量计算资源。除此之外,我们还可以观察到一些额外局限:首先,实验仅在VGGSound等相对简单的数据集上进行,对长视频、复杂场景的泛化能力未验证;其次,统一码本大小固定,可能限制表示能力;第三,评估主要依赖客观指标(FVD、FAD、DeSync),缺乏用户研究和主观评估;第四,AR生成的推理速度受token数量限制(1152个token),实时性可能不如基于扩散或flow matching的方法。
独立分析的弱点
AVTok存在几个可以改进的弱点。首先是显式跨模态交互的缺失,当前设计通过共享参数实现隐式融合,但可以引入显式的cross-modal attention机制或对比学习目标,增强模态间的对齐性。具体来说,可以在编码器中加入cross-modal attention层,让视频patch embedding直接attend到音频patch embedding,反之亦然,这样能够建立更直接的跨模态关联。其次是token数量的优化,当前使用固定的1024个视频holistic query和128个音频holistic query,可以借鉴AdapTok的自适应token分配机制,根据内容复杂度动态调整token数量,在保持质量的同时提高效率。第三是VFAL策略的泛化性,当前策略假设视频信息更丰富,但在某些数据集(如以音乐为主)中可能不成立,可以设计数据驱动的策略自动确定训练顺序或并行渐进训练。第四是评估指标的全面性,当前主要使用客观指标,可以增加主观评估、语义一致性评估和任务特定评估(如视频问答、音频分类),更好地反映实际应用价值。
未来方向
作者在结论中提出希望鼓励统一音视频tokenization方向的进一步探索,为构建统一的音视频生成大型多模态模型铺平道路。基于AVTok的成果,可以延伸出多个研究方向。首先是扩展到更多模态,如文本、深度、触觉等,构建真正统一的多模态tokenizer,这需要对现有架构进行更大规模的扩展,可能需要模块化设计和层次化码本。其次是与大型语言模型(LLM)的深度集成,AVTok的1D离散token天然适合LLM处理,可以探索将其作为视觉和听觉的通用接口,实现真正的多模态理解和生成统一框架。第三是长视频和长音频的处理,当前方法局限于16帧短视频,可以研究时序分块、层次化表示或latent扩散等技术在长序列上的应用。第四是实时生成优化,可以研究并行AR生成、缓存优化、模型蒸馏等技术,将AVTok应用于实时音视频交互场景。第五是可控生成,引入更细粒度的控制(如风格、情感、摄像机角度),通过条件注入、引导机制或adapter模块实现可控的音视频生成。
复现评估
论文提供了项目页面(https://hkust-longgroup.github.io/AVTok/),但未明确说明代码开源状态。从实现细节来看,复现难度中等偏高。数据方面,使用了TAVGBench和VGGSound公开数据集,相对容易获取。计算资源方面,模型参数量约208.4M,训练需要大量GPU资源,具体配置未详细说明,但从基于LARP和需要训练AR prior来看,可能需要多卡GPU集群。实现细节方面,论文提供了关键的配置参数(降采样因子、query数量、损失权重等),但某些细节(如具体的网络层数、注意力头数、优化器设置等)未完全描述,需要参考LARP原论文或开源代码。评估方面,作者提供了详细的指标计算方法和基准模型对比,复现实验相对容易。总体而言,如果作者开源代码,复现难度将显著降低;否则需要根据论文描述从头实现,需要相当的工程和调优经验。
论文图表
这张图展示了AVTok的核心贡献和整体架构。子图展示了AVTok的框架示意图,包括双流编码器、统一码本和双流解码器,清晰地表达了如何将音视频对编码为1D潜在表示。子图通过对比图展示了AVTok在视频重构(与LARP、AdapTok、OmniTokenizer对比)和音频重构(与SpectralCodec、UniCodec、WavTokenizer对比)上的competitive性能,用柱状图直观呈现了指标对比。子图展示了三个下游生成任务的示例:音频到视频生成(婴儿笑声到视频)、视频到音频生成(消防车警报到音频)、类别条件联合生成(木鸟啄树),每个示例都包含生成的音视频帧,直观展示了AVTok的实用价值。
Fig. 1是论文最重要的概览图,它在一页内完整展示了AVTok的三个核心方面:架构设计、性能表现、应用价值。对于读者来说,这张图提供了快速理解论文贡献的入口,是后续深入阅读的导航图。
这张图通过对比方式清晰地阐述了研究动机。左侧展示了现有方法的架构:视频和音频各自使用独立的预训练tokenizer,产生的嵌入空间在t-SNE可视化中显示为分离的类簇,视频嵌入和音频嵌入占据不同区域,表明存在表示间隙。右侧展示了AVTok的目标架构:统一tokenizer将两种模态编码到共享的token空间,t-SNE可视化中视频和音频嵌入在共享空间中紧密交织,表示语义对齐。
Fig. 2是理解研究动机的关键,它将抽象的表示间隙问题可视化,直观展示了为什么统一tokenization是必要的。t-SNE对比图特别有说服力,让读者立即理解问题的本质和解决方案的优势。