AdaPTwin:面向 Transformer 中的产品双胞胎的低成本自适应压缩
机器学习
2024-06-14 v1 声音
音频与语音处理
摘要
虽然基于大型 Transformer 模型在说话人无关语音识别中展现出卓越的性能,但其庞大的规模和计算要求使得在资源受限的环境中昂贵或不可行。本文我们提出一种低秩自适应压缩技术,称为 AdaPTwin,通过联合压缩 Transformer 注意力层中产品相关的权重矩阵对。我们的做法可以在保持对新说话人和声学条件的通用性的同时,优先考虑压缩后模型在特定说话人上的性能。值得注意的是,我们的方法仅需 8 小时的语音数据进行微调,微调时间不到 20 分钟,成本效益极高。我们通过将 Whisper 和 Distil-Whisper 模型压缩最高达 45%,且仅增加不到 2% 的词错误率来展示我们方法的有效性。
引用
@article{arxiv.2406.08904,
title = {AdaPTwin: Low-Cost Adaptive Compression of Product Twins in Transformers},
author = {Emil Biju and Anirudh Sriram and Mert Pilanci},
journal= {arXiv preprint arXiv:2406.08904},
year = {2024}
}
备注
12 pages, 3 figures, submitted to NeurIPS 2024