Hyperbolic State Transition Regularization
Implementation & benchmark of arXiv:2608.12862 — Topological shadowing for linear operators
Source paper: Topological shadowing for linear operators arXiv:2608.12862 ⓘ · analyzed Sep 1, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Constrain a recurrent or state-space transition matrix so that its eigenvalues avoid a configurable annulus around the unit circle. This creates a stable/unstable decomposition and should reduce the accumulation of numerical, quantization, and activation-update errors over long sequences while preserving controlled long-term memory.
Formulas
Mathematical statement
For a bijection f on a metric space (X,d), the paper defines a delta-pseudo-orbit as a two-sided sequence (x_n)_{n in Z} satisfying d(f(x_n),x_{n+1}) < delta(f(x_n)) for every integer n. It is epsilon-shadowed by x in X if d(f^n(x),x_n) < epsilon(f^n(x)) for every n. Topological shadowing means that every positive continuous tolerance epsilon admits a positive continuous delta for which every delta-pseudo-orbit is epsilon-shadowed. For a finite-dimensional linear operator A, the paper proves the spectral criterion sigma(A) subset {lambda in C: |lambda|<1} union {lambda in C: |lambda|>1}, where sigma(A) is the spectrum and lambda are eigenvalues. Thus no eigenvalue may lie on the unit circle. For a neural transition h_{t+1}=phi(Ah_t+Bx_t), use the practical margin condition ||lambda_i|-1| >= m for every eigenvalue lambda_i of A, with m>0. The regularizer L_gap(A) penalizes violations of this condition: L_gap(A)=(1/N) sum_i [max(0,m-||lambda_i|-1|)]^2, where N is the matrix dimension and m is the desired radial spectral gap. The theorem gives the qualitative hyperbolicity condition; the margin m and its finite-horizon benefit must be selected empirically.
Implementation notes
Integrate the method into the recurrent transition matrix A in h_{t+1}=phi(Ah_t+Bx_t), or into the learned state matrix of an SSM. Start with N=64 or 128 so that a full eigendecomposition is inexpensive. At each training step: (1) compute the complex eigenvalues lambda_i of A; (2) calculate r_i=abs(lambda_i) and the gap penalty L_gap=(1/N)*sum_i*max(0,m-abs(r_i-1))^2; (3) backpropagate through the eigendecomposition when supported; and (4) optimize L_task+alpha*L_gap. Use m=0.05 and alpha values in {1e-4,1e-3,1e-2}. For larger matrices, estimate only eigenvalues nearest the unit circle with Arnoldi iteration, or parameterize A in real 2-by-2 blocks with directly controlled radii. A simpler non-differentiable alternative is a post-update projection: for every eigenvalue with radius r in [1-m,1+m], replace its radius by 1-m if r<1 and by 1+m if r>=1, preserving phase and reconstructing A. The paper supplies the spectral condition, while the margin and optimizer coefficient are empirical hyperparameters. First test a 128-unit tanh RNN against unconstrained initialization, orthogonal initialization, and spectral-radius clipping on copy-memory and permuted sequential-MNIST tasks. Inject Gaussian noise into every recurrent update and evaluate clean accuracy, noisy accuracy, hidden-state divergence over 1,000 steps, gradient norms, and seed variance. The method succeeds if it lowers long-horizon noisy loss and trajectory divergence at equal parameter count and does not substantially reduce clean-sequence accuracy.
Verification
This idea has not been verified yet.
Verification happens in two stages: Stage 1 — a mechanism check on a toy system confirms the claimed mathematical phenomenon reproduces; Stage 2 — a benchmark implements the idea on a real (small) neural network task and compares it against a tuned baseline over 8 paired seeds with a permutation test.
Artifacts
Artifacts unavailable.