用双曲几何改进量化,让数据层次结构更精准
Hyperbolic Residual Quantization: Discrete Representations for Data with Latent Hierarchies
- 在双曲空间中做残差量化,天然适配层次分支结构
- 在词语层级任务中提升达20%,优于传统欧氏方法
- 适合处理知识图谱、分类体系等有潜在层级的数据
层次化数据广泛存在于生物分类、组织架构、法律条文和知识图谱等领域。残差量化(RQ)通过多级码本迭代量化残差,生成离散的多标记表示,但其依赖欧几里得几何,难以准确建模层级分支结构。本文提出双曲残差量化(HRQ),将数据原生嵌入双曲流形,采用双曲运算与距离度量进行残差量化。通过调整嵌入网络、残差计算和距离度量以适应双曲几何,HRQ引入了与层次分支自然契合的归纳偏置。我们在两个任务上评估:监督层级建模(使用WordNet同义词树)和层级发现(数据中存在潜在层级但不直接训练)。在两种场景下,HRQ生成的层次化标记在下游任务中表现均优于欧氏RQ,层级建模任务最高提升20%。结果表明,将双曲几何融入离散表示学习可显著增强捕捉潜在层级的能力。
原文摘要 · Abstract (English)
Hierarchical data arise in countless domains, from biological taxonomies and organizational charts to legal codes and knowledge graphs. Residual Quantization (RQ) is widely used to generate discrete, multitoken representations for such data by iteratively quantizing residuals in a multilevel codebook. However, its reliance on Euclidean geometry can introduce fundamental mismatches that hinder modeling of hierarchical branching, necessary for faithful representation of hierarchical data. In this work, we propose Hyperbolic Residual Quantization (HRQ), which embeds data natively in a hyperbolic manifold and performs residual quantization using hyperbolic operations and distance metrics. By adapting the embedding network, residual computation, and distance metric to hyperbolic geometry, HRQ imparts an inductive bias that aligns naturally with hierarchical branching. We claim that HRQ in comparison to RQ can generate more useful for downstream tasks discrete hierarchical representations for data with latent hierarchies. We evaluate HRQ on two tasks: supervised hierarchy modeling using WordNet hypernym trees, where the model is supervised to learn the latent hierarchy - and hierarchy discovery, where, while latent hierarchy exists in the data, the model is not directly trained or evaluated on a task related to the hierarchy. Across both scenarios, HRQ hierarchical tokens yield better performance on downstream tasks compared to Euclidean RQ with gains of up to $20\%$ for the hierarchy modeling task. Our results demonstrate that integrating hyperbolic geometry into discrete representation learning substantially enhances the ability to capture latent hierarchies.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。