用低维组合码本提升向量量化效率,更小代码本性能更强
LooC: Effective Low-Dimensional Codebook for Compositional Vector Quantization
- 将码向量视为特征的低维组合单元,扩大解空间并压缩代码本
- 通过插值外推机制保持细节,实现紧凑代码本下的高保真重建
- 可无缝接入各类VQ任务,适合追求高效压缩的模型开发者
向量量化(VQ)是通过码本近似连续特征向量的基础技术。随着数据与模型复杂度提升,亟需高容量但更紧凑的VQ方法。本文提出LooC——一种用于组合向量量化的有效低维码本方法。首先,通过重构码向量与特征向量的关系,引入参数高效的码本,显著扩展解空间;不再逐个匹配码向量,而是将其视为特征向量中的低维组合单元进行组合,从而实现更紧凑的码本结构并提升性能。其次,LooC引入无参数的插值外推机制,在量化过程中增强和平滑特征,更好保留细节与近似保真度。该设计促使码本全利用率,有效避免码本坍缩问题。第三,LooC可作为即插即用模块,适配多种基于VQ的下游任务。大量实验在不同任务、数据集与架构上验证,LooC以显著更小的码本规模达到业界领先性能。
原文摘要 · Abstract (English)
Vector quantization (VQ) is a prevalent and fundamental technique that discretizes continuous feature vectors by approximating them using a codebook. As the diversity and complexity of data and models continue to increase, there is an urgent need for high-capacity, yet more compact VQ methods. This paper aims to reconcile this conflict by presenting a new approach called LooC, which utilizes an effective Low-dimensional codebook for Compositional vector quantization. Firstly, LooC introduces a parameter-efficient codebook by reframing the relationship between codevectors and feature vectors, significantly expanding its solution space. Instead of individually matching codevectors with feature vectors, LooC treats them as lower-dimensional compositional units within feature vectors and combines them, resulting in a more compact codebook with improved performance. Secondly, LooC incorporates a parameter-free extrapolation-by-interpolation mechanism to enhance and smooth features during the VQ process, which allows for better preservation of details and fidelity in feature approximation. The design of LooC leads to full codebook usage, effectively utilizing the compact codebook while avoiding the problem of collapse. Thirdly, LooC can serve as a plug-and-play module for existing methods for different downstream tasks based on VQ. Finally, extensive evaluations on different tasks, datasets, and architectures demonstrate that LooC outperforms existing VQ methods, achieving state-of-the-art performance with a significantly smaller codebook.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。