✗ Failed on benchmark 2026

IMM Stale-Feedback Detector

Implementation & benchmark of arXiv:2608.18824 — Detecting Feedback-path Delay Injection Attacks Using Interacting Multiple Model Filtering

Usefulness7/10
Difficulty5/10
Novelty7/10

Source paper: Detecting Feedback-path Delay Injection Attacks Using Interacting Multiple Model Filtering arXiv:2608.18824 ⓘ · analyzed Sep 1, 2026

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

Idea description

Add an interacting-multiple-model monitor to a recurrent or distributed neural training loop, with one state estimator for each candidate feedback delay. The monitor detects when gradients, hidden-state feedback, or parameter acknowledgements become stale, allowing the system to reduce the learning rate, discard delayed updates, or switch to a safe synchronous mode before delayed feedback destabilizes training.

Formulas

$$z_{t+1}=A_tz_t+B_tu_t+w_t,\qquad y_t=H_{d_t}z_{t-d_t}+\nu_t,\qquad w_t\sim\mathcal N(0,Q),\ \nu_t\sim\mathcal N(0,R).$$
$$\mu_{j\mid i,t-1}=\frac{T_{ji}\pi_{t-1}^{j}}{\sum_{k=0}^{D}T_{ki}\pi_{t-1}^{k}},\qquad \bar z_{t-1}^{i}=\sum_{j=0}^{D}\mu_{j\mid i,t-1}\hat z_{t-1}^{j}.$$
$$r_t^i=y_t-H_i\hat z_{t\mid t-1}^{i},\quad S_t^i=H_iP_{t\mid t-1}^{i}H_i^\top+R,\quad \ell_t^i=\frac{\exp[-\tfrac12(r_t^i)^\top(S_t^i)^{-1}r_t^i]}{(2\pi)^{m/2}|S_t^i|^{1/2}}.$$
$$\pi_t^i=\frac{\ell_t^i\sum_jT_{ji}\pi_{t-1}^{j}}{\sum_k\ell_t^k\sum_jT_{jk}\pi_{t-1}^{j}},\qquad \textsc{alarm}(t)=\mathbf 1[\pi_t^0\leq\xi].$$

Mathematical statement

The paper models a known linear state process with delayed observations, $x_{t+1}=Ax_t+Bu_t+w_t$ and $y_t=Cx_{t-d_t}+\nu_{t-d_t}$, where $d_t$ is an unknown delay mode. For neural training, let $z_t$ be a low-dimensional monitored state such as loss, gradient norm, parameter norm, and validation residual; use a local linearization $z_{t+1}=A_tz_t+B_tu_t+w_t$ and candidate delays $d_i\in\{0,\ldots,D\}$. Each mode $i$ maintains $(\hat z_t^i,P_t^i)$ and predicts the observation with a delay-specific observation operator $H_i$. Given Markov mode transitions $T_{ji}=P(d_t=i\mid d_{t-1}=j)$, IMM mixing uses $\mu_{j\mid i,t-1}=T_{ji}\pi_{t-1}^j/\sum_kT_{ki}\pi_{t-1}^k$. The mode likelihood is the Gaussian innovation density $\ell_t^i=\mathcal N(r_t^i;0,S_t^i)$, where $r_t^i=y_t-H_i\hat z_{t\mid t-1}^i$ and $S_t^i=H_iP_{t\mid t-1}^iH_i^\top+R$. The posterior is $\pi_t^i=\ell_t^i\sum_jT_{ji}\pi_{t-1}^j/\sum_k\ell_t^k\sum_jT_{jk}\pi_{t-1}^j$. The alarm is raised when $\pi_t^0\leq\xi$, with mode zero denoting no delay. Under a persistent delay, the expected log posterior odds grow at approximately the per-step KL divergence between the true and no-delay innovation distributions.

Implementation notes

