计算机科学 > 机器学习
[提交于 2023年6月1日
(v1)
,最后修订 2024年7月16日 (此版本, v2)]
标题: FlexRound:基于逐元素除法的可学习量化后训练量化方法
标题: FlexRound: Learnable Rounding based on Element-wise Division for Post-Training Quantization
摘要: 后训练量化(PTQ)由于不需要完整的训练数据集或端到端训练,因此在资源受限设备上部署深度神经网络方面越来越受欢迎。 由于基于重建每一层或块输出的PTQ方案被证明能有效提高量化模型性能,最近的工作开发了算法来设计和学习一种新的权重舍入方案,以更好地重建每一层或块输出。 在本工作中,我们提出了一种简单但有效的新的权重舍入机制,称为\emph{灵活舍入},它基于逐元素除法而不是典型的逐元素加法,从而使FlexRound能够联合学习一个通用的量化网格大小以及每个预训练权重的不同尺度。 由于逐元素除法引起的导数倒数规则,FlexRound本质上能够在更新其相应尺度时利用预训练权重,从而根据它们的大小灵活地量化预训练权重。 我们在各种模型和任务上实证验证了FlexRound的有效性。 据我们所知,我们的工作是首次对图像分类、自然语言理解和自然语言生成任务进行全面实验。 此外,我们首次展示了大型语言模型可以高效量化,与半精度基线相比仅对性能产生可忽略的影响,这是通过逐块重建输出实现的。 我们的代码可在\url{https://github.com/onliwad101/FlexRound_LRQ}获取。
文献和引用工具
与本文相关的代码,数据和媒体
alphaXiv (什么是 alphaXiv?)
CatalyzeX 代码查找器 (什么是 CatalyzeX?)
DagsHub (什么是 DagsHub?)
Gotit.pub (什么是 GotitPub?)
Hugging Face (什么是 Huggingface?)
带有代码的论文 (什么是带有代码的论文?)
ScienceCast (什么是 ScienceCast?)
演示
推荐器和搜索工具
arXivLabs:与社区合作伙伴的实验项目
arXivLabs 是一个框架,允许合作伙伴直接在我们的网站上开发和分享新的 arXiv 特性。
与 arXivLabs 合作的个人和组织都接受了我们的价值观,即开放、社区、卓越和用户数据隐私。arXiv 承诺这些价值观,并且只与遵守这些价值观的合作伙伴合作。
有一个为 arXiv 社区增加价值的项目想法吗? 了解更多关于 arXivLabs 的信息.