✓✓ Beats tuned baseline 2026

Inverse-Gain Structured Privileged Distillation

Implementation & benchmark of arXiv:2608.10453 — From Privileged Control to Deployable Adaptation:Fusing Mechanism-Guided Task Reduction with Learned Behavior

Usefulness8/10
Difficulty5/10
Novelty7/10

Source paper: From Privileged Control to Deployable Adaptation:Fusing Mechanism-Guided Task Reduction with Learned Behavior arXiv:2608.10453 ⓘ · analyzed Sep 1, 2026

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

Idea description

Replace direct action imitation with a causal recurrent estimator of the inverse input gain. The neural network predicts the latent quantity needed by the expert controller, and a fixed algebraic wrapper converts that prediction into an action using the measured state difference and tracking error, thereby removing the additive disturbance exactly under the sampled timing model.

Formulas

$$\delta x_{1,k}:=\frac{x_{1,k}-x_{1,k-1}}{T_c}=a_kx_{2,k-1}+d_k$$
$$x_{2r,k}=\frac{k_1e_k-d_k}{a_k}=x_{2,k-1}+\frac{1}{a_k}\left(k_1e_k-\delta x_{1,k}\right),\qquad e_k=r-x_{1,k}$$
$$\hat q_k=f_\theta(h_k),\qquad \hat x_{2r,k}=x_{2,k-1}+\hat q_k\left(k_1e_k-\delta x_{1,k}\right)$$
$$\hat x_{2r,k}-x_{2r,k}=(\hat q_k-q_k)z_k,\qquad q_k=\frac{1}{a_k},\quad z_k=k_1e_k-\delta x_{1,k}$$

Mathematical statement

The plant is sampled as $x_{1,k}=x_{1,k-1}+T_c(a_kx_{2,k-1}+d_k)$ and $x_{2,k}=x_{2,k-1}+T_cu_{k-1}$, where $x_1$ is the regulated state, $x_2$ is the measured velocity-like state, $T_c$ is the sampling period, $a_k>0$ is an unknown multiplicative input gain, and $d_k$ is an unknown additive disturbance. Define the causal difference $\delta x_{1,k}=(x_{1,k}-x_{1,k-1})/T_c=a_kx_{2,k-1}+d_k$, the reference $r$, and tracking error $e_k=r-x_{1,k}$. The privileged expert uses $x_{2r,k}=(k_1e_k-d_k)/a_k$, with control gain $k_1>0$. Eliminating $d_k$ gives $x_{2r,k}=x_{2,k-1}+q_k(k_1e_k-\delta x_{1,k})$, where $q_k=1/a_k$. Thus the student only needs to estimate $q_k$ from causal history. Let $\hat q_k=f_\theta(h_k)$, where $h_k$ contains recent measured states, references, actions, and finite differences. The deployed action is $\hat x_{2r,k}=x_{2,k-1}+\hat q_k(k_1e_k-\delta x_{1,k})$. The crucial property is that disturbance dependence disappears from the wrapper; action error is $\hat x_{2r,k}-x_{2r,k}=(\hat q_k-q_k)(k_1e_k-\delta x_{1,k})$.

Implementation notes

1. Integration point: use this as a deployable policy head for a partially observed control network, or as a structured recurrent policy inside an RL or distillation system. The GRU or LSTM receives only deployment-visible quantities: a window of $(x_1,x_2,r)$, previous actions, and causally computed differences. It outputs a scalar inverse-gain estimate $\hat q_k$ or a bounded residual around a nominal inverse gain. The fixed wrapper computes $e_k$, $\delta x_{1,k}$, and $\hat x_{2r,k}$; an existing low-level actuator converts this reference into $u_k$. 2. Pseudocode: delta=(x1[k]-x1[k-1])/Tc; e=r-x1[k]; h=GRU(concat(x1,x2,r,u_prev)); qhat=q_min+(q_max-q_min)*sigmoid(head(h)); z=k1*e-delta; x2_ref=x2[k-1]+qhat*z; u=inner_controller(x2_ref,x2[k]). During privileged training, compute the expert action x2r=(k1*e-d)/a, but do not provide $a$ or $d$ to the student. Minimize Huber loss between x2_ref and x2r, plus a temporal smoothness penalty on $\hat q_k$, and optionally train on closed-loop rollouts rather than shuffled samples. 3. Computed exactly from the paper: the sampled difference, algebraic cancellation, and structured action wrapper. Estimated empirically: the recurrent map $f_\theta$, the admissible gain interval $[q_{min},q_{max}]$, sensor-noise filtering, and inner-loop dynamics. Use a causal moving average or one-pole filter for $\delta x_1$ only if the delay is acceptable. 4. First cheap experiment: simulate the two-state plant with $a_k$ randomly switching among $\{0.25,0.5,1,2\}$ and piecewise-constant disturbances, then hold out gain sequences and 60-second reference trajectories. Compare a direct MLP action imitator, a GRU action imitator, and the structured GRU against the privileged controller and a tuned nominal observer. Measure closed-loop tracking RMSE, action error, gain-estimation error, and failure rate. The quantitative prediction is that $|\hat x_{2r}-x_{2r}|$ scales linearly with $|z_k|$ and inverse-gain error according to $|\hat x_{2r}-x_{2r}|=|\hat q-q||z|$; binning by $|z|$ should produce slope $|\hat q-q|$. The structured policy should retain performance on unseen $(a,d)$ combinations, while direct action imitation should show larger closed-loop degradation despite similar offline action MSE.

