Phase-Polytope Robust Neural Dynamics
Implementation & benchmark of arXiv:2607.09194 — Cyclic Reformulation-Based Identification and Polytopic Uncertainty Modeling for Multirate Systems
Source paper: Cyclic Reformulation-Based Identification and Polytopic Uncertainty Modeling for Multirate Systems arXiv:2607.09194 ⓘ · analyzed Aug 30, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Use the M phase-aligned parameterizations produced by cyclic reformulation as an empirical ensemble of neural dynamics rather than selecting one phase or averaging only predictions. Their centroid supplies a nominal model, while their convex hull defines a low-dimensional uncertainty set used for robust rollout training and uncertainty-aware inference.
Formulas
Mathematical statement
Let theta^(m) be the parameter or function representation obtained from cyclic alignment m in {1,...,M}. The nominal centroid is theta_bar=(1/M) sum_m theta^(m), and the empirical polytope is P={theta(alpha): theta(alpha)=sum_m alpha_m theta^(m), alpha_m>=0, sum_m alpha_m=1}. For a loss ell, robust training can maximize over vertices or sampled convex combinations: L_rob=max_{alpha in Delta_M} ell(theta(alpha)), where Delta_M is the probability simplex. Since neural parameter coordinates are not invariant to neuron permutations or rescaling, the safer construction mixes function outputs or predicted trajectories: F_alpha(U)=sum_m alpha_m F_m(U). If phase errors are independent with variance sigma^2, the centroid has variance sigma^2/M.
Implementation notes
1. Integration point: maintain M lightweight phase heads or M copies of a small state-space transition. Heads may share hidden features. Each head is trained on data aligned to one cyclic offset. During training, compute all heads on the same minibatch and form both the centroid prediction and an adversarial convex mixture. 2. Pseudocode: compute predictions F[1:M]; set F_bar=mean(F); compute vertex losses l[m]=loss(F[m],Y); choose m_star=argmax_m l[m], or perform 3-5 projected-gradient ascent steps on alpha over the simplex; set F_mix=sum_m alpha[m]*F[m]; backpropagate loss(F_bar,Y)+lambda_rob*loss(F_mix,Y). At inference, report F_bar and the empirical spread across vertices. 3. Computed versus estimated: the simplex, centroid, and vertex losses are exact; the variance-reduction law is a hypothesis requiring empirical testing. Avoid averaging raw weights unless heads are explicitly aligned; use output-space mixing otherwise. 4. First cheap experiment: use a synthetic M=4 multirate nonlinear oscillator with additive observation noise and compare a single phase head, centroid prediction, ordinary deep ensemble, and centroid-plus-worst-vertex training using the same GRU or neural state-space backbone. Measure rollout RMSE, calibration, and maximum per-phase validation loss. The predicted signature is centroid error variance decreasing approximately as 1/M when phase errors are weakly correlated. Robust training should reduce the maximum vertex loss, and the worst-case loss should approach the average loss as phase discrepancies shrink. Failure of the 1/M trend or persistent large vertex spread falsifies the ensemble interpretation.
Verification
Stage 1 · Toy mechanism gate: Passed ✓
Stage 2 · Mechanism transferred to benchmark: Not reproduced ✗
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 small output-space phase-polytope experiment with four shared-feature heads, centroid prediction, exact worst-vertex robust training, and an independent-error variance sanity check. The variance check matched Var(mean)=sigma^2/M closely, and robust training reduced worst-vertex RMSE substantially at a modest centroid-RMSE cost. This is a promising but narrow signal rather than proof of improved dynamics modeling.
- Agent confidence
- 8/10
- Baseline
- Centroid-only training: centroid RMSE 0.02325, max vertex RMSE 0.36145, mean vertex RMSE 0.24270, prediction spread 0.05713.
- Idea
- Centroid-plus-worst-vertex training: centroid RMSE 0.02942, max vertex RMSE 0.03301, mean vertex RMSE 0.03196, prediction spread 0.000665; independent variance ratios for M=1,2,4,8,16 were 0.995, 1.003, 0.999, 0.998, 1.000.
Limitations: Only one-step prediction on a synthetic oscillator was tested; no multi-step rollout stability, calibration metric, FLOP/speed comparison, cyclic alignment procedure, sampled interior convex combinations, or independent deep-ensemble baseline was evaluated. The phase offsets were synthetic and fixed, and the single-head control has a different training objective, so the comparison is not a definitive end-to-end benchmark.
How to run: python3 phase_polytope_experiment.py
Stage 2 — Benchmark vs. tuned baseline (arXiv:2607.09194)
Ahead of the baseline (−2.6%) but not statistically significant (p=0.5281, wins 4 / 8 paired seeds).
- Benchmark
- Dynamics forecasting
- Model
- rnn_small
- Paired seeds
- 8
- Baseline mean
- 0
- Idea mean
- 0
- Effect (Δ)
- -0 (−2.6%; negative = idea better)
- Wins
- 4 / 8 paired seeds
- p-value
- 0.5281 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±9.1%
- Mechanism
- Not confirmed ✗
- Practical effect
- Inconclusive
- Baseline tuning
- swept over 3 configs
Only the registered built-in dynamics track was tested. The benchmark did not test an independent deep-ensemble baseline, calibration, or longer rollout horizons beyond the built-in target.
How to run: python3 bench_phase_polytope.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 output-space four-head phase-polytope robust system was evaluated against the centroid-only baseline using the same GRU backbone on the registered dynamics track. It had a small lower mean MSE, but the paired permutation test was non-significant (p=0.5281), so the system did not demonstrate a benchmark win; the trained-model 1/M variance signature was also not confirmed.
- Baseline
- Best baseline configuration lr=0.006, weight_decay=0.0; full 8-seed mean MSE=3.7997122717570164, std=8.625243996335504e-06.
- Idea
- Best idea configuration lr=0.006, weight_decay=0.0, lambda_rob=0.5; full 8-seed mean MSE=3.7003671650381875e-05, std=6.8240779328978215e-06; paired delta=-9.93451067188289e-07; p=0.5281.
Benchmark result
Ahead of the baseline (−2.6%) but not statistically significant (p=0.5281, wins 4 / 8 paired seeds).
- Benchmark
- Dynamics forecasting
- Model
- rnn_small
- Paired seeds
- 8
- Baseline mean
- 0
- Idea mean
- 0
- Effect (Δ)
- -0 (−2.6%; negative = idea better)
- Wins
- 4 / 8 paired seeds
- p-value
- 0.5281 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±9.1%
- Mechanism
- Not confirmed ✗
- Practical effect
- Inconclusive
- Baseline tuning
- swept over 3 configs
Only the registered built-in dynamics track was tested. The benchmark did not test an independent deep-ensemble baseline, calibration, or longer rollout horizons beyond the built-in target.
How to run: python3 bench_phase_polytope.py
Artifacts
- bench_phase_polytope.py 6.6 KB View Raw
- bench_report.json 8.5 KB View Raw
- phase_polytope_experiment.py 4.7 KB View Raw
- phase_polytope_results.json 1.5 KB View Raw
- report.md 1.5 KB View
- report_bench_2026-09-04T205435.md 4.4 KB View