跳到正文
arXiv cs.AI· Dohwan Kim, Jung-Woo Choi·· 12 小时前AI 评分22

MG-REPA:面向生成式目标说话人提取的多粒度表示对齐

MG-REPA: Multi-Granularity Representation Alignment for Generative Target Speaker Extraction

AI 导读

MG-REPA 通过将隐藏状态在帧级对齐 SSL 模型、在 utterance 级对齐说话人验证模型,加速生成式目标说话人提取训练。在 Libri2Mix 上,达到 MeanFlow-TSE 基线同等 OVRL 和 PESQ 分别只需 3.0 倍和 2.1 倍更少 epoch;训练 750 epoch 即在侵入指标上媲美已发表 2000 epoch 基线,并提升非侵入指标。

正文 · 原文

View PDF HTML (experimental)

Abstract:Generative target speaker extraction (TSE) models achieve high-quality extraction, but suffer from considerably slow training convergence. Representation Alignment (REPA) accelerates diffusion transformer training by aligning hidden states with a pre-trained Self-Supervised Learning (SSL) teacher. However, existing alignment in the speech and audio domain operates at a single granularity. For TSE, frame-level alignment leaves the global, time-invariant characteristics of the target speech implicit, whereas utterance-level alignment leaves the time-varying content implicit, even though extracting the target requires both. We propose Multi-Granularity Representation Alignment (MG-REPA), aligning the hidden states with an SSL model at the frame level and a speaker verification model at the utterance level. We further apply temporal normalization to the frame-level target, removing its utterance-level statistics so that each branch concentrates on its granularity. On Libri2Mix, MG-REPA reaches the OVRL and PESQ of a MeanFlow-TSE baseline with $3.0\times$ and $2.1\times$ fewer epochs, respectively. Trained for 750 epochs, it is competitive with the published 2,000-epoch baseline on intrusive metrics while improving non-intrusive metrics. An ablation demonstrates that MG-REPA outperforms alignment at a single granularity.
Subjects: Audio and Speech Processing (eess.AS); Artificial Intelligence (cs.AI)
Cite as: arXiv:2610.32161 [eess.AS]
  (or arXiv:2610.32161v1 [eess.AS] for this version)
  https://doi.org/10.48550/arXiv.2610.32161

arXiv-issued DOI via DataCite (pending registration)

Submission history

From: DoHwan Kim [view email]
[v1] Fri, 23 Oct 2026 04:51:51 UTC (520 KB)

来源:arXiv cs.AI · arxiv.org