用双曲几何提升视觉Transformer,让模型更稳定、更准确。
HexFormer: Hyperbolic Vision Transformer with Exponential Map Aggregation
- 在注意力机制中引入指数映射聚合,替代传统平均方式。
- 在多个数据集上优于欧式基线和已有双曲ViT,混合版本表现最佳。
- 双曲模型梯度更稳定,对预热策略不敏感,训练更高效。
跨模态数据如图像、文本和图常包含层次与关系结构,难以用欧氏几何建模。双曲几何天然适合表示此类结构。本文提出HexFormer,一种基于指数映射聚合的双曲视觉Transformer,用于图像分类。设计了两种变体:纯双曲编码器(HexFormer)和双曲编码器+欧式分类头的混合版本(HexFormer-Hybrid)。新注意力机制通过指数映射聚合,生成更准确稳定的表征,性能优于标准中心点平均。在多个数据集上的实验表明,该方法持续超越欧氏基线和先前双曲ViT,混合版本表现最强。同时分析显示,双曲模型梯度更稳定,对学习率预热策略敏感度更低,训练更具鲁棒性与效率。结果表明,双曲几何可有效提升视觉Transformer的精度与训练稳定性,且简单机制如指数映射聚合即具显著实用价值。
原文摘要 · Abstract (English)
Data across modalities such as images, text, and graphs often contains hierarchical and relational structures, which are challenging to model within Euclidean geometry. Hyperbolic geometry provides a natural framework for representing such structures. Building on this property, this work introduces HexFormer, a hyperbolic vision transformer for image classification that incorporates exponential map aggregation within its attention mechanism. Two designs are explored: a hyperbolic ViT (HexFormer) and a hybrid variant (HexFormer-Hybrid) that combines a hyperbolic encoder with an Euclidean linear classification head. HexFormer incorporates a novel attention mechanism based on exponential map aggregation, which yields more accurate and stable aggregated representations than standard centroid based averaging, showing that simpler approaches retain competitive merit. Experiments across multiple datasets demonstrate consistent performance improvements over Euclidean baselines and prior hyperbolic ViTs, with the hybrid variant achieving the strongest overall results. Additionally, this study provides an analysis of gradient stability in hyperbolic transformers. The results reveal that hyperbolic models exhibit more stable gradients and reduced sensitivity to warmup strategies compared to Euclidean architectures, highlighting their robustness and efficiency in training. Overall, these findings indicate that hyperbolic geometry can enhance vision transformer architectures by improving gradient stability and accuracy. In addition, relatively simple mechanisms such as exponential map aggregation can provide strong practical benefits.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。