Staged Depth-Pruning Distillation of a Flow-Matching Text-to-Speech Teacher: A Compact Hindi Speech Synthesizer
On the map: Diffusion models → Audio, speech and music
also in 1 more
Abstract (arXiv)
We present a practical recipe for building a compact Hindi text-to-speech (TTS) model by distilling a large flow-matching teacher (IndicF5, 337M-parameter DiT) under a severe data budget (~17.6 hours). Training a small model from scratch on this much data fails outright. Instead we warm-start the student from the teacher by pruning depth only: keeping the teacher's width, text dimension, attention heads, and mel/text I/O fixed so all non-block tensors copy one-to-one, and retaining an evenly-spaced subset of transformer blocks. We first measure how much depth the teacher tolerates (it remains near-functional at -27% blocks but collapses past -50%), then descend gradually (22 -> 16 -> 12 -> 8 -> 6 blocks), re-fine-tuning after each prune, with each step gated by an objective ASR word-error-rate (WER) check. The resulting students reach WER 0.00 on unseen sentences at 249M and 190M parameters, and remain robust down to 131M; at 102M we observe a clear capacity cliff that we attribute to the data budget rather than the recipe. We also document two train/inference feature- and library-parity failures (mel filterbank and rotary-embedding library versions) that silently degrade audio, and a version-independent fix. The method yields a high-quality Hindi voice that runs in real time on a 6 GB laptop GPU. An independent 50-sentence FLEURS benchmark compares the released 190M student against its teacher and MMS-TTS-hin.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Audio, speech and music, one of 14 directions in Diffusion models. Builds on 14 of its 20 references that are in the map. First posted on arXiv 19 Jul 2026.
Builds on
14 of 20 references.
Closest ancestors · 1
Same direction, newest first
Other references · 13
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2022Scalable Diffusion Models with Transformerscited by 6,054
- 2022Flow Matching for Generative Modelingcited by 5,702
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,700
- 2016Sequence-Level Knowledge Distillationcited by 801
- 2020ECAPA-TDNN: Emphasized Channel Attention, Propagation and Aggregation in TDNN Based Speaker Verificationcited by 628
- 2019Reducing Transformer Depth on Demand with Structured Dropoutcited by 439
- 2023Scaling Speech Technology to 1,000+ Languagescited by 411
- 2022UTMOS: UTokyo-SaruLab System for VoiceMOS Challenge 2022cited by 390
- 2024ShortGPT: Layers in Large Language Models are More Redundant Than You Expectcited by 383
- 2023Vocos: Closing the gap between time-domain and Fourier-based neural vocoders for high-quality audio synthesiscited by 153
- 2024Natural language guidance of high-fidelity text-to-speech with synthetic annotationscited by 67
1 more
- 2021Towards Building ASR Systems for the Next Billion Userscited by 47
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Depth through recurrence: Looped transformers for flow-matching TTS
- 2026DSFlow: Dual Supervision and Step-Aware Architecture for One-Step Flow Matching Speech Synthesis
- 2026Distill Locally, Schedule Globally: Flow Maps for Few-Step Text-to-Speech
- 2025Efficient Interleaved Speech Modeling through Knowledge Distillation
- 2025SlimSpeech: Lightweight and Efficient Text-to-Speech with Slim Rectified Flow
- 2022Nix-TTS: Lightweight and End-to-End Text-to-Speech via Module-wise Distillation
- 2026DLLM-TTS: Block Discrete Diffusion Language Model for Text-to-Speech Synthesis
- 2026BareWave: Waveform-Native Flow-Matching Text-to-Speech