← 返回 2026-05-19

WavFlow:直接在原始波形空间生成高保真音频 WavFlow: Audio Generation in Waveform Space

Feiyan Zhou, Luyuan Wang, Shoufa Chen, Zhe Wang, Zhiheng Liu, Yuren Cong, Xiaohui Zhang, Fanny Yang, Belinda Zeng 📅 2026-05-18 👍 10 2026-07-13 08:36
MMDiT x-prediction 原始波形建模 流匹配 视频转音频 音频生成

无需音频压缩器,直接在原始波形空间用流匹配生成高保真音频

前置知识

潜在空间压缩(Latent-space Compression)

现代音频/视频生成的主流范式:先用一个在大规模无标注音频上预训练好的编码器(如 EnCodec、SoundStream 这类神经编解码器,或音频 VAE)把原始波形压缩到一个低维、紧凑的潜在表示,再让扩散/流匹配模型在压缩空间里生成,最后用解码器还原成波形。这样做的好处是序列短、训练稳定,但坏处是输出质量被解码器重建保真度「封顶」,高频瞬态、精细相位信息等往往在瓶颈层丢失且无法恢复。

理解「为什么要绕开压缩层」是读懂本文动机的关键。WavFlow 的全部论证都建立在「压缩不是高质量合成的必要条件」这一反直觉主张之上。

条件流匹配(Conditional Flow Matching)

一种生成式建模框架,把噪声分布到数据分布之间的变换建模成一条连续的概率流。给定噪声 $x_0 \sim \mathcal{N}(0, I)$ 和干净数据 $x_1$,构造线性插值 $x_t = (1-t)x_0 + t x_1$,$t \in [0,1]$,对应的理想速度场为 $v^*(x_t,t) = x_1 - x_0$。模型学习一个速度场 $v_\theta(x_t, t, c)$ 把噪声「搬运」到数据,推理时通过求解 ODE $dx_t/dt = v_\theta$ 得到样本。相比传统扩散,目标函数更简单、训练更稳定。

WavFlow 的整套生成框架建立在条件流匹配之上,并且本文对预测目标(x-prediction)和损失公式(v-loss)的选择是核心创新之一。

x-prediction 与流形假设(Manifold Hypothesis)

在扩散/流匹配里,模型可以预测噪声(noise-pred)、速度(v-pred)或干净数据(x-pred)。流形假设指出:真实干净数据只分布在一个低维流形上,而噪声/速度却充满整个高维空间,因此预测低维的干净数据 $\hat{x}_1 = f_\theta(x_t,t,c)$ 本质上比预测全空间的噪声更容易学。WavFlow 借鉴了视觉领域 JiT 的工作,把 x-prediction 引入音频原始波形生成,让网络聚焦于恢复低维数据结构而非建模全空间噪声。

x-prediction + v-loss 的组合是本文最关键的训练技巧,直接决定了在原始波形这种高维、低能量信号上能否稳定优化。

波形 Patchify(Waveform Patchify)

借鉴 ViT 的图像 patchify 思想:把长度为 $T$ 的一维原始波形重排成一个 $C \times D$ 的二维 token 网格,每一行(长度 $D$)作为一个 token,类似于图像的一个 patch。$D$ 表示每个 token 包含的采样点数,定义了时间粒度;$C$ 是 token 总数。这一步完全无参数、无损,生成后用 unpatchify 还原回一维波形。WavFlow 默认 $D=200$,在 16kHz、8 秒片段下得到 $C=640$ 个 token,对应 12.5 毫秒粒度。

这是把高维一维波形喂给 Transformer 的关键工程创新,patch 粒度 $D$ 的选择直接在计算复杂度 $O(C^2)$ 和信息密度之间权衡,是全文最重要的超参数之一。

多模态扩散 Transformer(MMDiT)

MMDiT(Multimodal Diffusion Transformer)是把多个模态流(音频、视频、文本)的 token 拼进同一个注意力序列里做联合建模的架构,由 Sora/SD3 系列推广开来。WavFlow 采用 $L_{joint}$ 个联合块做跨模态融合,再接 $L_{fused}$ 个纯音频块做波形细化,通过 AdaLN 调制注入时间步和条件,用 RoPE 注入相对位置。视觉条件来自冻结的 CLIP/Synchformer,文本来自 CLIP 文本编码器。

