Trusted Polytopic Optimizer Steps
Implementation & benchmark of arXiv:2608.24019 — Trusted Polytopic Action Sets for Fast Planning in Underactuated Systems
Source paper: Trusted Polytopic Action Sets for Fast Planning in Underactuated Systems arXiv:2608.24019 ⓘ · analyzed Sep 1, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Represent a family of nearby neural-network parameter updates by a low-dimensional polytope around the current parameters, and retain only the convex inner region whose predicted nonlinear training dynamics remain close to actual dynamics. Optimize the training objective over this trusted family with a small quadratic program rather than testing many independent candidate steps. The method turns a scalar learning-rate choice into a reusable set of jointly safe update directions.
Formulas
Mathematical statement
Let \(\theta^*\) be current network parameters and \(\gamma\in\mathbb{R}^{n_\gamma}\) low-dimensional update coordinates. Define \(\theta(\gamma)=\theta^*+B\gamma\), where \(B\) contains gradient, momentum, curvature, or random directions. For optimizer or training state \(z\), use the affine prediction \(z_{\mathrm{lin}}(t;\gamma)=z^*(t)+R(t)\gamma\). Let \(\mathcal{A}=\{\gamma:G\gamma\le h\}\) be a polytope. If \(F_t\) is the actual nonlinear \(t\)-step training map, define violation \(v(\gamma)=\max_{1\le t\le H}\|F_t(\theta^*+B\gamma)-z^*(t)-R(t)\gamma\|_2\). The trusted set is an inner approximation \(\mathcal{A}_{\mathrm{trust}}\subseteq\{\gamma:v(\gamma)\le\varepsilon\}\). The useful property is bounded local nonlinear prediction error over a reusable convex region, not convexity of the neural loss.
Implementation notes
Integrate this before applying each Adam or SGD update. Choose \(n_\gamma=2\) to \(8\) directions: the preconditioned gradient, momentum, a diagonal-curvature direction, and normalized random directions. Set \(B=[b_1,\ldots,b_{n_\gamma}]\) and initialize a box \(\mathcal{A}=\{\gamma:|\gamma_i|\le r_i\}\). Compute the nominal \(H\)-step rollout from \(\theta^*\), retaining losses, optimizer states, and optional hidden activations. For every vertex and face midpoint, execute the actual rollout from \(\theta^*+B\gamma\) and compare it with the affine prediction using Jacobian-vector products; this estimates \(v(\gamma)\). Shrink violating facets, or iteratively add separating half-spaces and recompute a maximum-volume inscribed box or ellipsoid. Minimize the local quadratic model over the accepted polytope and apply that update. The paper-derived quantities are the affine action map, reachable-family representation, and trusted-inner-set construction; Jacobians, tolerance \(\varepsilon\), and rollout horizon \(H\) are estimated or selected empirically. First test a 2-layer MLP on MNIST or CIFAR-10 against SGD and Adam with equal backward-pass budgets. Sweep radius \(r\). The predicted signature is a critical radius \(r_c\): below it, maximum violation remains under \(\varepsilon\); above it, violation rises sharply and accepted-polytope volume collapses. The measured boundary should be reproducible across batches and the \(H\)-step loss prediction error should remain below \(\varepsilon\) inside the trusted set.
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 a two-coordinate trusted-polytopic optimizer MVP with nonlinear rollout prediction, box refinement using vertices and face midpoints, and quadratic-model candidate selection. The toy mechanism matched all three quantitative predictions: local violation slope 2.006 versus 2, trust radius versus tolerance slope 0.490 versus 0.5, and monotone increasing horizon error. In the tiny MLP test, the trusted method reached loss 0.1065 versus SGD 0.1266 after 40 updates, but required 4040 versus 40 gradient evaluations, so no speedup claim is supported.
- Agent confidence
- 8/10
- Baseline
- SGD: final/best loss 0.1266002150, 40 gradient evaluations
- Idea
- Trusted 2D polytope: final/best loss 0.1064823206, mean radius 0.09536, 4040 gradient evaluations; toy slopes 2.0056 and 0.4896 versus predictions 2.0 and 0.5
Limitations: The neural experiment is a tiny synthetic 2D binary classification task implemented in NumPy, not MNIST/CIFAR or a full Adam integration. The trust set is an approximate sampled box rather than a certified polytope, curvature is estimated directionally, and computational cost was not equalized beyond reporting gradient-evaluation counts; no multi-seed or large-scale benchmark was run.
How to run: python3 trusted_polytope.py
Stage 2 — Benchmark vs. tuned baseline (arXiv:2608.24019)
Significantly WORSE than the tuned baseline (+108.5%, p=0.0081, wins 0 / 8 paired seeds).
- Benchmark
- Tabular regression (Friedman)
- Model
- mlp_tiny
- Paired seeds
- 8
- Baseline mean
- 7.1906
- Idea mean
- 14.9916
- Effect (Δ)
- 7.801 (+108.5%; negative = idea better)
- Wins
- 0 / 8 paired seeds
- p-value
- 0.0081 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±13.8%
- Mechanism
- Confirmed ✓
- Practical effect
- Hurts
- Baseline tuning
- swept over 6 configs
Only the tabular track, mlp_tiny model, 10 epochs, 1200 training samples, and two optimizer directions were tested. The implementation uses one-step finite-difference probes rather than multi-step rollouts or exact JVPs, and no speed or FLOP advantage was demonstrated.
How to run: python3 stage2_bench.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 idea was implemented and evaluated on the registered tabular Friedman regression track with the shared mlp_tiny architecture and eight paired seeds. Tuned Adam achieved test MSE 7.1906, while the best trusted-polytopic configuration achieved 14.9916; paired delta was +7.8010 with permutation p=0.0081, so the idea was significantly worse. The trained-model mechanism signature nevertheless confirmed the predicted radius-to-nonlinearity relationship.
- Baseline
- Adam; best configuration lr=0.01, weight_decay=0.0001; mean test MSE 7.1905966401
- Idea
- Trusted polytopic optimizer; best configuration lr=0.01, radius=0.1; mean test MSE 14.9916086197
- Mechanism confirmed
- true
- Mechanism loglog slope
- 1.9431904882117401
- Paired delta
- 7.8010119795799255
- Permutation p value
- 0.0081
Benchmark result
Significantly WORSE than the tuned baseline (+108.5%, p=0.0081, wins 0 / 8 paired seeds).
- Benchmark
- Tabular regression (Friedman)
- Model
- mlp_tiny
- Paired seeds
- 8
- Baseline mean
- 7.1906
- Idea mean
- 14.9916
- Effect (Δ)
- 7.801 (+108.5%; negative = idea better)
- Wins
- 0 / 8 paired seeds
- p-value
- 0.0081 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±13.8%
- Mechanism
- Confirmed ✓
- Practical effect
- Hurts
- Baseline tuning
- swept over 6 configs
Only the tabular track, mlp_tiny model, 10 epochs, 1200 training samples, and two optimizer directions were tested. The implementation uses one-step finite-difference probes rather than multi-step rollouts or exact JVPs, and no speed or FLOP advantage was demonstrated.
How to run: python3 stage2_bench.py
Artifacts
- bench_report.json 8.8 KB View Raw
- report.md 1.5 KB View
- report_bench_2026-09-01T194315.md 4.2 KB View
- results.json 3.6 KB View Raw
- stage2_bench.py 7.1 KB View Raw
- trusted_polytope.py 5.9 KB View Raw