Memory Efficient Audio Synthesis with Decoupled Temporal Depth Diffusion Transformers
Summary
Apple's Siri Expressive Voices leverage a novel memory-efficient audio synthesis architecture, a detokenizer, to convert semantic audio tokens from the AFM 3 Core Advanced foundation model into high-fidelity audio on-device. Designed for the Apple Matrix Coprocessor (AMX), this architecture uses a three-component residual vector quantization (RVQ) representation with a streaming encoder, temporal decoder, and depth decoder, systematically decoupling temporal and depth processing. It features a single reusable depth decoder with Diffusion Transformer (DiT)-style stage conditioning and causal sliding window attention with fixed-window key-value caching, ensuring constant memory complexity. Deployed on the AMX, it achieves roughly 10 ms per generation step, 16x faster than real time, with a peak runtime memory of 21 MB and 329 MB of on-device assets. This enables continuous streaming synthesis of 20-320 seconds of audio, replacing conventional approaches' linear and quadratic memory scaling. The system improves Mean Opinion Score by +0.28 overall (4.15 vs. 3.87) and by +0.42 on conversational speech (4.24 vs. 3.82).
Key takeaway
For Machine Learning Engineers developing on-device audio synthesis, this architecture offers a blueprint for achieving high-fidelity results within severe memory and compute constraints. You should consider implementing decoupled temporal and depth processing with a single reusable depth decoder and causal sliding window attention to ensure constant memory complexity. This approach can deliver 16x real-time performance and significantly improve Mean Opinion Score on conversational speech, making advanced audio features viable for edge devices.
Key insights
Decoupling temporal and depth processing in Diffusion Transformers enables highly memory-efficient, high-fidelity on-device audio synthesis.
Principles
- Decouple temporal and depth processing for efficiency.
- Use a single reusable depth decoder.
- Causal sliding window attention ensures constant memory.
Method
Convert semantic audio tokens to RVQ using a streaming encoder, temporal decoder, and a single DiT-style stage-conditioned depth decoder, employing causal sliding window attention.
In practice
- Implement RVQ with decoupled decoders for on-device audio.
- Utilize fixed-window key-value caching for streaming.
- Achieve 16x real-time audio generation on constrained hardware.
Topics
- On-device Audio Synthesis
- Diffusion Transformers
- Residual Vector Quantization
- Memory Efficiency
- Real-time Audio
- Apple AMX
Best for: Research Scientist, AI Engineer, AI Scientist, Machine Learning Engineer, NLP Engineer
Related on AIssential
See Counsel's argued verdicts on the open AI decisions leaders are weighing →
Editorial summary, takeaway, and curation by AIssential. Original article published by Computation and Language.