MMDiT 是 WavFlow 的主干,理解它才能理解作者如何把「条件控制」嫁接到「原始波形生成」上,以及 VT2A 和 T2A 两个任务如何共用同一套架构。

分类器自由引导(Classifier-Free Guidance, CFG)

推理时提升生成质量的技巧:训练时以一定概率(本文 10%)把条件替换为可学习的空嵌入,推理时用 $\hat{v}_\theta = (1+w)v_\theta(x_t,t,c) - w v_\theta(x_t,t,\emptyset)$ 把条件和无条件两条预测线性组合,引导强度 $w$ 越大越向条件靠拢。WavFlow 发现 $w=4.5$ 时单样本质量最好(IS、IB 最高),但 $w=7.0$ 会让 FD 急剧变差、牺牲多样性。

CFG 既是推理质量的关键旋钮,也是 WavFlow 能「零成本」从 VT2A 扩展到 T2A 的机制(把视觉条件置零即可),理解它对解读实验结果必不可少。

研究动机

当前视频转音频(VT2A)和文本转音频(T2A)的几乎所有 SOTA 方法(Frieren、V2A-Mapper、MMAudio、HunyuanVideo-Foley 等)都遵循同一套「潜在空间配方」:先用预训练的神经编解码器或音频 VAE 把原始波形压缩成潜在表示,再用多模态扩散/流匹配 Transformer 学习给定视觉和文本特征下的条件分布,最后用解码器把潜在表示还原成波形。这套范式虽然有效,但留下一个根本问题:潜在空间压缩真的有必要吗?依赖一个独立的预训练 tokenizer 不仅增加了流水线复杂度,更致命的是把最终合成质量死死限制在该 backbone 的重建保真度上——高频瞬态、精细相位信息等关键细节在潜在瓶颈阶段就被破坏,事后任何后处理都无法恢复。这种内在的有损性促使作者追问:能否直接在原始波形空间生成?

本文的目标是本文的目标是构建一个名为 WavFlow 的生成框架,完全跳过音频压缩器,直接在原始波形空间(raw waveform space)端到端地合成高保真音频,同时保持精确的时间同步与语义对齐。具体而言,作者希望证明「潜在空间压缩并非高质量音频合成的必要前提」,并用一个更简单、更可扩展的替代方案,在标准的 VGGSound(VT2A)和 AudioCaps(T2A)基准上达到甚至超越成熟的潜在空间方法,从而为多模态音频生成开辟一条无需 tokenizer/VAE 的新路径。

与已有工作不同的是,直接在原始波形空间生成有三个长期未被克服的根本难题,恰恰构成本文的独特切入点。第一,原始波形维度极高(8 秒、16kHz 就是 128000 个采样点),直接建模序列在计算上极富挑战。第二,波形振幅动态范围大、能量却高度集中在零附近(RMS 普遍低于 0.2),导致训练时信噪比极差,流匹配目标难以优化。第三,配对的视频-音频数据稀缺,连广泛使用的 VGGSound 也只有约 20 万样本(500 小时),对必须从零学习声学结构、时间动力学和跨模态对齐的原始波形模型来说远远不够。作者没有回避这些难题,而是用波形 patchify、幅度提升、x-prediction 和大规模数据策划(约 500 万高质量三元组)逐一破解,填补了「多模态原始波形生成」这一空白。

核心方法

WavFlow 的整体思路可以用一句直觉概括:既然图像领域能用 patchify + x-prediction 直接在像素空间生成高质量图像(如 JiT),那么把同样的「流形假设 + patch 化」哲学搬到一维音频波形上,理应也能绕开压缩瓶颈。技术路线上,WavFlow 以多模态扩散 Transformer(MMDiT)为主干,给定多模态条件 $c$(视频+文本),用条件流匹配直接在观测空间生成原始波形 $x \in \mathbb{R}^T$。三个关键设计协同稳定了原始空间训练:幅度提升(RMS 归一化 + 全局缩放)把低能量信号抬到与高斯先验匹配的尺度;波形 patchify 把一维波形重排成 $C \times D$ 的二维 token 网格喂给 Transformer;x-prediction 让网络只聚焦于恢复低维数据流形。模型分 WavFlow-M(约 624M 参数,$L_{joint}=4, L_{fused}=8$)和 WavFlow-L(约 1.03B 参数,$L_{joint}=7, L_{fused}=14$)两个尺度,隐藏维度 $d=896$、14 个注意力头。

核心创新是把三件看似独立的「手术」有机组合,从而首次让原始波形多模态生成变得可行。其一是波形 patchify:把 128000 个采样点重排成 $C \times D$(默认 $640 \times 200$)的 token 网格,使 Transformer 既能处理高维输入,又能像 ViT 一样在每个 patch 上聚合信息,且 patch 与 unpatchify 完全无参数、无损,无需任何学习到的解码器或声码器。其二是 x-prediction 配合 v-loss:网络直接预测干净波形 $\hat{x}_1 = f_\theta(x_t, t, c)$,再按 $v_\theta = (\hat{x}_1 - x_t)/(1-t)$ 反推速度,既满足流形假设让优化更易,又把目标锚定在流匹配速度场上以兼顾高频保真度与生成多样性。其三是幅度提升 $x_{lift} = s_a \cdot \text{clamp}(\frac{r^*}{\text{rms}(x)} x, -1, 1)$($r^*=0.33, s_a=3.0$),把集中在零附近的低能量波形抬升到 $[-3,3]$ 范围,与单位方差高斯先验对齐,使去噪目标变得可优化。与已有方法的本质区别在于:彻底删除了「编码器-解码器」这一中间层,证明压缩不再是天花板。

方法步骤详情

完整流程分训练与推理两阶段。训练前先做数据策划:从专有媒体数据(MovieGen 训练子集)和开源 VGGSound/AudioCaps/Freesound 出发,经三阶段流水线筛选——丢弃时长不足 8 秒或静音占比超 80% 的片段、用 audiobox-aesthetics 滤掉 PQ<6.0 的低质样本、用 PANNs 剔除分类置信度最低 10%,再按 VGGSound 类别分布平衡出约 500 万高质量片段。训练时,单条音频先转单声道,做幅度提升把 RMS 拉到 0.33 再放大 3 倍;接着波形 patchify 成 $640 \times 200$(16kHz)或 $1764 \times 200$(44.1kHz)的网格;按 $x_t = (1-t)x_0 + t x_1$ 构造插值,$t$ 服从 logit-normal 分布,模型预测 $\hat{x}_1$,损失 $\mathcal{L} = \mathbb{E}[\| \frac{\hat{x}_1 - x_t}{1-t} - \frac{x_1 - x_t}{1-t} \|^2]$;视觉/文本条件各以 10% 概率替换为可学习空嵌入以支持 CFG。全局条件 $c_g$ 由 CLIP 视觉、文本均值池化特征与时间步嵌入相加给出语义引导,帧级条件 $c_e$ 来自冻结 Synchformer 的同步特征(含可学习分段位置编码,最近邻插值到长度 $C$),通过 AdaLN 注入。推理时从波形 token 空间采高斯噪声,用 Euler 求解器走 50 步、CFG 强度 4.5 生成 token 网格,再 unpatchify 还原成一维波形,并按 $1/s_a$ 反缩放、归一化到 $-23$ LUFS 保证播放响度舒适。

技术新颖性

新颖性体现在「首次」与「反共识」两点。在音频生成领域,原始波形建模此前只存在于神经声码器(WaveNet、WaveGrad、DiffWave)这类从局部频谱特征重建波形的场景,缺乏把全局语义线索直接映射到原始波形的能力,更没人成功把它用于大规模多模态生成。WavFlow 是首个在多模态扩散 Transformer 框架内、直接在原始波形空间端到端生成高保真音频并达到 SOTA 级别的方法,从工程上推翻了「必须有压缩器」的行业共识。技术层面,它把视觉 JiT 的 x-prediction 流形假设、ViT 的 patchify 思想、以及针对音频特性的幅度提升三者系统化整合;并通过消融严谨证明:x-pred + v-loss 优于 v-pred(FD_PaSST 63.05 vs 77.19)、噪声偏移(noise shift)在音频上反而有害($s=5$ 时 FD_PaSST 从 63.05 飙到 92.21,与图像完全相反)、patch 粒度 $D=200$ 是计算效率与质量的饱和点——这些针对原始波形的负向发现本身也极具价值。

