Temporal Geometry of Deep Networks: Hyperbolic Representations of Training Dynamics for Intrinsic Explainability
cs.LG, cs.AI, cs.NE, physics.data-an
Submitted: 2026-10-02
Updated: 2026-10-02
Comments: Published as a main conference paper at ICLR 2026. 10 Main Pages, 22 pages of supplementary material
Code: https://github.com/xxlya/BrainGNN_Pytorch
License: http://creativecommons.org/licenses/by/4.0/
The gist: Intrinsic explainability remains a challenging problem, particularly in contexts where multilayer perceptrons (MLPs) require dynamic re-training within an optimization environment.
Terminology
Abstract
Intrinsic explainability remains a challenging problem, particularly in contexts where multilayer perceptrons (MLPs) require dynamic re-training within an optimization environment. This paper investigates how MLPs and their training dynamics can be represented and studied in non-Euclidean spaces; our representation features the Poincaré model of hyperbolic geometry. We aim to capture the geometric evolution of their weighted topology and self-organization over time. Instead of restricting the analysis to single checkpoints---as per established measure-based explainability methods---we construct temporal parameter graphs, i.e., snapshots over time T steps of the optimization/training process for MLPs. This reflects the view that neural networks encode information not only in their weights but also in the trajectory traced during training. Drawing on the idea that many complex networks admit embeddings in hidden metric spaces where distances correspond to connection likelihood, we present a geometric and temporal graph-based metalearning framework for obtaining dynamic hyperbolic representations of the underlying neural parameter graphs. Our model embeds temporal parameter graphs in the Poincaré model ball, and learns from them while maintaining equivariance to within-snapshot neuron permutations and invariance to permutations of past snapshots. In doing so, the approach preserves functional equivalence over time and recovers the latent evolving geometry of the network. Experiments on regression and classification tasks with trained MLPs show strong meta-network performance, accompanied by hyperbolic temporal representations. This reveals how the network structure emerges over time under specific training environments, thus providing insights into the network's self-organization.
Sources
- Geometric Deep Learning: Grids, Groups, Graphs, Geodesics, and Gauges
- Riemannian Adaptive Optimization Methods
- DynDepNet: Learning Time-Varying Dependency Structures from fMRI Data via Dynamic Graph Structure Learning
- Hyperbolic Graph Convolutional Neural Networks
- Trust-Region Eigenvalue Filtering for Projected Newton
- A Simple Framework for Contrastive Learning of Visual Representations
- Fully Hyperbolic Neural Networks
- BrainNNExplainer: An Interpretable Graph Neural Network Framework for Brain Network based Disease Analysis
- Deep Learning on Implicit Neural Representations of Shapes
- From data to functa: Your data point is a function and you can treat it like one
- Classifying the classifier: dissecting the weight space of neural networks
- DynGEM: Deep Embedding Method for Dynamic Graphs
- Hyperbolic embedding of multilayer networks
- Rethinking Functional Brain Connectome Analysis: Do Graph Deep Learning Models Help
- Graph Neural Networks for Learning Equivariant Representations of Neural Networks
- Understanding Generalization in Deep Learning via Tensor Methods
- Federated Learning with Heterogeneous Architectures using Graph HyperNetworks
- DARTS: Differentiable Architecture Search
- Structure Is Not Enough: Leveraging Behavior for Neural Network Weight Reconstruction
- Equivariant Deep Weight Space Alignment
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks