Certified Bi-Lipschitz Recurrent Cell
Implementation & benchmark of arXiv:2607.10026 — Robustly Invertible Nonlinear Dynamics and the BiLipREN: From Inversion-Based Control to Generative Trajectory Modelling
Source paper: Robustly Invertible Nonlinear Dynamics and the BiLipREN: From Inversion-Based Control to Generative Trajectory Modelling arXiv:2607.10026 ⓘ · analyzed Aug 30, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Replace an unconstrained RNN or state-space layer with an implicit recurrent cell whose nonlinear algebraic loop is well posed and whose forward dynamics are contracting and strongly input-output monotone. The same certificate guarantees a causal inverse with bounded gain, so sequence predictions should be insensitive to initial-state mismatch while remaining responsive to input perturbations.
Formulas
Mathematical statement
Let x_t in R^n be the hidden state, u_t in R^m the input, w_t in R^q an implicit activation variable, and y_t in R^m the output. Define v_t = C_1 x_t + D_12 u_t + D_11 w_t + b_v and w_t = sigma(v_t), where every scalar activation is slope restricted by 0 <= (sigma(a) - sigma(b)) / (a - b) <= 1. The state and output equations are x_(t+1) = A x_t + B_1 w_t + B_2 u_t + b_x and y_t = C_2 x_t + D_21 w_t + D_22 u_t + b_y. A sufficient certificate uses P positive definite and positive scalars sigma and eta satisfying the paper's dissipativity LMI, with supply-rate matrices Q = -sigma I, S = I, and R = -eta I. The resulting incremental dynamics obey a contraction inequality V_(t+1) <= rho V_t + gamma times the squared input difference, where V_t = delta_x_t^T P delta_x_t and rho is less than one. Strong input-output monotonicity gives delta_y_t^T delta_u_t >= sigma times the squared input difference plus eta times the squared output difference. The inverse swaps sigma and eta, so forward and inverse gains are both finite.
Implementation notes
1. Integration point: implement this cell as a drop-in replacement for the recurrent block in a small GRU, vanilla RNN, or state-space model. Use hidden dimension n, input and output dimension m, and choose q approximately equal to n. The implicit activation loop is solved by 10 to 30 fixed-point iterations, initialized with the previous timestep's w or with zero. For a conservative MVP, parameterize D_11 as tanh(Dhat_11) divided by s with s greater than one, so that its spectral norm is below one. 2. Pseudocode: given x_t and u_t, set b_w = C_1 x_t + D_12 u_t + b_v; iterate w <- sigma(D_11 w + b_w) until the fixed-point residual is below tolerance; then compute x_(t+1) = A x_t + B_1 w + B_2 u_t + b_x and y_t = C_2 x_t + D_21 w + D_22 u_t + b_y. Train with the task loss plus an optional inverse-consistency loss using a tied inverse cell. 3. Compute from the paper: the architecture, slope restriction, dissipativity conditions, and storage-matrix certificate. Estimate empirically: the largest closed-loop Jacobian singular value, contraction factor rho, and input-output gains. For the full version, optimize a Cholesky factor P = L L^T and penalize positive eigenvalues of the LMI residual. 4. First experiment: compare this cell with an unconstrained tanh RNN of equal parameter count on sequential MNIST or the copy task. Perturb the initial hidden state and measure ||delta_x_t|| divided by ||delta_x_0||. The prediction is exponential decay with slope approximately log(rho), and the measured slope should agree with the certificate within 20 percent. Sweep the recurrent scale and locate the divergence boundary where the estimated largest closed-loop Jacobian singular value crosses one. The certified cell should remain bounded below this boundary, while the unconstrained RNN should show exploding sensitivity near the boundary. Also measure inverse reconstruction gain; it should increase sharply as eta approaches zero.
Verification
Stage 1 · Toy mechanism gate: Passed ✓
Stage 2 · Mechanism transferred to benchmark: Confirmed ✓
Stage 2 · Practical benchmark result: Significantly worse ✗
Methodology: 8 paired seeds, permutation test (p<0.05) against a learning-rate-tuned baseline. How verification works
Stage 1 — Mechanism check agent confidence 8/10
Built an implicit tanh REN cell with fixed-point algebraic-loop solving, an explicit scalar version of the paper’s LMI certificate, Jacobian/contraction diagnostics, monotonicity checks, and an unconstrained tanh-RNN comparison. The mechanism manifested: the fixed-point sweep transitioned at |D11|=1 as predicted, and the Jacobian crossed one at B1=0.806 versus the conservative predicted B1=0.75, a 7.4% discrepancy. The certified cell also had positive LMI margin, lower sequence MSE, and much lower initial-state sensitivity. The finite upper monotonicity boundary was not reached in the tested D12 range, so that part remains unverified.
- Agent confidence
- 8/10
- Baseline
- Unconstrained scalar tanh RNN: MSE=0.23964; mean output sensitivity to initial-state perturbation=0.47598.
- Idea
- Implicit certified REN: LMI minimum eigenvalue=0.33391; MSE=0.001926; mean initial-state sensitivity=0.02706; fixed-point transition at D11=1.0; observed Jacobian crossing B1=0.8056 versus predicted 0.75; output gain increased from 0.353 at D12=0.02 to 1.077–1.57 across the sweep while the strong-monotonicity margin remained positive.
Limitations: Only scalar toy systems and a hand-configured one-dimensional sequence task were tested; there was no training, sequential-MNIST or copy task, multidimensional P optimization, inverse-cell reconstruction experiment, FLOP/speed study, or broad random-seed sweep. The monotonicity sweep did not extend to a certificate-margin failure boundary.
How to run: python3 experiment.py
Stage 2 — Benchmark vs. tuned baseline (arXiv:2607.10026)
Significantly WORSE than the tuned baseline (+51.7%, p=0.0257, wins 1 / 8 paired seeds).
- Benchmark
- Dynamics forecasting
- Model
- rnn_small
- Paired seeds
- 8
- Baseline mean
- 0.0007
- Idea mean
- 0.0011
- Effect (Δ)
- 0.0004 (+51.7%; negative = idea better)
- Wins
- 1 / 8 paired seeds
- p-value
- 0.0257 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±34.7%
- Mechanism
- Confirmed ✓
- Practical effect
- Hurts
- Baseline tuning
- swept over 3 configs
Only the registered dynamics track was tested. No full matrix-valued optimized P/LMI certificate, inverse reconstruction task, longer training budget, or broader architectural sweep was evaluated.
How to run: python3 bench_stage2.py
Verdict computed by deterministic test code from paired-seed statistics — not by the language model.
Stage 2 — Benchmark latest Failed ✗
Agent confidence: 9/10
The certified implicit recurrent cell was evaluated against a matched GRU baseline on the registered dynamics track using paired seeds, shared training budget, and a parity-complete learning-rate sweep. The mechanism signature was confirmed by trained-model behavior, but the idea lost significantly on task MSE: 0.0011225013 versus 0.0007400292, paired permutation p=0.02575.
- Baseline
- rnn_small GRU, best lr=0.006, test MSE mean=0.0007400291942758486, std=0.00016879101698107971
- Idea
- Certified implicit tanh recurrent cell, best lr=0.006, test MSE mean=0.0011225013367948122, std=0.0002539576109580576
Benchmark result
Significantly WORSE than the tuned baseline (+51.7%, p=0.0257, wins 1 / 8 paired seeds).
- Benchmark
- Dynamics forecasting
- Model
- rnn_small
- Paired seeds
- 8
- Baseline mean
- 0.0007
- Idea mean
- 0.0011
- Effect (Δ)
- 0.0004 (+51.7%; negative = idea better)
- Wins
- 1 / 8 paired seeds
- p-value
- 0.0257 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±34.7%
- Mechanism
- Confirmed ✓
- Practical effect
- Hurts
- Baseline tuning
- swept over 3 configs
Only the registered dynamics track was tested. No full matrix-valued optimized P/LMI certificate, inverse reconstruction task, longer training budget, or broader architectural sweep was evaluated.
How to run: python3 bench_stage2.py
Artifacts
- bench_report.json 2.6 KB View Raw
- bench_stage2.py 4.6 KB View Raw
- experiment.py 6.7 KB View Raw
- report.md 1.7 KB View
- report_bench_2026-08-30T152819.md 4.1 KB View
- results.json 6.9 KB View Raw