WavFlow 架构。原始音频表示为二维 patch 网格,经过一系列联合块和融合 Transformer 块,利用多模态条件实现精确的语义和时间控制。
Figure 2: WavFlow 架构。原始音频表示为二维 patch 网格,经过一系列联合块和融合 Transformer 块,利用多模态条件实现精确的语义和时间控制。
从原始数据集到最终高质量、平衡训练混合物的多阶段数据策划概览。
Figure 3: 从原始数据集到最终高质量、平衡训练混合物的多阶段数据策划概览。
15 段随机音频片段预处理前(左)后(右)的振幅直方图。原始波形尖锐零中心、能量差异大;预处理后振幅更一致、被抬升。
Figure 6: 15 段随机音频片段预处理前(左)后(右)的振幅直方图。原始波形尖锐零中心、能量差异大;预处理后振幅更一致、被抬升。
波形 patchify 示意。一维波形被重排成 $C \times D$ 的二维 token 网格;对不能整除的长度自然做零填充,unpatchify 时截断恢复原始 $T$ 个采样点。
Figure 7: 波形 patchify 示意。一维波形被重排成 $C \times D$ 的二维 token 网格;对不能整除的长度自然做零填充,unpatchify 时截断恢复原始 $T$ 个采样点。

实验结果

实验从 VT2A、T2A、消融三个维度系统验证。在 VGGSound-Test(1.5 万视频)的 VT2A 任务上(Table 1),即便是中等规模的 WavFlow-M-16kHz(624M)就已超越多个成熟潜在空间系统,FD_PANNs 6.37、IS_PANNs 17.24、DeSync 0.47;放大到 WavFlow-L-16kHz(1.03B)则在分布保真度上反超 SOTA 的 MMAudio-L-44.1kHz(FD_PaSST 59.98 vs 60.60),并匹配其感知与对齐指标(IS_PANNs 17.40、DeSync 0.44);进一步在 16kHz 检查点上微调的 WavFlow-L-44.1kHz 把 FD_PaSST 推到全部方法最佳的 55.82。在 AudioCaps-Test(4800 样本)的 T2A 任务上(Table 2),尽管 WavFlow-M-16kHz 是统一模型而非 T2A 专用,仍取得迄今最佳的 FD_PANNs 10.63 和 IS_PANNs 12.62,超过专门为 T2A 设计的 GenAU-Large 与 MMAudio。消融实验(Tables 3-5)显示:patch 粒度与数据规模存在耦合,低数据量下小 $D$(细粒度)是唯一有效杠杆(200K 时 $D$ 从 512 降到 200 使 FD_PaSST 136.45→90.24),而大数据能部分补偿大 $D$;x-prediction 全面优于 v-prediction;RMS 归一化 + 3.0× 缩放组合在多样性与保真度上最优(IS_PANNs 15.58)。在 MovieGen-Audio-Bench 上 WavFlow 的 IS 8.95、DeSync 0.77 也匹敌或超越 MMAudio/MovieGen。整体结论:大规模高质量数据下,端到端原始波形合成可以消除中间潜在表示而不牺牲性能。

