PTQ4ADM: Post-Training Quantization for Efficient Text Conditional Audio Diffusion Models
Denoising diffusion models have emerged as state-of-the-art in generative tasks across image, audio, and video domains, producing high-quality, diverse, and contextually relevant data. However, their broader adoption is limited by high computational costs and large memory footprints. Post-training quantization (PTQ) offers a promising approach to mitigate these challenges by reducing model complexity through low-bandwidth parameters. Yet, direct application of PTQ to diffusion models can degrade synthesis quality due to accumulated quantization noise across multiple denoising steps, particularly in conditional tasks like text-to-audio synthesis. This work introduces PTQ4ADM, a novel framework for quantizing audio diffusion models(ADMs). Our key contributions include (1) a coverage-driven prompt augmentation method and (2) an activation-aware calibration set generation algorithm for text-conditional ADMs. These techniques ensure comprehensive coverage of audio aspects and modalities while preserving synthesis fidelity. We validate our approach on TANGO, Make-An-Audio, and AudioLDM models for text-conditional audio generation. Extensive experiments demonstrate PTQ4ADM’s capability to reduce the model size by up to 70% while achieving synthesis quality metrics comparable to full-precision models(<5% increase in FD scores). We show that specific layers in the backbone network can be quantized to 4-bit weights and 8-bit activations without significant quality loss. This work paves the way for more efficient deployment of ADMs in resource-constrained environments.
Authors
- Prabhu Shankar (ORCID: https://orcid.org/0000-0001-9909-8527)
- Nader Bouacida (ORCID: https://orcid.org/0000-0002-4571-8484)
- Aditya Krishnan
- Prasant Mohapatra (ORCID: https://orcid.org/0000-0002-2768-5308)
- Jayneel Vora (ORCID: https://orcid.org/0000-0002-0286-6380)
Institutions
- UC Davis Health (US)
- University of California, Davis (US)
- University of Tampa (US)
Publication Details
- Published
- 2025-03-12
- DOI
- https://doi.org/10.1109/icassp49660.2025.10890341
- Primary Topic
- Music and Audio Processing
- Type
- article
- Field-Weighted Citation Impact
- 0.00