△ Mechanism confirmed, baseline not beaten 2026

Pick-to-Learn Scenario Compression for Safe NN Calibration

Implementation & benchmark of arXiv:2607.16084 — Pick-to-Learn Calibration of an MPC Policy for an Origin-to-Destination Flight Problem

Usefulness7/10
Difficulty5/10
Novelty7/10

Source paper: Pick-to-Learn Calibration of an MPC Policy for an Origin-to-Destination Flight Problem arXiv:2607.16084 ⓘ · analyzed Aug 30, 2026

AI-generated research hypothesis, automatically tested. Not peer-reviewed.

Idea description

Replace uniform tuning of neural-network hyperparameters with a Pick-to-Learn-style compression procedure that selects the few scenarios most informative for constraint satisfaction. A scenario can be a domain-randomization seed, adversarial perturbation, task instance, or rollout. Tune the network or optimizer on the selected compression set, then evaluate fresh scenarios using a finite-sample certificate for the probability of violating a prescribed robustness, safety, or stability constraint.

Formulas

$$C(D_N)\subseteq D_N,\qquad |D_N|=N=400,\qquad |C(D_N)|=k=2.$$
$$V(\theta)=\Pr_{\xi\sim P}\left[g(\theta,\xi)>0\right].$$
$$\Pr_{D_N}\left\{V\left(\theta(D_N)\right)\leq \varepsilon(N,k,\beta)\right\}\geq 1-\beta.$$
$$\varepsilon(400,2,10^{-5})=0.048,\qquad \beta=10^{-5}.$$

Mathematical statement

Let xi denote an iid uncertainty realization, such as an environment, perturbation, or task instance. Let D_N = {xi_1, ..., xi_N} be N observed scenarios, theta be neural-network and training-policy parameters, and g(theta, xi) <= 0 be the required scenario constraint. Pick-to-Learn constructs a data-dependent compression set C(D_N) contained in D_N with size k much smaller than N, and calibrates theta using the selected informative scenarios. The fresh-scenario violation probability is V(theta) = P_{xi approximately distributed as P}[g(theta, xi) > 0]. The scenario-compression guarantee has the form P_{D_N}{V(theta(D_N)) <= epsilon(N,k,beta)} >= 1 - beta. In the paper's reported case, N = 400, k = 2, beta = 10^{-5}, and epsilon = 0.048, so the certified probability of violating the constraint on a fresh scenario is at most 4.8 percent with confidence 1 - 10^{-5}. The useful mechanism is that a small set of active or informative scenarios can preserve an out-of-sample constraint guarantee, provided the scenarios are iid and the calibration procedure satisfies the assumptions of the theorem.

Implementation notes

First, place this method around an existing neural-network training and hyperparameter-calibration loop. Let theta include learning rate, weight decay, dropout, spectral-normalization strength, controller gains, or a small set of trainable policy parameters. Define g(theta, xi) as a measurable constraint such as validation loss minus a threshold, maximum rollout deviation, gradient-norm explosion, safety violation, or failure of a certified Lipschitz bound. Generate N iid scenarios and initialize an empty compression set. A practical greedy algorithm is: train or calibrate theta on the current set; score every unselected scenario by g(theta, xi); select the scenario with the largest positive violation or largest change in the active constraint set; add it to the compression set; recalibrate theta; repeat until k scenarios are selected or no scenario changes the active set. Finally retrain on the selected set and evaluate all N scenarios for diagnostics. The compression set and training constraints are computed from sampled data, while the fresh violation probability is estimated only with an independent test set. The exact epsilon(N,k,beta) should be taken from the full Pick-to-Learn theorem rather than replaced by a heuristic confidence interval. The first cheap experiment should use an MLP on MNIST-C with 400 corruption scenarios and a fixed corruption-loss threshold. Compare greedy selection of k = 2, 4, and 8 scenarios against random selection with the same retraining budget, then evaluate on 10,000 fresh corruptions. The predicted signature is that greedy compression reaches near-complete training-scenario coverage with far fewer scenarios, and for N = 400, k = 2, beta = 10^{-5}, the fresh violation rate should be near or below 4.8 percent when the theorem assumptions hold. Non-iid scenarios or uncontrolled adaptive tuning should invalidate or weaken this boundary.

Verification