在 VGGSound-Test 上的视频-文本转音频(VT2A)SOTA 对比。FD/KL/DeSync 越低越好,IS/IB/CLAP 越高越好,最佳加粗、次佳下划线。
Table 1: 在 VGGSound-Test 上的视频-文本转音频(VT2A)SOTA 对比。FD/KL/DeSync 越低越好,IS/IB/CLAP 越高越好,最佳加粗、次佳下划线。
在 AudioCaps-Test 上的文本转音频(T2A)SOTA 对比。
Table 2: 在 AudioCaps-Test 上的文本转音频(T2A)SOTA 对比。
不同数据规模下 patchify 粒度($C \times D$)对波形合成的影响,展示时间分辨率与数据多样性的权衡。
Table 3: 不同数据规模下 patchify 粒度($C \times D$)对波形合成的影响,展示时间分辨率与数据多样性的权衡。
WavFlow 流匹配目标的消融。结果表明 x-prediction 配 v-loss 在高层特征相似性与生成多样性间最优。
Table 4: WavFlow 流匹配目标的消融。结果表明 x-prediction 配 v-loss 在高层特征相似性与生成多样性间最优。
原始波形预处理消融。结果表明 RMS 归一化配 3.0× 缩放性能最优。
Table 5: 原始波形预处理消融。结果表明 RMS 归一化配 3.0× 缩放性能最优。
所有 WavFlow 变体的训练超参。除 WavFlow-L-44k(44.1kHz)外均用 16kHz。
Table 6: 所有 WavFlow 变体的训练超参。除 WavFlow-L-44k(44.1kHz)外均用 16kHz。
数据混合物与字幕粒度对 VT2A 的影响(WavFlow-M-16k,1M 规模,VGGSound-Val 评测)。
Table 7: 数据混合物与字幕粒度对 VT2A 的影响(WavFlow-M-16k,1M 规模,VGGSound-Val 评测)。
噪声偏移对 VT2A 的影响(WavFlow-M-16k,1M,VGGSound-Test)。
Table 9: 噪声偏移对 VT2A 的影响(WavFlow-M-16k,1M,VGGSound-Test)。
推理超参(CFG 强度、ODE 步数)对 VT2A 的影响(WavFlow-M-16k,1M,VGGSound-Val)。
Table 10: 推理超参(CFG 强度、ODE 步数)对 VT2A 的影响(WavFlow-M-16k,1M,VGGSound-Val)。
MovieGen-Audio-Bench 评测。因无参考音频,采用无参考指标 IS/CLAP/IB/DeSync。
Table 11: MovieGen-Audio-Bench 评测。因无参考音频,采用无参考指标 IS/CLAP/IB/DeSync。
不同数据规模下 patchify 粒度与关键指标的关系。注意 D=512 扩展到 3M 时性能退化,表明低粒度 token 的容量瓶颈。
Figure 4: 不同数据规模下 patchify 粒度与关键指标的关系。注意 D=512 扩展到 3M 时性能退化,表明低粒度 token 的容量瓶颈。
VGGSound 验证集上指标随训练 epoch 的变化,横跨 200K/1M/3M/5M 四个数据规模。虚线标示最终采用的收敛点。
Figure 5: VGGSound 验证集上指标随训练 epoch 的变化,横跨 200K/1M/3M/5M 四个数据规模。虚线标示最终采用的收敛点。
查看结构化数据
任务指标本文基线提升
视频转音频 VT2A(VGGSound-Test) FD_PaSST(越低越好) WavFlow-L-44.1kHz 达 55.82,WavFlow-L-16kHz 达 59.98,均为全场最佳或次佳 SOTA 的 MMAudio-L-44.1kHz 为 60.60 最佳配置 55.82 相比 MMAudio 的 60.60 降低约 4.78,证明原始波形建模在分布保真度上反超潜在空间 SOTA
视频转音频 VT2A(VGGSound-Test) DeSync 时间同步(越低越好) WavFlow-L-16kHz 为 0.44,与 MMAudio 并列最佳 V2A-Mapper 1.23、Frieren 0.85、MovieGen 1.00 从 V2A-Mapper 的 1.23 降到 0.44,时间同步大幅领先,证明直接波形生成不损害视听对齐
文本转音频 T2A(AudioCaps-Test) FD_PANNs / IS_PANNs WavFlow-M-16kHz:FD_PANNs 10.63、IS_PANNs 12.62 前 SOTA MMAudio-L-44.1kHz FD_PANNs 15.04、IS_PANNs 12.08;GenAU-Large FD 16.51 FD_PANNs 从 15.04 降到 10.63(降约 29%),IS_PANNs 从 12.08 升到 12.62,统一模型在 T2A 上也刷新最佳
VT2A 跨域泛化(MovieGen-Audio-Bench) IS / DeSync(无参考指标) WavFlow-L:IS 8.95、DeSync 0.77 MMAudio IS 8.40、DeSync 0.77;MovieGen IS 8.89、DeSync 1.00 IS 超越 MMAudio 和 MovieGen,DeSync 并列最佳,证明对 AI 合成视频同样泛化良好

局限与改进

