Back to Search Start Over

HVT: A Comprehensive Vision Framework for Learning in Non-Euclidean Space

Authors :
Fein-Ashley, Jacob
Feng, Ethan
Pham, Minh
Publication Year :
2024

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.

Details

Database :
arXiv
Publication Type :
Report
Accession number :
edsarx.2409.16897
Document Type :
Working Paper