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
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
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
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
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
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
- bench_report.json 2.7 KB View Raw
- bench_run.py 4.1 KB View Raw
- custom_inverse_gain_track.py 1.5 KB View Raw
- experiment.py 4.5 KB View Raw
- report.md 1.5 KB View
- report_bench_2026-09-01T102825.md 4.4 KB View
- results.json 0.8 KB View Raw
- verification_report.json 1.4 KB View Raw