作者明确承认的局限是:WavFlow 目前不支持显式的语音或歌唱合成,因为生成的发声不构成有意义的语言;要扩展到这些领域需要更细的语言粒度和更大的语音数据集。此外,作者在附录里坦诚了若干「反直觉」的边界:噪声偏移(noise shift)在音频上不仅无效还会持续伤害所有指标($s=5$ 时 FD_PaSST 从 63.05 升到 92.21),与图像领域截然相反;patch 粒度 $D=512$ 在大数据(3M)下反而遭遇容量瓶颈(FD_PaSST 从 1M 的 81.81 退化到 89.51)。从外部观察,本文的 5M 训练数据来自 Meta 专有的 MovieGen 子集,对学术复现极不友好;同时 44.1kHz 模型是 16kHz 的微调而非从零训练,高保真并非纯原始空间学习的产物;另外大规模训练(全局 batch 10752、400 epoch、H100 集群)的成本未被讨论,对算力受限的研究者门槛很高。

独立分析的弱点

第一,数据可得性是最大短板:核心的 500 万高质量视频-文本-音频三元组来自 MovieGen 专有数据,普通研究者无法获取,且开源替代(VGGSound 仅 200K)作者自己在消融中证明会导致 FD_PaSST 退化到 90 以上,这使「原始波形生成可行」的结论与大规模私有数据深度绑定。改进方向是构建开源的千万级多模态音频数据集,或研究更高效的小数据正则化。第二,高保真(44.1kHz)结果依赖 16kHz 检查点的监督微调,并非纯原始空间训练的胜利,严格意义上削弱了「压缩非必需」的论证力度;可尝试从零训练 44.1kHz 模型以做更干净的对照。第三,序列长度随采样率平方增长($C$ 从 640 涨到 1764,复杂度 $O(C^2)$),长音频(>8 秒)或更高采样率的扩展性未被讨论,patchify 的计算瓶颈会凸显;可借鉴层级化设计或稀疏注意力。第四,全篇缺少语音/歌唱能力,且未报告推理延迟、显存占用与生成时长比(RTF),工程实用性存疑;可补充效率评测并做少步蒸馏。第五,消融中噪声偏移、T2A 跨任务的视觉置零策略等结论较为初步,缺乏对其失效机理的更深入理论解释。

未来方向

作者提出的未来方向是把框架扩展到联合建模环境声与人类语音/歌唱,需要更大规模语料和细粒度语言描述。基于本文成果可延伸的方向包括:一是把 WavFlow 与语音识别/合成结合,赋予模型语言级控制;二是探索更细 patch($D<160$)或层级化 patchify 以支持长音频和高采样率;三是把「压缩非必需」的范式迁移到音乐生成、3D 声场、空间音频等领域,验证其普适性;四是研究少步蒸馏(如 4-8 步)与一致性模型在原始波形空间的应用,目前 50 步 ODE 推理偏慢;五是开源中等规模多模态音频数据集,让学术界能独立验证「数据规模 vs. 压缩器」的权衡边界;六是把 CFG 与 T2A 的视觉置零机制推广为更灵活的条件组合(如音频续写、声学场景控制)。此外,噪声偏移在音频失效这一现象值得理论分析,或能催生针对音频特性的新训练目标。

复现评估

复现难度整体偏高。代码与项目页已开源(https://github.com/facebookresearch/WavFlow),架构细节(两个尺度的层数、$d=896$、14 头)、训练超参(AdamW $\beta=(0.9,0.95)$、常数学习率 $10^{-4}$、EMA 0.9999、梯度裁剪 1.0、BF16、batch 10752/8192/1536)和关键消融(Table 6/8/9/10)都交代得相当完整,纯算法部分可复现。但最大障碍在于数据:核心训练集是 Meta 专有的 MovieGen 媒体数据子集(5M VT2A + 1M T2A),不对外发布,而作者自己证明仅用开源 VGGSound/AudioCaps(20 万-30 万)性能会大幅下滑,无法达到论文水准。算力方面,需要 H100 集群跑 400-650 epoch、全局 batch 上万,对中小团队极不友好。44.1kHz 模型还需 16kHz 收敛检查点做 SFT。综合判断:模型结构、损失、预处理高度可复现,但数据与算力构成实质壁垒,普通研究者最多能复现小规模(200K-1M)的相对趋势,难以复现最终 SOTA 数字。