arrow
返回

Better than Optimal: Improving Adaptive Stochastic Quantization Using Shared Randomness

delete2025-12-01
delete0
PRE
AI
R
Ran Ben Basat *
Y
Yaniv Ben-Itzhak
M
Michael Mitzenmacher
S
Shay Vargaftik
DOI:10.1145/3771564delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
量化用于将高精度数据转换为低精度表示,为减少机器学习任务中的内存使用、计算复杂度和通信成本提供了基础技术。最近的研究表明,自适应随机量化(ASQ)方法——即在输入分布未知给解量化器的情况下提供无偏优化的量化值——对于实时压缩大型张量是高效的。本文引入了自适应无偏量化(AUQ)问题,将其推广至允许量化器和解量化器共享已知随机性的场景。我们证明了两种问题可实现精度之间的渐近差距,显示AUQ可实现二次方的更高精度。随后,我们介绍了Simba,一种AUQ技术,可提升最优ASQ方法的精度。直观上,Simba使用多组量化值,且用于量化特定值的组别由共享随机性决定。有趣的是,我们发现Simba通过利用近似ASQ解作为起点进行改进,可同时提升最优ASQ方法的速度和精度。实测表明,对于常见分布,Simba比当前最优ASQ方法QUIVER最高提升约24倍的精度,同时速度提升达两个数量级。最后,我们演示了Simba可通过替换QUIC-FL(一种先进的分布式均值估计技术)基于求解器的查找表为Simba计算的表,来提升其精度。
Keyword:
Compression
Quantization
Shared Randomness
Machine Learning
Stochastic Quantization
Unbiasedness

期刊

P
Proceedings of the ACM on Measurement and Analysis of Computing Systems
IF:
2.7
论文数:
45
被引数:
1.0K

机构

B
broadcom
学者数:
363
论文数: 242
被引数: 0
U
university college london
学者数:
8.9K
论文数: 4.8K
被引数: 1
U
university of london
学者数:
21.5W
论文数: 19.7W
被引数: 305
学者 查看更多机构
引用论文

引用论文

Q-BERT: Hessian Based Ultra Low Precision Quantization of BERT
err2020-04-03
err0
errOAAI
errSheng Shen; Zhen Dong; Jiayu Ye; Linjian Ma; Zhewei Yao; Amir Gholami; Michael W. Mahoney; Kurt Keutzer
err分享
err收藏
Beyond Throughput and Compression Ratios: Towards High End-to-end Utility of Gradient Compression
err
IF0
err2024-11-18
err0
PREAI
errWenchen Han; Shay Vargaftik; Michael Mitzenmacher; Brad Karp; Ran Ben Basat
err分享
err收藏
Detecting routing loops in the data plane
err2020-11-23
err0
PREAI
errKučera,Jan; Basat,Ran Ben; Kuka,Mário; Antichi,Gianni; Yu,Minlan; Mitzenmacher,Michael
err分享
err收藏
没有更多内容