mlsanityData to 14 Oct 2026

Does per-frame early exit pay? A compute-matched study of dynamic depth for on-device speech enhancement

arXiv 2609.29867 ↗First posted 24 Sep 2026eess.AS cs.LG cs.SDcited by 0

New in Efficient LLM inference and serving; no direction yet. Efficient LLM inference and serving

Abstract (arXiv)

Deep learning-based speech enhancement is increasingly deployed on-device in hearing aids, headsets, and earbuds. Most of these devices, however, can only accelerate static int8 graphs, so a depth-varying network must be implemented as several graphs, orchestrated by a policy. In this paper, we supervise every intermediate depth of one causal model, then we fine-tune its output heads to guarantee that deeper outputs are never worse than shallower ones. Using this training protocol, we can derive a family of static models that are more Pareto-efficient than their equivalently-sized counterparts trained from scratch on the same budget. Specifically, we achieve up to 0.11 higher PESQ for equivalent compute, and match the best PESQ at 30% less compute. We then quantize the models to int8 and measure the latency-quality frontier on an STM32N6 microcontroller. On VoiceBank-DEMAND, the dynamic enhancer lies on the same frontier as the static models, rather than trading quality for dynamic execution. Running the policy on the companion Cortex-M55 takes only 26 μs per frame, while splitting the enhancer into separate NPU graphs adds 2.2% latency overhead. The cost of dynamic execution is therefore small.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 27 of its 33 references that are in the map. First posted on arXiv 24 Sep 2026.

Builds on

27 of 33 references.

References · 27

  1. 2016Categorical Reparameterization with Gumbel-Softmaxcited by 2,739
  2. 2018ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Designcited by 1,136
  3. 2016Deep Networks with Stochastic Depthcited by 979
  4. 2019Once-for-All: Train One Network and Specialize it for Efficient Deploymentcited by 787
  5. 2021A White Paper on Neural Network Quantizationcited by 472
  6. 2019Reducing Transformer Depth on Demand with Structured Dropoutcited by 439
  7. 2018SDR - half-baked or well done?cited by 423
  8. 2021Dynamic Neural Networks: A Surveycited by 408
  9. 2018Conv-TasNet: Surpassing Ideal Time-Frequency Magnitude Masking for Speech Separationcited by 406
  10. 2014Deeply-Supervised Netscited by 367
  11. 2019Depth-Adaptive Transformercited by 236
  12. 2019Universally Slimmable Networks and Improved Training Techniquescited by 221
15 more
  1. 2020Real Time Speech Enhancement in the Waveform Domaincited by 186
  2. 2021Towards efficient models for real-time deep noise suppressioncited by 29
  3. 2020Data augmentation and loss normalization for deep noise suppressioncited by 23
  4. 2020A consolidated view of loss functions for supervised deep learning-based speech enhancementcited by 19
  5. 2023Towards Anytime Classification in Early-Exit Architectures by Enforcing Conditional Monotonicitycited by 16
  6. 2023Subnetwork-to-go: Elastic Neural Network with Dynamic Training and Customizable Inferencecited by 9
  7. 2023Dynamic nsNet2: Efficient Deep Noise Suppression with Early Exitingcited by 7
  8. 2024Scalable Speech Enhancement with Dynamic Channel Pruningcited by 5
  9. 2023TODM: Train Once Deploy Many Efficient Supernet-Based RNN-T Compression For On-device ASR Modelscited by 4
  10. 2025Adaptive Slimming for Scalable and Efficient Speech Enhancementcited by 4
  11. 2025Dynamic Slimmable Networks for Efficient Speech Separationcited by 3
  12. 2019Progressive Speech Enhancement with Residual Connectionscited by 2
  13. 2025Knowing When to Quit: Probabilistic Early Exits for Speech Separationcited by 2
  14. 2026From Diet to Free Lunch: Estimating Auxiliary Signal Properties using Dynamic Pruning Masks in Speech Enhancement Networkscited by 2
  15. 2025Dynamically Slimmable Speech Enhancement Network with Metric-Guided Trainingcited by 1

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Beyond Model Size: Redesigning LiSenNet for embedded speech enhancement
  2. 2024Modulating State Space Model with SlowFast Framework for Compute-Efficient Ultra Low-Latency Speech Enhancement
  3. 2024Dense-TSNet: Dense Connected Two-Stage Structure for Ultra-Lightweight Speech Enhancement ↗
  4. 2026RT-SEMamba: Real-Time Speech Enhancement Mamba via Progressive Knowledge Distillation
  5. 2025Dynamic Frequency-Adaptive Knowledge Distillation for Speech Enhancement
  6. 2021Weight, Block or Unit? Exploring Sparsity Tradeoffs for Speech Enhancement on Tiny Neural Accelerators
  7. 2025EDNet: A Versatile Speech Enhancement Framework with Gating Mamba Mechanism and Phase Shift-Invariant Training
  8. 2024Ultra-Low Latency Speech Enhancement - A Comprehensive Study