Skip to main content

Module hyperbolic

Module hyperbolic 

Source
Expand description

Hyperbolic Attention Module

Implements attention mechanisms in hyperbolic space using:

  • Poincaré ball model (traditional)
  • Lorentz hyperboloid model (novel - faster, more stable)

Re-exports§

pub use poincare::exp_map;
pub use poincare::frechet_mean;
pub use poincare::log_map;
pub use poincare::mobius_add;
pub use poincare::mobius_scalar_mult;
pub use poincare::poincare_distance;
pub use poincare::project_to_ball;
pub use hyperbolic_attention::HyperbolicAttention;
pub use hyperbolic_attention::HyperbolicAttentionConfig;
pub use mixed_curvature::MixedCurvatureAttention;
pub use mixed_curvature::MixedCurvatureConfig;
pub use lorentz_cascade::busemann_score;
pub use lorentz_cascade::einstein_midpoint;
pub use lorentz_cascade::horosphere_attention_weights;
pub use lorentz_cascade::lorentz_distance;
pub use lorentz_cascade::lorentz_inner;
pub use lorentz_cascade::project_hyperboloid;
pub use lorentz_cascade::CascadeHead;
pub use lorentz_cascade::LCAConfig;
pub use lorentz_cascade::LorentzCascadeAttention;

Modules§

hyperbolic_attention
Hyperbolic Attention Mechanism using Poincaré ball model
lorentz_cascade
Lorentz Cascade Attention (LCA) - A Novel Hyperbolic Attention Mechanism
mixed_curvature
Mixed-Curvature Attention combining Euclidean and Hyperbolic spaces
poincare
Poincaré Ball Model Operations for Hyperbolic Geometry