1. Integration point: begin with a small GRU or a distributed MLP trained by one asynchronous worker. Define $z_t\in\mathbb R^p$ as a monitored state containing current loss, exponentially smoothed loss, gradient norm, parameter norm, update norm, and worker-to-server residual. Keep the optimizer unchanged and run the IMM monitor after every optimizer step. Candidate modes are delays $d_i=0,1,\ldots,D$, with $D=4$ for the MVP. For a nonlinear network, estimate $A_t$ using a diagonal or low-rank local linear model fitted online from the last 100 monitor states, or use a constant autoregressive matrix learned during clean warmup. 2. Pseudocode: initialize $\pi^i=1/(D+1)$, $\hat z^i=z_0$, and $P^i=I$. At each step, compute $\mu_{j|i}$, mix means and covariances, propagate each mode with $\hat z^i\leftarrow A_t\bar z^i+B_tu_t$ and $P^i\leftarrow A_t\bar P^iA_t^\top+Q$, construct the delay-specific predicted observation using the corresponding ring-buffer entry $\hat z_{t-d_i}^i$, calculate $r_t^i,S_t^i,\ell_t^i$, normalize the posterior $\pi_t^i$, and update each EKF covariance. If $\pi_t^0\leq\xi$ for three consecutive steps, reduce the learning rate by 4x and reject updates older than the median detected delay. 3. Computed versus estimated: IMM equations and posterior normalization are exact; $A_t,Q,R$ and transition matrix $T$ are estimated from a clean calibration run. Set $R$ to the empirical covariance of monitor residuals and $Q$ to 0.01--0.1 times that covariance. Estimate $\xi$ from the 99.5th percentile of the minimum clean no-delay posterior over calibration windows. 4. First cheap experiment: train a two-layer MLP on MNIST with one asynchronous worker, then inject a fixed delay of $d=2$ steps into worker gradients at step 2,000. Compare SGD, SGD with gradient-age clipping, and SGD plus IMM. Record posterior trajectories, detection delay, false alarms, and loss spikes. The quantitative prediction is that the posterior for mode $d=2$ will dominate after the attack and that expected log odds $\log(\pi_t^2/\pi_t^0)$ will have an approximately linear slope equal to the empirical innovation KL divergence. Measure the clean alarm rate below 0.5%, and verify that a delay of two steps is detected within 20--50 updates when its innovation KL divergence is at least 0.1 nats per step. Sweep injected delay and learning rate; the learned stability boundary should shift to larger tolerable learning rates when delayed updates are rejected.

Verification

Failed on benchmark

Stage 1 · Toy mechanism gate: Passed ✓

Stage 2 · Mechanism transferred to benchmark: Not reproduced ✗

Stage 2 · Practical benchmark result: No measurable effect

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 an IMM stale-feedback detector for candidate delays 0..4 using exact Gaussian innovation likelihoods, Markov mode mixing, posterior updates, and alarms. The mechanism manifested: posterior normalization error was 4.4e-16, persistent delay-2 reached median posterior 0.96 versus 0.006 for no-delay, and cumulative innovation log-likelihood slope divided by empirical KL was 0.99-1.01 across noise sweeps. Detection time increased from 6 to 23 steps as KL decreased, but the practical control benchmark produced no alarm and no improvement over baseline, so this supports the detector math rather than a demonstrated neural-training win.

Agent confidence
8/10
Baseline
Delayed quadratic SGD final losses at learning rates 0.8/1.0/1.2: 4.66e-27, 0, and 1.24e-3; median maximum absolute state was 2.0; no IMM intervention occurred.
Idea
IMM toy: delay-2 final posterior 0.961, no-delay posterior 0.0057, and median alarm time 8.5 steps at noise 0.03; KL-slope ratios were 0.991-0.999 in the main sweep; median alarm times were 6, 8.5, 10.5, and 23 steps for noise 0.015, 0.03, 0.06, and 0.12.

