arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2610.12474cs.SDcs.LGeess.AS

TTS-Q:感知引导的文本到语音扩散Transformer的后训练量化

TTS-Q: Perception-Guided Post-Training Quantization of Text-to-Speech Diffusion Transformers

Xiaoqin Feng, Yuqi Li, Weilun Feng, Yi-Cheng Lin, Kai Li, Xianglong Wang, Kuo Yang, Chuanguang Yang, Yingli Tian, Wei Li, Zhongwei Cheng, Lin Chen

AI总结:

本文针对TTS扩散Transformer的后训练量化问题,提出感知引导的TTS-Q方法,通过混合精度调度恢复4位量化后的音色,同时降低GPU内存,为资源受限的TTS部署提供参考。

摘要:

零样本文本到语音(TTS)可从数秒音频中克隆语音,但其质量源于数十亿参数的扩散Transformer,该模型的规模和多步采样使其无法应用于最需要它的设备端、辅助及实时场景。后训练量化(PTQ)可在无需重新训练的情况下缩小此类模型,但它是为语言和视觉任务设计的,而非语音任务。在4位量化下,高性能量化器虽能让可懂度接近全精度水平,但会持续破坏说话人音色,导致错误率保持较低的同时,语音偏离目标说话人。本文探究量化为何会损害语音、损伤发生在何处,以及能否在不重新训练的情况下修复损伤。为此,我们对10亿参数和35亿参数的TTS扩散Transformer开展了首个4位权重与激活(W4A4)PTQ研究。音色损伤源于权重和激活量化,但仅激活部分具备可被精度调度利用的时间步结构。因此,我们提出TTS-Q,该方法从任务匹配的校准候选中选取难量化项,并应用预算均衡的混合精度调度,为后期前馈激活分配更高精度。TTS-Q在两种规模下均能最佳恢复音色,且不损失可懂度。此外,部署的校准W4A4基准可降低GPU峰值内存,为资源受限的TTS部署提供实用参考。

英文摘要:

Zero-shot text-to-speech (TTS) can clone a voice from seconds of audio. Yet the quality comes from billion-parameter diffusion transformers whose size and multi-step sampling put them out of reach of the on-device, assistive, and real-time applications that need them most. Post-training quantization (PTQ) shrinks such models without retraining, but it has been built for language and vision, not speech. At 4 bits, a strong quantizer leaves intelligibility close to full precision but consistently erodes speaker timbre, so error rates stay low while the voice drifts from the target speaker. We ask why quantization hurts speech, where the damage lives, and whether it can be repaired without retraining. To find out, we run the first 4-bit weight-and-activation (W4A4) PTQ study of a TTS diffusion transformer at 1B and 3.5B. The timbre cost comes from both the weight and the activation quantization, but only the activation part has the timestep structure a precision schedule can exploit. We therefore introduce TTS-Q, which selects hard-to-quantize items from task-matched calibration candidates and applies a budget-balanced mixed-precision schedule that assigns higher precision to late-step feed-forward activations. TTS-Q recovers timbre best at both scales, at no cost to intelligibility. Separately, the deployed calibrated-W4A4 reference reduces peak memory on GPU, providing a practical reference for resource-constrained TTS deployment.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