Efficient Streaming Audio-Visual Target Speaker Extraction for Real-World Acoustic Scenes
Organizations: School of Computer Technology and Application, Qinghai University, Xining 810016, China · Department of Computer Science and Technology, Institute for AI, BNRist, Tsinghua Laboratory of Brain and Intelligence (THBI), IDG/McGovern Institute for Brain Research, Tsinghua University, Beijing 100084, China · Intelligent Computing and Application Laboratory of Qinghai Province, Qinghai University, Xining 810016, China · Chinese Institute for Brain Research (CIBR), Beijing 100010, China
Abstract
Streaming audio-visual target speaker extraction (AVTSE) is essential for latency-sensitive applications. However, current causal systems face two coupled gaps. First, standard benchmarks fix the number of speakers at two and ignore the near-/far-field interference, reverberation, and sudden noise of real scenes. Second, high-performance causal architectures are too heavy for edge deployment. Lightweight alternatives try to reclaim capacity by repeatedly invoking a small separator, which offsets the savings. To address the first gap, we release RealSSA, a realistic AVTSE benchmark consisting of RealSSA-Sim and RealSSA-Real. RealSSA-Sim provides controllable simulated scenes with 1--5 dynamic near-field interfering speakers; scenes with far-field interference additionally include 10--15 far-field interfering speakers. RealSSA-Real provides a held-out real-recorded evaluation set. To address the second gap, we propose Falcon, a causal AVTSE method. Its Separator performs multi-scale time-frequency separation in a single encoder--decoder pass. We train Falcon with an encoder-space speaker contrastive loss. This loss suppresses near-field leakage at zero inference cost and transfers across backbones. Falcon achieves state-of-the-art extraction quality on RealSSA-Sim, LRS2-2Mix, LRS3-2Mix, and VoxCeleb2-2Mix. Compared with the causal AV-TFGridNet baseline, it cuts parameters by 91.9%, computation by 20.2 times, and GPU inference latency by 11.1 times. Code and demo are available at https://cslikai.cn/Falcon/.
Figures & tables
| Benchmark | NF speakers | FF speakers | Reverb control | Env. noise | Sudden noise | Real-recorded eval |
|---|---|---|---|---|---|---|
| NTCD-TIMIT+noise [ 2017 ] | 2 | ✗ | ✗ | ✓ | ✗ | ✗ |
| LRS2-2Mix [ 2018a ] | 2 | ✗ | ✗ | ✗ | ✗ | ✗ |
| LRS3-2Mix [ 2018b ] | 2 | ✗ | ✗ | ✗ | ✗ | ✗ |
| VoxCeleb2-2Mix [ 2018 ] | 2 | ✗ | ✗ | ✗ | ✗ | ✗ |
| LRS3+WHAM! [ 2023 ] | 2 | ✗ | ✗ | ✓ | ✗ | ✗ |
| RealSSA (ours) | 2–6 | ✓ | ✓ | ✓ | ✓ | ✓ |
| Scene | Space size | RT60 | Near speakers | Far speakers | Env. noise SIR | Sudden noise |
|---|---|---|---|---|---|---|
| Small meeting | 0.3–0.6 s | 1–5, 0.5–1.5 m | none | 5–15 dB | 30%, 1 event | |
| Medium indoor | 0.5–0.9 s | 1–3, 1.0–2.5 m | 10–15, 3–15 m | 5–10 dB | 40%, 1–2 events | |
| Large public | 1.0–2.0 s | 1–4, 1.0–3.0 m | 10–15, 5–20 m | 2–8 dB | 50%, 1–3 events | |
| Outdoor | 0.05–0.1 s | 1–4, 1.0–4.0 m | 10–15, 5–25 m | – dB | 40%, 1–3 events |
| Method | RealSSA-Real | RealSSA-Sim | LRS2-2Mix | LRS3-2Mix | VoxCeleb2-2Mix | ||||
|---|---|---|---|---|---|---|---|---|---|
| DNSMOS | SI-SNRi | SDRi | SI-SNRi | SDRi | SI-SNRi | SDRi | SI-SNRi | SDRi | |
| AV-ConvTasNet [ 2021 ] | 2.580 | 1.2 | 2.6 | 9.8 | 10.3 | 12.5 | 12.8 | 6.9 | 8.2 |
| AV-Sepformer [ 2023 ] | 2.813 | 5.5 | 6.3 | 10.6 | 11.1 | 13.4 | 13.7 | 6.9 | 8.3 |
| AV-TF-GridNet [ 2023 ] | 3.030 | 7.6 | 8.1 | 12.7 | 13.1 | 14.3 | 14.8 | 5.8 | 7.7 |
| RTFSNet-12 [ 2024 ] | 3.066 | 4.9 | 5.7 | 12.4 | 12.8 | 14.0 | 14.4 | 8.4 | 10.7 |
| CTCNet [ 2024b ] | 2.840 | 5.4 | 6.2 | 8.0 | 8.8 | 9.4 | 10.2 | 2.5 | 4.7 |
| Method | RealSSA-Sim | RealSSA-Real | ||
|---|---|---|---|---|
| WER (%) | Similarity | WER (%) | Similarity | |
| AV-Sepformer [ 2023 ] | 32.64 | 38.69 | 34.61 | 38.25 |
| AV-TF-GridNet [ 2023 ] | 27.31 | 40.44 | 34.95 | 39.88 |
| RTFSNet-12 [ 2024 ] | 31.91 | 36.51 | 35.56 | 35.78 |
| CTCNet [ 2024b ] | 34.44 | 37.08 | 38.81 | 35.33 |
| AV-Mossformer2 [ 2025 ] | 32.44 | 37.34 | 32.00 | 36.42 |
| Method | Params (M) | MACs (G) | Latency (ms) | Memory (MB) | ||
|---|---|---|---|---|---|---|
| CPU | GPU | RAM | GPU Inf. | |||
| AV-TF-GridNet [ 2023 ] | 6.2 | 213.8 | 5646.3 | 567.8 | 136.3 | 1015.8 |
| AV-Mossformer2 [ 2025 ] | 56.7 | 224.2 | 16670.7 | 114.9 | 330.9 | 539.8 |
| Swift-Net-12 [ 2025 ] | 0.5 | 45.3 | 4163.8 | 172.1 | 482.3 | 411.7 |
| Dolphin [ 2026a ] | 5.6 | 18.1 | 2411.0 | 87.9 | 560.2 | 390.3 |
| Falcon (Ours) | 0.5 | 10.6 | 1361.2 | 51.3 | 125.3 | 364.7 |
| Method | VoxCeleb2 | RealSSA-Sim (Ours) | ||
|---|---|---|---|---|
| NISQA | DNSMOS | NISQA | DNSMOS | |
| AV-TF-GridNet | 2.495 | 2.757 | 2.413 | 3.030 |
| AV-Mossformer2 | 1.995 | 2.715 | 2.256 | 2.912 |
| Swift-Net-12 | 2.481 | 3.064 | 3.484 | 3.276 |
| Dolphin | 1.593 | 2.608 | 2.590 | 2.960 |
| Falcon (Ours) | 2.431 | 2.878 | 3.615 | 3.276 |
| Stages | SI-SNRi | SDRi |
|---|---|---|
| 1 | 7.1 | 7.7 |
| 2 | 7.4 | 7.9 |
| 3 (Ours) | 7.7 | 8.2 |
| Model | Loss | SI-SNRi | SDRi |
|---|---|---|---|
| Falcon | Without | 7.3 | 7.8 |
| Falcon | + Ours | 7.7 | 8.2 |
| SwiftNet-12 | Baseline | 7.3 | 7.8 |
| SwiftNet-12 | + Ours | 7.5 | 8.0 |
| Dolphin | Baseline | 6.7 | 7.3 |
| Dolphin | + Ours | 7.0 | 7.6 |
| Strategy | SI-SNRi | SDRi |
|---|---|---|
| Fixed | 7.5 | 8.1 |
| Fixed | 7.4 | 7.9 |
| Anneal | 7.5 | 8.0 |
| Anneal (Ours) | 7.7 | 8.2 |
Appendix figures & tables17 assets
Supplementary material from the paper’s appendix.
Appendix
| Corpus | Role | Hours | Speakers | Native SR | Used SR |
|---|---|---|---|---|---|
| LRS3 [ 10 ] | Target (audio–visual) | 438 | 9,545 | ||
| LibriSpeech [ 29 ] | Near-/far-field interferers | 1000 | 2,484 | ||
| WHAM! [ 12 ] | Continuous environmental noise | 80 | – | ||
| Hive [ 30 ] | Semantic environmental and sudden events | 544 | – |
| Scene | Space size | RT60 | Near speakers | Far speakers | Near SIR | Env. noise SIR | Applied Far SIR | Sudden noise |
|---|---|---|---|---|---|---|---|---|
| Small meeting | 0.3–0.6 s | 1–5, 0.5–1.5 m | none | 2–5 dB | 5–15 dB | – | 30%, 1 event | |
| Medium indoor | 0.5–0.9 s | 1–3, 1.0–2.5 m | 10–15, 3–15 m | 2–8 dB | 5–10 dB | 20–35 dB | 40%, 1–2 events | |
| Large public | 1.0–2.0 s | 1–4, 1.0–3.0 m | 10–15, 5–20 m | 2–8 dB | 2–8 dB | 20–35 dB | 50%, 1–3 events | |
| Outdoor | 0.05–0.1 s | 1–4, 1.0–4.0 m | 10–15, 5–25 m | 2–15 dB | to dB | 20–35 dB | 40%, 1–3 events |
| Scene | Environmental noise labels | Sudden-noise labels |
|---|---|---|
| Small meeting | air conditioning, fan, clock, typing, printer, keyboard, tick, hiss | writing, clicking, crumple, rustle, knock, telephone, door, drawer, squeak, paper |
| Medium indoor | microwave, blender, dishes, pots/pans, refrigerator | cutlery, glass, cupboard, frying, cash register, gurgling, steam, kettle, sneeze, cough, laughter |
| Large public | footsteps, shuffle, speech, babble, crowd | sliding door, siren, child speech, crying, roll, alarm, buzzer, announcement |
| Outdoor | wind, rain, traffic, stream, city ambience | car, bus, truck, motorcycle, car passing, skidding, reversing beep, horn, police/ambulance/fire siren, dog, bird, cricket, jackhammer, hammer, power tool |
| All scenes (shared) | – | key jangling, coin dropping, slam, bang, thump, crash, squeal |
| Split | Samples | Total hours | Avg. duration | Avg. speakers | Avg. Near SIR | Avg. Applied Far SIR |
|---|---|---|---|---|---|---|
| Train | s | dB | dB | |||
| Val | s | dB | dB | |||
| Test | s | dB | dB |
| Simulated scene | Real-world environment | Sessions | Total duration (m:s) |
|---|---|---|---|
| Small meeting | Study room | ||
| Classroom | |||
| Medium indoor | Internet cafe | ||
| Billiard hall | |||
| Large public | Shopping mall | ||
| Supermarket |
| Symbol | Module | Value |
|---|---|---|
| STFT encoder output channels | ||
| Audio bottleneck output channels | ||
| Separator hidden channels | ||
| Visual hidden channels | ||
| Speaker embedding dimension | ||
| TFRB frequency-LSTM hidden size |
| Item | Value |
|---|---|
| Optimizer | AdamW, learning rate , weight decay |
| LR scheduler | ReduceLROnPlateau, factor , patience |
| Early stopping | Patience on validation loss |
| Max epochs | |
| Batch size per GPU | |
| Hardware | NVIDIA RTX 3090 |
| Method | Small meeting | Medium indoor | Large public | Outdoor | ||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| AV-ConvTasNet [ 39 ] | 0.1 | 0.4 | 0.7 | 0.7 | 1.6 | 1.5 | 1.5 | 1.3 | 1.4 | 1.6 | 1.5 | 0.9 | 1.5 | 2.0 | 1.9 | |
| AV-DPRNN [ 26 ] | 4.0 | 4.4 | 4.3 | 4.3 | 4.1 | 5.1 | 4.9 | 4.8 | 5.4 | 5.4 | 5.5 | 5.6 | 6.6 | 6.8 | 7.0 | 7.1 |
| AV-Sepformer [ 27 ] | 4.0 | 4.4 | 4.3 | 4.4 | 4.2 | 4.9 | 4.8 | 4.7 | 5.4 | 5.4 | 5.6 | 5.6 | 7.1 | 7.3 | 7.5 | 7.4 |
| CTCNet [ 6 ] | 4.0 | 4.4 | 4.4 | 4.5 | 4.3 | 5.0 | 4.9 | 4.8 | 5.3 | 5.3 | 5.5 | 5.7 | 6.6 | 6.9 | 7.1 | 7.3 |
| RTFSNet-12 [ 16 ] | 2.5 | 3.4 | 3.5 | 3.9 | 3.7 | 4.5 | 4.6 | 4.5 | 4.7 | 4.9 | 4.9 | 5.3 | 4.8 | 5.4 | 5.8 | 6.0 |
| Recurrent Unit | SI-SNRi | SDRi | Params (M) | MACs (G) |
|---|---|---|---|---|
| RNN | 6.2 | 7.1 | 0.08 | 1.90 |
| SRU | 7.4 | 7.9 | 0.15 | 2.71 |
| GRU | 7.3 | 7.9 | 0.20 | 4.42 |
| LSTM (Ours) | 7.7 | 8.2 | 0.25 | 5.69 |
| Output formulation | SI-SNRi | SDRi |
|---|---|---|
| Complex-valued mask (Falcon) | 7.7 | 8.2 |
| Real-valued mask | 7.3 | 7.9 |
| Direct real/imaginary prediction | 7.1 | 7.2 |
| Overlap range | Samples | Mean overlap | SI-SNRi | SDRi |
|---|---|---|---|---|
| – | ||||
| – | ||||
| – | ||||
| – | ||||
| – |
| Method | Missing pattern | |||||
|---|---|---|---|---|---|---|
| AV-Sepformer [ 27 ] | Random Drop | 5.26 / 5.88 | 4.98 / 5.68 | 4.57 / 5.38 | 2.35 / 4.02 | / |
| AV-Sepformer [ 27 ] | Block Drop | 5.26 / 5.88 | 2.42 / 3.65 | / | / | / |
| RTFSNet [ 16 ] | Random Drop | 4.01 / 4.61 | 3.87 / 4.50 | 3.42 / 4.15 | 1.69 / 3.02 | / |
| RTFSNet [ 16 ] | Block Drop | 4.01 / 4.61 | 2.77 / 3.66 | 0.51 / 2.20 | / 0.31 | / |
| CTCNet [ 6 ] | Random Drop | 5.07 / 5.72 | 4.83 / 5.53 | 4.30 / 5.09 | 2.90 / 4.03 | / |
| CTCNet [ 6 ] | Block Drop | 5.07 / 5.72 | 3.04 / 4.10 | / 1.76 | / | / |
Target Speaker Extraction
Target speaker extraction isolates a chosen voice from overlapping speech using speaker identity, visual, spatial, textual, or neural cues.
See all 38 papers