PS4: Proxy-Supervised Joint Training for Real Target Speaker Extraction

· Source: Artificial Intelligence · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Speech & Audio Processing · Depth: Expert, quick

Summary

PS4 is a novel proxy-supervised training framework designed for Target Speaker Extraction (TSE) in real conversational mixtures, addressing the scarcity of large-scale supervised training data. The framework's first contribution is the creation of a substantial corpus comprising 71,771 training samples, sourced from four public datasets covering both Chinese and English scenarios. Each sample integrates an overlapping speech mixture, per-speaker enrollment audio, a ground-truth transcript, and frame-level voice activity labels. Second, PS4 introduces a proxy-supervised joint training strategy that fine-tunes a BSRNN-based TSE model. This fine-tuning process utilizes four complementary differentiable objectives: ASR cross-entropy, speaker similarity, frame-level voice activity detection, and perceptual audio quality. Starting from a pre-trained checkpoint, only the BSRNN separator is updated. PS4 achieved 2nd place overall on the REAL-T challenge leaderboard, demonstrating the best speaker similarity and timing F1 among all submissions.

Key takeaway

For AI Scientists developing Target Speaker Extraction models for real conversational environments, PS4 demonstrates a viable path to overcome data scarcity. You should explore proxy-supervised joint training, integrating diverse objectives such as ASR cross-entropy, speaker similarity, frame-level voice activity detection, and perceptual audio quality. This strategy, particularly when fine-tuning only the separator of a pre-trained BSRNN model, can yield strong performance, as evidenced by PS4's top speaker similarity on the REAL-T challenge.

Key insights

Proxy-supervised joint training with multiple objectives enables effective Target Speaker Extraction from real conversational mixtures despite limited supervised data.

Principles

Method

Fine-tune a BSRNN-based TSE model using a proxy-supervised joint training strategy. This involves four differentiable objectives: ASR cross-entropy, speaker similarity, frame-level VAD, and perceptual audio quality. Only the separator is updated.

In practice

Topics

Best for: NLP Engineer, AI Scientist, Research Scientist

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.