面向视频生成语音的层次化Codec扩散Transformer Hierarchical Codec Diffusion for Video-to-Speech Generation
首个显式建模语音token层级先验的离散扩散VTS模型。
前置知识
残差矢量量化(RVQ)Codec
一种分层语音编解码器,先由编码器把语音压成连续特征,再通过多层矢量量化 ($VQ_1 \to VQ_2 \to \cdots \to VQ_{12}$) 逐层残差细化,每层输出 1024 大小的码本索引。低层 token 承载说话人内容与音色,高层承载韵律,共 12 层。
HiCoDiT的核心正是把RVQ的12层切分为低层(x^r1:r2)与高层(x^r3:r12),再分别注入不同视觉条件。不理解VQ层级语义就理解不了论文为何要分块。
离散扩散模型(DDM)与掩码预测
把数据视作离散状态,通过连续时间离散马尔可夫链加噪到 $[\text{MASK}]$,训练 score 网络估计任意时刻去掩码的具体分数 concrete score,采样时用 Euler 方法从全 $[\text{MASK}]$ 逐步恢复真实 token,代表工作 SEDD、MD4。
HiCoDiT的score network s_θ就是离散扩散的sθ,优化目标是DSE(去噪分数熵),推理靠Euler sampler 64步从全掩码恢复语音token。必须理解concrete score与transition rate矩阵Q̅的关系。
自适应实例层归一化(AdaLN)与DiT
DiT 用 AdaLN 把外部条件编码为缩放 $\gamma$ 与偏置 $\beta$,公式 $h'=(1+\gamma)\cdot\frac{h-\mu}{\sigma}+\beta$。本文提出 dual-scale AdaLN,通道维度做全局风格调制,时间维度做局部韵律调制。
HiCoDiT用dual-scale AdaLN把身份与情感两类'类级别'视觉条件注入高层block,而把唇动'帧同步'条件用通道拼接注入低层block。理解两种调制粒度差异是读懂条件设计的关键。
音视频语音表征AV-HuBERT
通过掩码多模态聚类预测的自监督预训练模型,联合学习人脸视频与语音的共享表示,常用于唇读与VTS中抽取内容对齐特征,本文用其Large版本最后一层隐藏态作为唇动表征clip。
HiCoDiT的lip adapter直接用AV-HuBERT-Large最后层hidden state,因其包含最强的视听语义对齐信号。不清楚AV-HuBERT就不知道作者为何强调'content-timbre-centric'对齐。
ArcFace与Poster2人脸识别/表情识别
ArcFace是加性角度间隔损失训练的深度人脸识别模型,用于抽取身份嵌入cid;Poster2是改进的金字塔交叉融合Transformer,在AffectNet等表情基准上SOTA,用于抽取每帧情感类别再0.5秒窗口平滑得到情感序列cemo。
身份与情感两条视觉支路分别对应timbre和prosody,文中表7显示去除GE2E损失后SpkSim从56.78%掉到34.10%,替换Poster2后EmoAcc从79.41%掉到76.29%,证明这两条支路缺一不可。
研究动机
视频生成语音(VTS)旨在从无声视频推断并合成与说话人唇形、身份、情感一致的语音,应用涵盖无声电影配音、失语症辅助沟通、隐私敏感场景等。现有方法存在三类核心缺陷:一是把语音当作扁平序列,完全忽视语音从粗到细的层级结构,如NaturalL2S、Face2Speech、FaceStyleSpeech等只做内容或音色对齐;二是用mel谱回归路线(如EmoDubber)直接预测连续频谱,建模容量受限且易出现过平滑;三是即便Flow-based FTV引入层级视觉编码器,仍把视觉特征笼统注入holistic语音表征,没有显式把'低层语义token—唇动/身份'与'高层韵律token—情感'解耦对齐。这三类缺陷在LRS3基准上表现为:AlignDiT WER 31.37、FTV WER 30.37、EmoDubber WER 41.52,而LSE-C(同步置信)最高仅7.08、EmoAcc最高仅76.11,说明语音内容清晰度与韵律表达仍未同时达到最优。
本文的目标是本文目标明确:提出HiCoDiT——首个把RVQ codec的12层token切分为低层(语义/音色)与高层(韵律)、并通过离散扩散Transformer显式建模'层级—视觉条件'对齐的VTS框架。作者希望同时在三个层面取得突破:在VQ层级上,让低层VQ 1-2负责内容与音色,高层VQ 3-12负责韵律;在视觉条件上,用唇动+身份调制低层、情感调制高层,并通过dual-scale AdaLN分离全局音色与局部韵律;在性能上,目标是在LRS3上以无音频引导方式达到WER 29.41、DNSMOS 3.50、EmoAcc 79.41、LSE-C 7.15等全面SOTA,并在LRS2、真实电影OOD数据上验证鲁棒性。
与已有工作不同的是,既有方法的共同盲点是'层级—视觉条件'映射被隐式编码,HiCoDiT的差异化切入角度是:第一,首次把RVQ的token层级先验作为显式归纳偏置注入VTS,提出低/高层codec diffusion blocks对应不同视觉条件,做到层级解耦;第二,首次把mask-based discrete diffusion(SEDD/MD4范式)从零训练应用于VTS,既保留DDM的高效推理(64步Euler),又规避连续diffusion在mel谱上的频谱过平滑;第三,提出dual-scale AdaLN——通道维度归一化抓全局音色、时间维度25倍上采样抓局部韵律,精细化条件注入粒度,使'全局类属性'与'局部时序属性'在同一block内同时建模。这三点共同构成论文在VTS范式上从'扁平对齐'到'层级对齐'的关键跨越。
核心方法
HiCoDiT的核心直觉是'语音天然是层级的,视觉条件也应该按层级注入'。给定无声视频V,先抽三条视觉支路——AV-HuBERT-Large得到唇动序列clip、ArcFace得到身份嵌入cid、Poster2(0.5秒平滑)得到情感序列cemo;同时由预训练RVQ codec(MaskGCT)把真实语音编码为12层token,切分为低层x^low=x^r1:r2与高层x^high=x^r3:r12,再经离散扩散(对数线性噪声调度σ(t))逐步掩码得到x_t^low与x_t^high。score network s_θ由HiCoDiT本体担任,包含8个低层块与8个高层块,逐层迭代预测concrete score,最后由codec decoder还原波形。低层块用通道拼接融合唇动+身份(MLP调制出γ1id/β1id/α2id等参数),高层块用dual-scale AdaLN注入情感——通道级MLP预测αemo,c/γemo,c/βemo,c,时间级MLP预测γemo,t并Kronecker积上采样25倍对齐50Hz token。
核心创新可凝练为'三首次一组合':(1)首次在VTS中显式建模RVQ语音token的层级先验,把低/高层token与不同视觉模态一一对应,实现层级—条件解耦;(2)首次把mask-based discrete diffusion从零训练用于VTS,取代连续扩散在mel谱上的低效与过平滑;(3)首次提出dual-scale AdaLN,通过通道维+时间维双重归一化把全局音色与局部韵律解耦注入;组合层面,创新点(1)(2)(3)叠加形成'层级token×层级视觉条件×层级归一化'的三层嵌套结构,与FTV的'层级视觉编码+连续flow matching'、AlignDiT的'扁平对齐DiT'形成本质区别——前者只对齐视觉层级,后者根本不建模层级。
方法步骤详情
流程六步。(1) 视觉抽取:AV-HuBERT-Large 得唇动 $c_{lip}$,ArcFace 对齐 GE2E 得 $c_{id}$,Poster2 平滑得 $c_{emo}$。(2) Token 化:MaskGCT 把语音编为 $12\times 1024$ 码字 $x^{r_{1:12}}$,切低层 $x^{low}=x^{r_{1:2}}$ 与高层 $x^{high}=x^{r_{3:12}}$。(3) 加噪:SEDD log-linear $\sigma(t)$ 调至 $[\text{MASK}]$。(4) Score:低层 8 块通道拼接唇动,$c_{id}$ 预测 $\gamma_{id}/\beta_{id}$ 调制;高层 8 块 dual-scale AdaLN 同时被情感调制,$\gamma_{emo,t}$ 经 $\otimes\mathbf{1}_{25}$ 上采样 25 倍到 50Hz。(5) 训练:$L_{total}=\sum L_{DSE}+100\ell_1(c_{id},c_{GE2E})$,CFG 10% 丢弃,AdamW lr 1e-4 batch 32 200k 迭代。(6) 推理:Euler 64 步,CFG $w_{all}=2.5$、$w_{id}=1.25$、$w_{emo}=1.5$、$w_{lip}=2.0$。
技术新颖性
新颖性分四点。第一,把 RVQ 层级先验作为显式归纳偏置是根本新意,VQ1-2 与 VQ3-12 在 score network 中物理分离,让'唇动+身份'与'情感'不再争夺同一组参数,这是 FTV 用层级视觉编码却仍做 holistic 解码所不具备的。第二,选 masked SEDD 离散扩散而非 absorbing continuous,避免 mel 谱回归频谱模糊,只在 codebook 索引上运算,推理效率优于 mel 连续扩散。第三,dual-scale AdaLN 通过通道归一化捕获全局音色,通过 25 倍 Kronecker 上采样捕获逐帧韵律,公式 $h' \otimes \mathbf{1}_{25} \cdot \gamma_{emo,t} \cdot (1+\gamma_{emo,c}) \cdot \frac{h-\mu(h)}{\sigma(h)} + \beta_{emo,c}$。第四,$L_{id}=\ell_1(c_{id},c_{GE2E})$ 约束视觉—声纹嵌入,表 7 显示去掉后 SpkSim 从 56.78% 骤降到 34.10%。
实验结果
实验在 VoxCeleb2 训练、LRS3/LRS2 零样本测试。(1) LRS3(表 1):HiCoDiT 无音频引导取得 WER 29.41、DNSMOS 3.50、EmoAcc 79.41、LSE-C 7.15、LSE-D 6.58(均最佳),优于 AlignDiT、FTV;加音频引导后 WER 28.98、SpkSim 0.6715(最佳)。(2) LRS2(表 2):同样 WER 39.99、DNSMOS 3.35、EmoAcc 68.21、MCD 8.74(均最佳)。(3) 主观(表 3,4):MOSnat 3.17 超过 GT 3.07,MOSsyn 3.50 超过 GT 3.40,A/B 以 57.0% 击败 AlignDiT、52.1% 击败 FTV,更以 53.9% 胜过 GT。(4) 消融与 OOD:表 5 去层级建模 WER 升到 30.65;表 7 w/o GE2E $L_{id}$ 时 SpkSim 骤降到 34.10%;电影 OOD(表 6)WER 58.7 远低于 EmoDubber 88.3。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| LRS3 视频生成语音 (无音频引导) | WER(词错误率,越低越好) | 29.41 | FTV 30.37, AlignDiT 31.37, EmoDubber 41.52, MTL 76.61, Lip2Wav 98.68 | 相对FTV降3.2%,相对AlignDiT降6.3%,远优于mel谱路线EmoDubber |
| LRS3 视频生成语音 (无音频引导) | EmoAcc(情感识别准确率,越高越好) | 79.41% | AlignDiT 76.11%, FTV 73.19%, EmoDubber 72.01%, MTL 61.24% | 相对AlignDiT升+3.30pp,相对FTV升+6.22pp,验证dual-scale AdaLN对情感注入的有效性 |
| LRS3 视频生成语音 (无音频引导) | DNSMOS(语音质量,越高越好) | 3.50 | AlignDiT 3.24, FTV 3.22, EmoDubber 2.95, DiffV2S 2.56 | 升+0.26相对AlignDiT,显示离散扩散生成的语音比mel谱路线更自然清晰 |
| LRS3 视频生成语音 (无音频引导) | LSE-C/LSE-D(唇音同步置信/距离) | LSE-C 7.15 / LSE-D 6.58 | FTV LSE-C 7.08, AlignDiT LSE-C 6.95, EmoDubber LSE-C 6.88 | LSE-C升+0.07相对FTV,LSE-D降-0.08,显示唇动—低层token对齐精度提升 |
| LRS2 视频生成语音 (零样本) | WER / EmoAcc / MCD / DNSMOS | WER 39.99, EmoAcc 68.21%, MCD 8.74, DNSMOS 3.35 | FTV WER 38.09, AlignDiT WER 42.26, EmoDubber WER 47.60 | 多指标领先,仅FTV在WER单项略低1.9pp但EmoAcc/MCD/DNSMOS全面不及 |
| 主观MOS评估 (20人30样本) | MOSnat / MOSexp / MOSsyn (1-5分) | 3.17 / 2.88 / 3.50 | FTV 2.80 / 2.90 / 3.48, AlignDiT 2.47 / 2.63 / 3.13, GT 3.07 / 3.30 / 3.40 | MOSnat与MOSsyn超越GT,MOSexp仅略低于FTV 0.02,显示感知质量全面领先 |
| A/B偏好测试 | 胜率(%) | vs AlignDiT 57.0% / vs FTV 52.1% / vs GT 53.9% | AlignDiT 38.1% / FTV 41.8% / GT 45.5% | 胜出AlignDiT+18.9pp、FTV+10.3pp,甚至以53.9%胜过真实语音GT(显示GT评测集瑕疵) |
| 层级建模与dual-scale AdaLN消融 (LRS3) | WER / DNSMOS / UTMOS / EmoAcc | Full: WER 29.41, DNSMOS 3.50, UTMOS 3.84, EmoAcc 79.41 | w/o Hierarchical WER 30.65 DNSMOS 3.36 EmoAcc 76.98; w/o Dual-scale AdaLN WER 29.60 DNSMOS 3.45 EmoAcc 78.55 | 层级建模贡献最大(EmoAcc +2.43pp,DNSMOS +0.14);dual-scale AdaLN次之(EmoAcc +0.86pp) |
| 视觉条件消融 (LRS3) | SpkSim / EmoAcc | Full: SpkSim 56.78%, EmoAcc 79.41% | w/o GE2E L_id: SpkSim 34.10% (-22.68pp);w/o Poster2(换Poster): EmoAcc 76.29% (-3.12pp) | GE2E身份损失是音色保留的核心驱动,Poster2比Poster更适合细粒度情感识别 |
| 真实电影OOD测试 (CinePile 160样本) | WER / EmoAcc / DNSMOS | WER 58.7, EmoAcc 82.0%, DNSMOS 3.5 | EmoDubber WER 88.3, AlignDiT WER 80.8 | WER相对AlignDiT降-22.1pp,EmoAcc升+6.8pp,真实复杂场景鲁棒性显著 |
局限与改进
作者未单独设置Limitations小节,但根据行文与数据可观察到五点局限。其一,表1显示HiCoDiT在LRS3 SpkSim 0.5678低于FTV 0.5981,作者承认'受限于训练集多样性';加入音频引导后才升至0.6715,说明纯视觉身份建模仍有显著提升空间。其二,真实电影OOD测试中SpkSim 50.1反低于AlignDiT 58.5,提示域外身份泛化能力存在退化。其三,训练仅在VoxCeleb2 261.5小时(7种基本情感、3438说话人)上完成,情感类别粒度粗,无法处理复合情感或文化特定表达。其四,推理依赖64步Euler采样,虽然比连续DDPM快,但仍慢于自回归LLM风格TTS,实际部署的实时性未报告。其五,实验仅在LRS2/LRS3与CinePile电影上测试,未覆盖中文、手语、远程会议、儿童等场景,泛化边界尚未充分验证,主观测试也仅20人参与,统计显著性需要更大样本。
独立分析的弱点
独立分析可识别五点可改进之处。第一,层级切分点固定为 VQ1-2 与 VQ3-12,但层级比例是否最优未做消融;若改为 VQ1-3 vs VQ4-12 或 VQ1 vs VQ2-12 可能进一步提升,建议扫描 $r_2 \in \{2,3,4\}$ 的层级划分。第二,dual-scale AdaLN 的时间维度 25 倍 Kronecker 上采样是硬编码,对其他采样率或可变长视频缺乏自适应,改进方向是引入可学习的动态上采样或 attention-based 时间对齐。第三,$L_{id}$ 仅在训练时使用真值 GE2E embedding,但视觉身份 $c_{id}$ 与声纹 GE2E 本质跨模态,若用对比学习 InfoNCE 替代 $\ell_1$ 可学到更鲁棒的跨模态身份空间,且推理时无需真值。第四,情感支路 Poster2 只输出类别概率+0.5 秒平滑,丢失了情感强度的连续谱,改进方向是引入连续维度 (valence/arousal) 回归头。第五,训练仅 200k 迭代 batch 32,未做 loss 曲线报告,改进方向是绘制完整训练曲线并对比不同训练长度。
未来方向
作者隐含方向是把'层级token×层级视觉条件'范式扩展到更复杂场景,基于此可延伸四条研究方向:(1)把层级扩散框架推广到多说话人对话场景VTS,如访谈节目、电话会议,需引入speaker-aware speaker turn embedding,层级切分可拓展到说话人级VQ层;(2)结合LLM做'可控语音生成',用自然语言指令直接控制情感强度、说话风格、语速,通过dual-scale AdaLN的γ/β参数把语言条件直接映射到韵律调制;(3)扩展到TTS、Voice Conversion、Speech Editing等多任务,把RVQ层级先验作为通用语音生成归纳偏置,建立统一的层级离散扩散语音基础模型;(4)探索更长horizon的视频输入(>10秒)与超长语音生成,通过层级VQ的递归结构实现coarse-to-fine的token-by-token生成,降低单步去噪成本。
复现评估
复现评估分四方面。开源情况:论文明确声明代码与语音 demo 发布在 https://github.com/Jiaxin-Ye/HiCoDiT,可下载权重与推理脚本,但训练代码完整性需进一步确认。数据可获得性:VoxCeleb2 需向官方申请(学术免费),LRS2/LRS3 公开,AV-HuBERT/ArcFace/Poster2/MaskGCT 全部开源。算力需求:训练 200k 迭代、batch 32、AdamW lr 1e-4,未明确 GPU 数;按模型规模 (8+8 Transformer block, 通道 768) 粗估至少 8×A100 80G 跑 3-5 天,推理单 GPU 即可。复现难度:技术栈涉及 RVQ codec、masked discrete diffusion、AdaLN、CFG 组合,但作者给出所有超参 ($\lambda=100$、$w_{all}=2.5/2.25$、$w_{id}=1.25$、$w_{emo}=1.5$、$w_{lip}=2.0$、64 步 Euler),中等团队 2-3 周可复现核心结果。
论文图表
RVQ codec结构示意与层级消融曲线:(a)子图展示VQ1→VQ12逐层残差量化的物理结构;(b)子图横轴为累积解码层数,纵轴报告内容/音色/韵律三类分数,显示VQ1-2集中改善speaker-aware语义(30.35%/20.10%相对增益),VQ3-12则渐进改善韵律(10.85%增益),定量证明层级切分点选择有经验依据。
为'为何把VQ1-2作为低层、VQ3-12作为高层'这一关键设计选择提供了定量证据,是支撑全文motivation的核心图。