Komuro Time-Warp Expansivity Regularizer
Implementation & benchmark of arXiv:2608.02186 — Komuro Expansivity and Periodic Orbit Growth for Multi-Singular Hyperbolic Flows
Source paper: Komuro Expansivity and Periodic Orbit Growth for Multi-Singular Hyperbolic Flows arXiv:2608.02186 ⓘ · analyzed Aug 31, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Apply Komuro-style expansivity to a continuous-time neural latent flow by requiring distinct latent trajectories to separate even when the second trajectory is allowed an arbitrary increasing time reparametrization. This targets neural ODE world models and irregularly sampled sequence models, where ordinary pointwise separation can mistake clock-speed differences for different states.
Formulas
Mathematical statement
Let phi_t(x) be the neural flow generated by dz/dt=f_theta(z), let d be a latent metric, and let alpha be an increasing homeomorphism with alpha(0)=0. Komuro expansivity states that for every epsilon>0 there is delta(epsilon)>0 such that d(phi_t(x),phi_alpha(t)(y))<delta for all t implies y=phi_s(x) for some |s|<epsilon. In the neural implementation, estimate the worst time-warped discrepancy D_theta(x,y) by minimizing over a restricted monotone warp family, and penalize non-orbit pairs whose discrepancy is below a margin.
Implementation notes
1. Integration point: use a latent neural ODE, controlled differential equation, or continuous-time state-space model with latent state z and vector field f_theta. Add the regularizer to reconstruction or prediction loss after producing two rollouts. Sample negative pairs x,y from different trajectories and positive pairs by setting y=phi_s(x) for a small random time shift s. 2. Pseudocode: integrate z_x(t)=ODESolve(f_theta,z_x(0),grid) and z_y(t); represent a monotone warp as alpha(t)=t+sum_k softplus(a_k)B_k(t), normalized so alpha(0)=0; optimize the warp coefficients for a few gradient steps or use soft-DTW; compute a smooth maximum over aligned distances; apply max(0,m-D)^2 only to negative pairs; update theta with L_task+lambda L_exp. 3. Computed quantities are rollouts, warped distances, and the loss. The theorem's true delta is unavailable, so estimate delta_emp from held-out pairs and vary q, epsilon, and warp complexity. 4. First cheap experiment: train a 2D or 4D neural ODE on a synthetic limit-cycle-plus-saddle system, comparing ordinary contrastive separation with this loss. Measure false closeness of negative pairs over long horizons and prediction error under a twofold clock-speed perturbation. The predicted signature is a sharp empirical transition in false-closeness probability near delta_emp(epsilon), while same-orbit pairs remain close after alignment; the benefit should persist under irregular sampling and should not be explained only by increased latent norm.
Verification
Stage 1 · Toy mechanism gate: Passed ✓
Stage 2 · Mechanism transferred to benchmark: Confirmed ✓
Stage 2 · Practical benchmark result: Inconclusive
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 a deterministic MVP of Komuro-style time-warp discrepancy with an affine monotone warp search and hinge expansivity statistics. All three required mechanism predictions manifested: inverse-speed warp recovery had maximum error 2.41e-4, optimized discrepancy equaled the radial-gap lower bound across the sweep, and hinge activation matched the empirical q=0.1 quantile transition. The controlled comparison reduced clock-perturbed pair discrepancy by 99.90% (0.3126 to 0.000303), but this is a mathematical toy signal rather than evidence of a trained neural ODE win.
- Agent confidence
- 8/10
- Baseline
- Ordinary alpha=1 max discrepancy on clock-perturbed pairs: mean 0.312585
- Idea
- Warp-optimized max discrepancy: mean 0.000303; relative reduction 0.99903. Max inverse-speed alpha error 0.0002407; radius-gap D monotone and D=|dr|; q=0.1 delta_emp=0.357631 with observed activation 0.10.
Limitations: No learned neural ODE training, reconstruction or prediction task, irregular sampling, nonlinear warp optimization, arbitrary surjective homeomorphisms, FLOP/runtime comparison, or downstream generalization test was performed. The circular analytic flow makes the predictions unusually clean and does not establish benefit in realistic latent dynamics.
How to run: python3 komuro_mvp.py
Stage 2 — Benchmark vs. tuned baseline (arXiv:2608.02186)
No significant difference vs. the tuned baseline (+8.1%, p=0.0754, wins 2 / 8 paired seeds).
- Benchmark
- Dynamics forecasting
- Model
- rnn_small
- Paired seeds
- 8
- Baseline mean
- 0
- Idea mean
- 0
- Effect (Δ)
- 0 (+8.1%; negative = idea better)
- Wins
- 2 / 8 paired seeds
- p-value
- 0.0754 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±10.2%
- Mechanism
- Confirmed ✓
- Practical effect
- Inconclusive
- Baseline tuning
- swept over 3 configs
The built-in dynamics track provides flattened pendulum windows rather than explicit irregularly sampled or clock-speed-perturbed trajectory pairs, so the regularizer uses shuffled windows and restricted affine monotone slopes (0.75, 1.0, 1.3333333) as a proxy. No nonlinear warp family, neural ODE integration, long-horizon rollout metric, runtime/FLOP comparison, or larger-scale model was tested.
How to run: python3 bench_komuro.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
Built the Komuro time-warp expansivity regularizer as a training-loss modification for the structurally matched dynamics track using the shared rnn_small GRU. The mechanism signature was confirmed quantitatively: warped hidden-trajectory discrepancy fell 4.56% (1.6830 to 1.6063), but test MSE worsened from 2.2823e-5 to 2.4673e-5; the paired permutation p-value was 0.07535, so there was no significant win.
- Baseline
- Best lr=0.006; full 8-seed test MSE mean 2.282289460708853e-05, std 2.305330446600017e-06
- Idea
- Best lr=0.006, lambda=0.03, margin=0.25; full 8-seed test MSE mean 2.467335889377864e-05, std 3.053583916547981e-06
- Mechanism signature
- ordinary hidden D=1.6830359101295471, warped hidden D=1.6062670052051544, relative reduction=0.04561334934231065, confirmed=true
- Paired delta
- 1.8504642866901122e-06 (idea worse; 2/8 per-seed wins)
- Permutation p value
- 0.07535
Benchmark result
No significant difference vs. the tuned baseline (+8.1%, p=0.0754, wins 2 / 8 paired seeds).
- Benchmark
- Dynamics forecasting
- Model
- rnn_small
- Paired seeds
- 8
- Baseline mean
- 0
- Idea mean
- 0
- Effect (Δ)
- 0 (+8.1%; negative = idea better)
- Wins
- 2 / 8 paired seeds
- p-value
- 0.0754 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±10.2%
- Mechanism
- Confirmed ✓
- Practical effect
- Inconclusive
- Baseline tuning
- swept over 3 configs
The built-in dynamics track provides flattened pendulum windows rather than explicit irregularly sampled or clock-speed-perturbed trajectory pairs, so the regularizer uses shuffled windows and restricted affine monotone slopes (0.75, 1.0, 1.3333333) as a proxy. No nonlinear warp family, neural ODE integration, long-horizon rollout metric, runtime/FLOP comparison, or larger-scale model was tested.
How to run: python3 bench_komuro.py
Artifacts
- bench_komuro.py 4.8 KB View Raw
- bench_report.json 3.7 KB View Raw
- komuro_mvp.py 4.9 KB View Raw
- report.md 1.5 KB View
- report_bench_2026-08-31T201603.md 4.6 KB View
- results.json 3.5 KB View Raw