arXiv · 2409.16897
HVT: A Comprehensive Vision Framework for Learning in Non-Euclidean Space
Abstract
Data representation in non-Euclidean spaces has proven effective for capturing hierarchical and complex relationships in real-world datasets. Hyperbolic spaces, in particular, provide efficient embeddings for hierarchical structures. This paper introduces the Hyperbolic Vision Transformer (HVT), a novel extension of the Vision Transformer (ViT) that integrates hyperbolic geometry. While traditional ViTs operate in Euclidean space, our method enhances the self-attention mechanism by leveraging hyperbolic distance and M\"obius transformations. This enables more effective modeling of hierarchical and relational dependencies in image data. We present rigorous mathematical formulations, showing how hyperbolic geometry can be incorporated into attention layers, feed-forward networks, and optimization. We offer improved performance for image classification using the ImageNet dataset.
Explore related subjects
Keep this discovery
Jacob Fein-Ashley, Ethan Feng, Minh Pham. 2024-09-25. HVT: A Comprehensive Vision Framework for Learning in Non-Euclidean Space. https://arxiv.org/abs/2409.16897
Cite the original work for its findings. Save a collection to share your selection of sources.