dRAE: 具有超球面编码的表示自编码器

Hugging Face Daily Papers 论文

摘要

本文提出Hyper-Spherical Quantization (HSQ)来解决将视觉表示离散化时的码本崩溃问题,实现了高保真重建和可扩展的码本预算,最高可达131,072,利用率达100%。

在本工作中,我们旨在将高维视觉表示离散化,以弥合与语言模型之间的差距——这是一个不小的挑战,因为现有的量化方法存在码本崩溃问题,无法在保持语义一致性的同时进行扩展。我们识别出根本原因是度量不匹配:标准的欧几里得码本目标与表示空间的各向异性几何从根本上不匹配,导致码本嵌入具有高方差的尺度大小和不均匀的角度分布,阻碍了可扩展性。为了解决这个问题,我们提出了Hyper-Spherical Quantization (HSQ),它通过角度路由将语义内容与特征量级解耦,防止码本分配被量级而非语义所主导。由此得到的离散表示自编码器 (dRAE) 实现了高保真重建,同时保持了语义完整性并支持可扩展的码本预算。大量实验表明,随着词汇表大小扩展到131,072,性能持续提升,同时实现了100%的码本利用率、简化的训练流程,以及在理解和生成任务上的强劲表现。
查看原文
查看缓存全文

缓存时间: 2026/07/28 06:33

论文页 - dRAE: 超球面码的表示自编码器

来源:https://huggingface.co/papers/2607.22148

摘要

在这项工作中,我们旨在对高维视觉表示进行离散化,以弥合与语言模型之间的鸿沟——这是一个非平凡挑战,因为现有的量化方法存在码本崩溃问题,在扩展时难以保持语义一致性。我们将根本原因归因于度量不匹配:标准欧几里得码本目标与表示空间的各向异性几何结构根本不一致,导致码本嵌入出现高方差幅度尺度和不均匀的角度分布,阻碍了可扩展性。为解决此问题,我们提出超球面量化(HSQ),通过角度路由将语义内容与特征幅度解耦,防止码分配被尺度而非语义所主导。由此产生的离散表示自编码器(dRAE)实现了高保真重建,同时保持了语义完整性,并支持可扩展的码本预算。大量实验表明,当词汇量扩展到131,072时,性能持续提升,同时实现100%的码本利用率、简化的训练流程,以及在理解和生成任务上的强劲表现。

查看 arXiv 页面 (https://arxiv.org/abs/2607.22148) 查看 PDF (https://arxiv.org/pdf/2607.22148) 项目页面 (https://drae-hsq.github.io/) GitHub3 (https://github.com/martian422/dRAE) 添加到收藏 (https://huggingface.co/login?next=%2Fpapers%2F2607.22148)

在您的智能体中获取本论文:

hf papers read 2607\.22148

没有最新的 CLI?curl \-LsSf https://hf\.co/cli/install\.sh \| bash

引用本论文的模型0

没有模型链接本论文

在模型 README.md 中引用 arxiv.org/abs/2607.22148 即可从本页面链接该论文。

引用本论文的数据集0

没有数据集链接本论文

在数据集 README.md 中引用 arxiv.org/abs/2607.22148 即可从本页面链接该论文。

引用本论文的空间0

没有空间链接本论文

在空间 README.md 中引用 arxiv.org/abs/2607.22148 即可从本页面链接该论文。

包含本论文的收藏集0

没有收藏集包含本论文

将本论文添加到收藏集 (https://huggingface.co/new-collection) 即可从本页面链接该论文。

相似文章

D-Quant:面向 KV 缓存量化的可漂移熵编码

arXiv cs.CL

D-Quant 是一种灵活的 KV 缓存量化框架,采用可漂移熵编码将可变长度表示转换为固定大小的比特流,从而在注意力内核中实现高效的并行反量化,并实现高达 7 倍的内存减少和 3.5 倍的吞吐量提升,同时保持接近 BF16 的性能。

HiRAE: 带残差预算的层级表征自编码

Hugging Face Daily Papers

HiRAE 是一种层级表征自编码器,它提高了图像重建保真度和生成模型兼容性,降低了FID并增强了文本到图像生成中的对齐性。

稀疏自编码器作为符号匹配的共享坐标系

Hugging Face Daily Papers

论文提出 SAE-IDF,利用稀疏自编码器和按稀有度加权的特征构建共享坐标系,以在异构系统间匹配符号单元;在 12 项任务上性能超越基于稠密表示的 LLM 基线,同时保持人类可解释性。