# New non-Euclidean neural quantum states from hyperbolic Lorentz recurrent architectures

arXiv: https://arxiv.org/abs/2604.24337 · HTML: https://mlsanity.com/paper/2604.24337/

## Key facts

- Cited by 1 papers in the map (data through 14 Oct 2026).
- Builds on 18 of its 50 references that are in the map.
- First posted on arXiv 27 Apr 2026.

## Builds on (most cited first, 18 of 18)

- [Poincar'e Embeddings for Learning Hierarchical Representations](https://mlsanity.com/paper/1705.08039/) (2017)
- [Hyperbolic Neural Networks](https://mlsanity.com/paper/1805.09112/) (2018)
- [Representation Tradeoffs for Hyperbolic Embeddings](https://mlsanity.com/paper/1804.03329/) (2018)
- [Riemannian Adaptive Optimization Methods](https://mlsanity.com/paper/1810.00760/) (2018)
- [Hyperbolic Entailment Cones for Learning Hierarchical Embeddings](https://mlsanity.com/paper/1804.01882/) (2018)
- [Hyperbolic Deep Neural Networks: A Survey](https://mlsanity.com/paper/2101.04562/) (2021)
- [Hyperbolic Attention Networks](https://mlsanity.com/paper/1805.09786/) (2018)
- [Continuous Hierarchical Representations with Poincar'e Variational Auto-Encoders](https://mlsanity.com/paper/1901.06033/) (2019)
- [Constant Curvature Graph Convolutional Networks](https://mlsanity.com/paper/1911.05076/) (2019)
- [Fully Hyperbolic Neural Networks](https://mlsanity.com/paper/2105.14686/) (2021)
- [Gated Feedback Recurrent Neural Networks](https://mlsanity.com/paper/1502.02367/) (2015)
- [A Fully Hyperbolic Neural Model for Hierarchical Multi-Class Classification](https://mlsanity.com/paper/2010.02053/) (2020)
- [HyperCore: The Core Framework for Building Hyperbolic Foundation Models with Comprehensive Modules](https://mlsanity.com/paper/2504.08912/) (2025)
- [Recurrent neural network wave functions for Rydberg atom arrays on kagome lattice](https://mlsanity.com/paper/2405.20384/) (2024)
- [Supplementing Recurrent Neural Network Wave Functions with Symmetry and Annealing to Improve Accuracy](https://mlsanity.com/paper/2207.14314/) (2022)
- [Are queries and keys always relevant? A case study on Transformer wave functions](https://mlsanity.com/paper/2405.18874/) (2024)
- [Hyperbolic recurrent neural network as the first type of non-Euclidean neural quantum state ansatz](https://mlsanity.com/paper/2505.22083/) (2025)
- [Deep Learning Calabi-Yau four folds with hybrid and recurrent neural network architectures](https://mlsanity.com/paper/2405.17406/) (2024)

## Built on by (most cited first, 1 of 1)

- [Hyperbolic Restricted Boltzmann Machine Neural Quantum State](https://mlsanity.com/paper/2609.26032/) (2026)

## Abstract (arXiv)

In this work, we construct new non-Euclidean neural quantum states (NQS) based on hyperbolic Lorentz recurrent architectures (RNN/GRU). These constructions, together with the Poincare RNN NQS also newly constructed here, extend the class of previously introduced non-Eucllidean NQS which consists only of Poincare hyperbolic GRU. Using the Heisenberg J1J2 and J1J2J3 models consisting of 100 spins in the Variational Monte Carlo (VMC) setting, we show that the four hyperbolic RNN/GRU NQS variants are always able to furnish better representations of the ground state wavefunctions of the quantum systems than their respective Euclidean counterparts with the same architecture. In our experiments, among the four hyperbolic NQS, Lorentz RNN stands out in particular because despite having almost three times fewer parameters, it is capable of surpassing the more complex Poincare GRU and Lorentz GRU to emerge as the best overall hyperbolic NQS ansatz on many instances involving different J2 and (J2,J3) couplings. Given the findings from this work showing that the four newly constructed hyperbolic RNN/GRU NQS ansatze are able to outperform the well-established Euclidean RNN/GRU NQS in Heisenberg spin models, we establish the utility and efficiency of the hyperbolic Lorentz RNN/GRU NQS as well as the Poincare RNN/GRU NQS for future variational studies of quantum many-body systems, especially those exhibiting a hierarchical structure in the form of the different degrees of nearest-neighbor interactions.