Response-Sufficient Neural Memory
Implementation & benchmark of arXiv:2608.20328 — The Memory Hidden in Response Fluctuations: Trajectory-Level Fluctuation-Response Theory and Inequalities for Non-Markovian Jump Dynamics
Source paper: The Memory Hidden in Response Fluctuations: Trajectory-Level Fluctuation-Response Theory and Inequalities for Non-Markovian Jump Dynamics arXiv:2608.20328 ⓘ · analyzed Sep 1, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Replace correlation-based memory pruning in an RNN or state-space model by measuring how hidden-state history changes the response to individual past input events. Train a compressed memory coordinate only if it preserves the event-consequence kernel for the target observable, such as future loss, prediction, or control return. A memory representation is accepted when the conditional variance of this kernel within compressed-state groups is small, even if dwell-time or autocorrelation statistics remain large.
Formulas
Mathematical statement
For an event record with counting increment \(\mathrm{d}n_a(t)\), where \(a\) indexes an input or internal event, let \(\gamma_a(t)\) be its history-conditioned intensity and define the martingale residual \(\mathrm{d}\varepsilon_a(t)=\mathrm{d}n_a(t)-\gamma_a(t)\mathrm{d}t\). For a future scalar observable \(Y_T\), define the event-consequence kernel \(K_a(t)=\mathbb{E}[Y_T\mid\mathcal{H}_{t-},\mathrm{d}n_a(t)=1]-\mathbb{E}[Y_T\mid\mathcal{H}_{t-},\mathrm{d}n_a(t)=0]\), where \(\mathcal{H}_{t-}\) is the realized history. The trajectory fluctuation expansion is \(Y_T-\mathbb{E}Y_T=\sum_a\int_0^T K_a(t)\,\mathrm{d}\varepsilon_a(t)\). If \(z_t=\phi(\mathcal{H}_t)\) is a compressed neural memory, define the response-heterogeneity gap \(G=\mathbb{E}[\sum_a\int_0^T\gamma_a(t)\operatorname{Var}(K_a(t)\mid z_t,a)\,\mathrm{d}t]\ge 0\). Response sufficiency means that \(G\) is close to zero.
Implementation notes
1. Integration point: apply this to an RNN, GRU, or selective state-space model processing event streams, token sequences, or irregular sensors. Let the hidden state be \(h_t\), and add a bottleneck memory \(z_t=\phi_\psi(h_t)\) with dimension much smaller than \(h_t\). Choose a future observable \(Y_T\), such as validation loss over the next horizon, a future logit, cumulative reward, or a forecast at lag \(\tau\). 2. Pseudocode: sample a trajectory and record event type \(a\), time \(t\), hidden state \(h_t\), and target \(Y_T\). Create a paired counterfactual by masking or delaying the event, or use automatic differentiation through the recurrent update, to estimate \(K_a(t)\approx Y_T(\text{event})-Y_T(\text{no event})\). Predict intensity \(\hat\gamma_a(t)\) from \(h_t\). Train \(\phi_\psi\) and an auxiliary predictor \(q_\omega(z_t,a)\) with weighted loss \(\hat\gamma_a(t)(K_a-q_\omega)^2\), while retaining the task loss. 3. The decomposition and nonnegativity are theoretical; \(K_a\), \(\gamma_a\), and \(G\) are estimated using interventions, Monte Carlo rollouts, or truncated backpropagation. Use cross-fitting for evaluation. 4. First experiment: train a GRU on synthetic delayed-copy event sequences containing one highly autocorrelated irrelevant variable and one weakly correlated but target-relevant variable. Compare full memory, autocorrelation-pruned memory, and response-sufficient memory. The prediction is that long-horizon error tracks \(\hat G\), not autocorrelation; the response method should achieve at least a fivefold lower kernel variance at equal bottleneck size.
Verification
Stage 1 · Toy mechanism gate: Passed ✓
Stage 2 · Mechanism transferred to benchmark: Not reproduced ✗
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 9/10
Built a response-sufficient memory bottleneck and an analytic Gaussian-Poisson event-stream verification. The martingale residual check had 0.06% relative variance error; the estimated response-heterogeneity gap followed the predicted lambda^2 scaling with 0.19% median relative error and the predicted (1-rho^2) scaling with 0.56% median relative error. At rho=0.35, the response-aware coordinate reduced held-out kernel MSE from 0.885 for the nuisance/correlation proxy to approximately 4e-22, demonstrating the intended mechanism. The future observable MSE itself did not improve because this toy includes irreducible Poisson noise and does not train a recurrent predictor.
- Agent confidence
- 9/10
- Baseline
- Nuisance/correlation proxy: kernel MSE 0.8853 at rho=0.35; estimated gap 6.1684. Future-observable MSE was 6.9250 in the direct response comparison (essentially equal to the baseline because irreducible martingale noise dominates).
- Idea
- Response-sufficient supervised bottleneck: kernel MSE approximately 4e-22 at rho=0.35; estimated gap 0.0368 versus the nuisance gap 6.1684. Lambda-sweep median relative error to G proportional to lambda^2: 0.00187; rho-sweep median relative error to G proportional to 1-rho^2: 0.00563.
Limitations: The implementation uses an analytically available event-consequence kernel rather than counterfactual rollouts or automatic differentiation. It is a scalar supervised bottleneck demonstration, not a trained GRU/SSM on delayed-copy sequences; no wall-clock, memory-size, or end-to-end long-horizon task win was established. The binned conditional-gap estimator has finite-bin bias, especially near rho=0.99.
How to run: python3 experiment.py && /home/maxwelhelp/main/bin/python3 - <<'PY'
import numpy as np
from experiment import draw
from response_memory import ResponseSufficientMemory
x,r,k,y,c=draw(50000,.35,1.0,777)
for h in (x,r):
m=ResponseSufficientMemory(1).fit(h[:,None],k)
print(m.kernel_mse(h[:,None],k))
PY
Stage 2 — Benchmark vs. tuned baseline (arXiv:2608.20328)
Significantly WORSE than the tuned baseline (+256.1%, p=0.0081, wins 0 / 8 paired seeds).
- Benchmark
- Sequence modeling
- Model
- transformer_tiny
- Paired seeds
- 8
- Baseline mean
- 0.3066
- Idea mean
- 1.0918
- Effect (Δ)
- 0.7852 (+256.1%; negative = idea better)
- Wins
- 0 / 8 paired seeds
- p-value
- 0.0081 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±21.0%
- Mechanism
- Not confirmed ✗
- Practical effect
- Hurts
- Baseline tuning
- swept over 3 configs
Only the built-in sequence forecasting track was tested, using a transformer rather than an RNN/SSM. The auxiliary response target was the future scalar observable at every position rather than explicit event-counterfactual rollouts or intensity-weighted martingale residuals; no long-horizon memory-compression, wall-clock, or parameter/memory savings study was performed. The runner uses a compact 1,200/400 train/test subset and 8 epochs for the required time budget.
How to run: python3 stage2_response_memory.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 a response-sufficient bottleneck for the structurally matched sequence forecasting track, with an auxiliary future-response prediction loss and trained-model intervention signature. The mathematical sanity check matched the predicted heterogeneity gap with 0.26% relative error, and the neural intervention response correlated strongly with the learned response prediction (0.957), confirming the mechanism signal. However, the idea significantly worsened test MSE: 1.09184 versus the tuned baseline's 0.30664, paired delta +0.78520 with permutation p=0.0081, so it does not give a benchmark win.
- Baseline
- sequence/transformer_tiny, lr=0.001 selected by sweep; 8-seed test MSE mean 0.3066449799, std 0.0326071854
- Idea
- response bottleneck, lr=0.001 and auxiliary weight alpha=0.2 selected from shared-lr sweep; 8-seed test MSE mean 1.0918433443, std 0.0919865673; paired delta +0.7851983644, p=0.0081, 0/8 wins
- Math sanity
- Predicted G=10.3808250, observed G=10.3542673, relative error=0.00255834
- Trained model signature
- Event intervention observed/predicted response correlation=0.9570617; observed std=0.00484657, predicted std=0.00489357; confirmed=true
Benchmark result
Significantly WORSE than the tuned baseline (+256.1%, p=0.0081, wins 0 / 8 paired seeds).
- Benchmark
- Sequence modeling
- Model
- transformer_tiny
- Paired seeds
- 8
- Baseline mean
- 0.3066
- Idea mean
- 1.0918
- Effect (Δ)
- 0.7852 (+256.1%; negative = idea better)
- Wins
- 0 / 8 paired seeds
- p-value
- 0.0081 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±21.0%
- Mechanism
- Not confirmed ✗
- Practical effect
- Hurts
- Baseline tuning
- swept over 3 configs
Only the built-in sequence forecasting track was tested, using a transformer rather than an RNN/SSM. The auxiliary response target was the future scalar observable at every position rather than explicit event-counterfactual rollouts or intensity-weighted martingale residuals; no long-horizon memory-compression, wall-clock, or parameter/memory savings study was performed. The runner uses a compact 1,200/400 train/test subset and 8 epochs for the required time budget.
How to run: python3 stage2_response_memory.py
Artifacts
- bench_report.json 3.7 KB View Raw
- experiment.py 4.7 KB View Raw
- report.md 2.2 KB View
- report_bench_2026-09-01T172800.md 6.1 KB View
- response_memory.py 1.9 KB View Raw
- results.json 3.9 KB View Raw
- stage2_response_memory.py 6.8 KB View Raw