Limitations: No MLP, GRU, asynchronous worker, gradient-age rejection, or GPU training experiment was completed. The successful verification assumes a known scalar linear process and known delayed state history; online estimation of A, Q, R and IMM performance under neural monitor noise were not tested. The control benchmark did not trigger an alarm, so learning-rate mitigation remains unvalidated.

How to run: python3 imm_experiment.py > results.json && /home/maxwelhelp/main/bin/python3 mini_control.py > control_results.json

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

No significant difference vs. the tuned baseline (−-0.0%, p=1, wins 0 / 8 paired seeds).

Benchmark
Tabular regression (Friedman)
Model
mlp_tiny
Paired seeds
8
Baseline mean
8.5679
Idea mean
8.5679
Effect (Δ)
0 (−-0.0%; negative = idea better)
Wins
0 / 8 paired seeds
p-value
1 (permutation test, 20 000 shuffles)
Smallest detectable effect
±0.0%
Mechanism
Not confirmed ✗
Practical effect
No effect
Baseline tuning
swept over 3 configs
Limitations:

Only the matched built-in tabular track was tested; vision, sequence, and dynamics were not tested. The canonical bench has no asynchronous acknowledgement or injected-delay protocol, so the main comparison had no actual stale feedback to detect. The monitor uses a simplified online diagonal observation model rather than fitted A_t/P_t delayed-state IMM filtering, and the explicit diagnostic was not used as the primary task-metric comparison.

How to run: python3 bench_imm.py && /home/maxwelhelp/main/bin/python3 nn_signature.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

Built an IMM stale-feedback monitor and integrated it into a custom Adam training loop for the matched tabular Friedman#1 plus mlp_tiny benchmark. Across the required 8 paired seeds, the best idea configuration exactly matched the tuned baseline (delta_mean=0, p=1.0), so there is no observed training-metric win. A separate trained-MLP two-step stale-gradient diagnostic also produced zero alarms, meaning the stage-1 detection effect did not transfer quantitatively to this neural setup.

Baseline
Friedman#1 tabular / mlp_tiny: tuned lr=0.01, 8-seed mean test MSE 8.567867696285248, std 0.9045233203943077; sweep means lr=0.001:25.60164165496826, lr=0.003:13.860755920410156, lr=0.01:8.352817177772522.
Diagnostic
On 8 trained MLP runs with explicit two-step stale-gradient application, clean alarm rate 0.0 and stale-delay-2 alarm rate 0.0; stale final no-delay posterior ranged 0.8641-0.9253, while clean final no-delay posterior ranged 0.9713-1.0000.
Idea
IMM monitor with persistent alarm-triggered 4x Adam learning-rate reduction: best lr=0.01, 8-seed mean test MSE 8.567867696285248, std 0.9045233203943077; paired delta 0.0, p=1.0, 0/8 wins, verdict no measurable effect.

Benchmark result

No significant difference vs. the tuned baseline (−-0.0%, p=1, wins 0 / 8 paired seeds).

Benchmark
Tabular regression (Friedman)
Model
mlp_tiny
Paired seeds
8
Baseline mean
8.5679
Idea mean
8.5679
Effect (Δ)
0 (−-0.0%; negative = idea better)
Wins
0 / 8 paired seeds
p-value
1 (permutation test, 20 000 shuffles)
Smallest detectable effect
±0.0%
Mechanism
Not confirmed ✗
Practical effect
No effect
Baseline tuning
swept over 3 configs
Limitations:

Only the matched built-in tabular track was tested; vision, sequence, and dynamics were not tested. The canonical bench has no asynchronous acknowledgement or injected-delay protocol, so the main comparison had no actual stale feedback to detect. The monitor uses a simplified online diagonal observation model rather than fitted A_t/P_t delayed-state IMM filtering, and the explicit diagnostic was not used as the primary task-metric comparison.

How to run: python3 bench_imm.py && /home/maxwelhelp/main/bin/python3 nn_signature.py

Artifacts

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