Verification

Beats tuned baseline

Stage 1 · Toy mechanism gate: Passed ✓

Stage 2 · Mechanism transferred to benchmark: Confirmed ✓

Stage 2 · Practical benchmark result: Significant win ✓

Methodology: 8 paired seeds, permutation test (p<0.05) against a learning-rate-tuned baseline. How verification works

Stage 1 — Mechanism check agent confidence 9/10

Built the inverse-gain structured policy wrapper and a fixed-seed toy verification. Algebraic disturbance cancellation and the factorized action-error identity held to about 5e-14 across disturbance amplitudes, and the inverse-gain noise sweep matched the predicted linear scaling closely. The offline learning comparison did not show a practical win: structured GRU RMSE was 0.513 versus 0.501 for the MLP and 0.555 for the GRU, so the mechanism is validated but superiority is not established.

Agent confidence
9/10
Baseline
MLP action imitation: RMSE 0.5012, MAE 0.3039; GRU action imitation: RMSE 0.5549, MAE 0.3636
Idea
Structured GRU action imitation: RMSE 0.5134, MAE 0.3127; exact cancellation max error 4.97e-14; inverse-gain noise/action-MAE scaling matched prediction within about 1%

Limitations: The learning comparison is offline, teacher-forced action prediction rather than closed-loop control. Sensor noise, online adaptation, inner-loop actuator dynamics, stability/failure-rate analysis, FLOP accounting, and repeated random seeds were not tested. results.json contains the original learning run; verification_report.json contains the additional mechanism sweeps.

How to run: python3 experiment.py

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

Beats the tuned baseline by 59.0% (p=0.0081, wins 8 / 8 paired seeds; metric: lower is better).

Benchmark
Inverse_gain_control
Model
rnn_small
Paired seeds
8
Baseline mean
3.671
Idea mean
1.5069
Effect (Δ)
-2.1641 (−59.0%; negative = idea better)
Wins
8 / 8 paired seeds
p-value
0.0081 (permutation test, 20 000 shuffles)
Smallest detectable effect
±10.4%
Mechanism
Confirmed ✓
Practical effect
Helps
Baseline tuning
swept over 3 configs
Limitations:

The registered custom track is a small supervised sampled-control benchmark and does not test long-horizon closed-loop stability, sensor-noise robustness, actuator saturation, or hardware transfer. The mechanism signature uses privileged expert targets to infer observed inverse gain and therefore validates the neural-scale factorization rather than deployment-time gain estimation.

How to run: python3 bench_run.py

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

Stage 2 — Benchmark latest Worked ✓

Agent confidence: 8/10

The inverse-gain structured GRU significantly outperformed the matched direct-action GRU on the registered inverse_gain_control dynamics track. Test MSE improved from 3.6709986031 to 1.5069266558 on all 8 paired seeds, with permutation p=0.0081; the trained-model mechanism signature confirmed the predicted action-error factorization.

Baseline
Direct-action GRU, tuned lr=0.003: mean test MSE 3.670998603105545, std 0.63283622936479.
Idea
Inverse-gain structured GRU, tuned lr=0.003: mean test MSE 1.5069266557693481, std 0.23555139171685704.
Paired delta
-2.164071947336197; 8/8 idea wins; permutation p=0.0081.

Benchmark result

Beats the tuned baseline by 59.0% (p=0.0081, wins 8 / 8 paired seeds; metric: lower is better).

Benchmark
Inverse_gain_control
Model
rnn_small
Paired seeds
8
Baseline mean
3.671
Idea mean
1.5069
Effect (Δ)
-2.1641 (−59.0%; negative = idea better)
Wins
8 / 8 paired seeds
p-value
0.0081 (permutation test, 20 000 shuffles)
Smallest detectable effect
±10.4%
Mechanism
Confirmed ✓
Practical effect
Helps
Baseline tuning
swept over 3 configs
Limitations:

The registered custom track is a small supervised sampled-control benchmark and does not test long-horizon closed-loop stability, sensor-noise robustness, actuator saturation, or hardware transfer. The mechanism signature uses privileged expert targets to infer observed inverse gain and therefore validates the neural-scale factorization rather than deployment-time gain estimation.

How to run: python3 bench_run.py

Artifacts

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