Memory Efficient Audio Synthesis with Decoupled Temporal Depth Diffusion Transformers

· Source: Computation and Language · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Software Development & Engineering · Depth: Expert, quick

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

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

Topics

Best for: Research Scientist, AI Engineer, AI Scientist, Machine Learning Engineer, NLP Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Computation and Language.