Mechanism confirmed, baseline not beaten

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 readable Pick-to-Learn MVP with exact theorem-bound evaluation, greedy largest-violation compression, constrained calibration, and iid fresh-scenario diagnostics. The mechanism manifested: epsilon decreased roughly by half when N doubled, increased with weaker beta and larger k, and greedy k=2 achieved 0.774% fresh violation versus 54.162% for random k=2 in the toy problem. The paper’s displayed equation gives 3.53% for N=400,k=2,beta=1e-5, differing from its quoted 4.8%; this implementation reports the literal equation result rather than silently using 4.8%.

Agent confidence
8/10
Baseline
Random compression: fresh violation 54.162% (k=2), 36.594% (k=4), 29.870% (k=8); theorem epsilon at N=400,k=2,beta=1e-5 is 3.530%.
Idea
Greedy compression: fresh violation 0.774% (k=2), 0.394% (k=4), 0.394% (k=8); theorem epsilon sweep is 13.549%, 6.963%, 3.530%, 1.777% for N=100,200,400,800 and rises monotonically with k.

Limitations: The benchmark is a two-parameter affine convex surrogate, not an MLP on MNIST-C, MPC, or rollout calibration. Only one seed and one random-compression draw were used; no formal theorem claim is made for adaptive neural-network training, and the shifted-distribution diagnostic is not a certificate test.

How to run: python3 p2l_experiment.py

Stage 2 — Benchmark vs. tuned baseline (arXiv:2607.16084)

Significantly WORSE than the tuned baseline (+573.0%, p=0.0081, wins 0 / 8 paired seeds).

Benchmark
Tabular regression (Friedman)
Model
mlp_tiny
Paired seeds
8
Baseline mean
15.3127
Idea mean
103.0531
Effect (Δ)
87.7404 (+573.0%; negative = idea better)
Wins
0 / 8 paired seeds
p-value
0.0081 (permutation test, 20 000 shuffles)
Smallest detectable effect
±118.3%
Mechanism
Confirmed ✓
Practical effect
Hurts
Baseline tuning
swept over 3 configs
Limitations:

The registered tabular track was used because the idea concerns training-time calibration and optimizer/training dynamics. The implementation uses a practical greedy residual heuristic rather than claiming the exact theorem for adaptive neural-network training; k was fixed at 20, and vision corruption-specific experiments were not run.

How to run: python3 bench_p2l.py

Verdict computed by deterministic test code from paired-seed statistics — not by the language model.

Stage 2 — Benchmark latest Failed ✗

Agent confidence: 10/10

Implemented greedy Pick-to-Learn-style residual scenario compression for the registered tabular Friedman#1 MLP track. The core numerical proxy contracted monotonically with increasing N and the trained-model signature confirmed high-loss-tail selection, but compression to k=20 caused a large, statistically significant test-MSE regression. Therefore the idea did not win the standard benchmark metric.

Baseline
mean test MSE 15.31271517276764, std 0.822924635150327, best configuration lr=0.006 epochs=12
Comparison
idea minus baseline delta_mean +87.74035584926605; permutation p_value 0.0081; 0 of 8 paired wins
Idea
mean test MSE 103.05307102203369, std 20.493133534705294, best configuration lr=0.006 epochs=12

Benchmark result

Significantly WORSE than the tuned baseline (+573.0%, p=0.0081, wins 0 / 8 paired seeds).

Benchmark
Tabular regression (Friedman)
Model
mlp_tiny
Paired seeds
8
Baseline mean
15.3127
Idea mean
103.0531
Effect (Δ)
87.7404 (+573.0%; negative = idea better)
Wins
0 / 8 paired seeds
p-value
0.0081 (permutation test, 20 000 shuffles)
Smallest detectable effect
±118.3%
Mechanism
Confirmed ✓
Practical effect
Hurts
Baseline tuning
swept over 3 configs
Limitations:

The registered tabular track was used because the idea concerns training-time calibration and optimizer/training dynamics. The implementation uses a practical greedy residual heuristic rather than claiming the exact theorem for adaptive neural-network training; k was fixed at 20, and vision corruption-specific experiments were not run.

How to run: python3 bench_p2l.py

Artifacts

Implementation overview ⬇ Download all as ZIP 6 files · code, reports and structured results