Intrinsic Lorentz Neural Network
Organizations: University of Science and Technology of China · University of Trento · Peking University
Abstract
Real-world data frequently exhibit latent hierarchical structures, which can be naturally represented by hyperbolic geometry. Although recent hyperbolic neural networks have demonstrated promising results, many existing architectures remain partially intrinsic, mixing Euclidean operations with hyperbolic ones or relying on extrinsic parameterizations. To address it, we propose the Intrinsic Lorentz Neural Network (ILNN), a fully intrinsic hyperbolic architecture that conducts all computations within the Lorentz model. At its core, the network introduces a novel point-to-hyperplane fully connected layer (FC), replacing traditional Euclidean affine logits with closed-form hyperbolic distances from features to learned Lorentz hyperplanes, thereby ensuring that the resulting geometric decision functions respect the inherent curvature. Around this fundamental layer, we design intrinsic modules: GyroLBN, a Lorentz batch normalization that couples gyro-centering with gyro-scaling, consistently outperforming both LBN and GyroBN while reducing training time. We additionally proposed a gyro-additive bias for the FC output, a Lorentz patch-concatenation operator that aligns the expected log-radius across feature blocks via a digamma-based scale, and a Lorentz dropout layer. Extensive experiments conducted on CIFAR-10/100 and two genomic benchmarks (TEB and GUE) illustrate that ILNN achieves state-of-the-art performance and computational cost among hyperbolic models and consistently surpasses strong Euclidean baselines. The code is available at magenta{this url}.
Figures & tables
| Models | Airport | Cora | PubMed |
|---|---|---|---|
| GCN ( Kipf & Welling, 2017 ) | |||
| GAT ( Veličković et al., 2017 ) | |||
| SGC ( Wu et al., 2019 ) | |||
| HGNN ( Liu et al., 2019 ) | |||
| HGCN ( Chami et al., 2019 ) | |||
| HGAT ( Chami et al., 2019 ) |
Appendix figures & tables1 asset
Supplementary material from the paper’s appendix.
Appendix
| Hyperparameter | CIFAR-10&100 | TEB | GUE |
|---|---|---|---|
| Epochs | 200 | 150 | 150 |
| Batch size | 128 | 256 | 256 |
| Learning rate (LR) | 1e-1 | 8e-4 | 9e-4 |
| Drop LR epochs | 60, 120, 160 | 100,130 | 100,130 |
| Drop LR gamma | 0.2 | 0.1 | 0.1 |
| Weight decay | 5e-4 | 6e-3 | 5e-3 |
Hyperbolic Representation Learning
Hyperbolic Representation Learning uses negatively curved spaces to encode hierarchies and scale-free relations in compact, geometry-aware embeddings.
See all 86 papers