Unverified 2026

AT-stable stochastic binary layer

Implementation & benchmark of arXiv:2608.23413 — A quantitative replica-symmetric bound of Sherrington--Kirkpatrick model in the entire de Almeida--Thouless region

Usefulness5/10
Difficulty5/10
Novelty7/10

Source paper: A quantitative replica-symmetric bound of Sherrington--Kirkpatrick model in the entire de Almeida--Thouless region arXiv:2608.23413 ⓘ · analyzed Aug 29, 2026

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

Idea description

Add a mean-field stochastic binary recurrent layer with an explicit susceptibility controller. The layer estimates the response statistic \(\chi=\beta^2N^{-1}\sum_i\operatorname{sech}^4(u_i)\) and either penalizes or clips it below \(1-\delta\), preventing the high-gain regime in which replicas with identical weights develop strongly divergent states. The expected benefit is more stable long-horizon recurrence and lower variance across stochastic forward passes.

Formulas

$$q=\mathbb{E}\left[\tanh^2\left(h+\beta\sqrt{q}\,Z\right)\right],\qquad Z\sim N(0,1).$$
$$\chi_{\mathrm{AT}}=\beta^2\mathbb{E}\left[\operatorname{sech}^4\left(h+\beta\sqrt{q}\,Z\right)\right]<1.$$
$$\mathbb{E}\left\langle(R_{12}-q)^2\right\rangle=O(N^{-1}),\qquad R_{12}=\frac{1}{N}\sum_{i=1}^N\sigma_i^{(1)}\sigma_i^{(2)}.$$
$$\mathcal{L}_{\mathrm{AT}}=\mathcal{L}_{\mathrm{task}}+\lambda\left[\max\left(0,\hat\chi-(1-\delta)\right)\right]^2,\qquad \hat\chi=\beta^2\frac{1}{N}\sum_{i=1}^N\operatorname{sech}^4(u_i).$$

Mathematical statement

The paper defines the replica-symmetric scalar order parameter \(q\) by \(q=\mathbb{E}[\tanh^2(h+\beta\sqrt{q}Z)]\), where \(h>0\) is an external field, \(\beta>0\) is interaction strength, and \(Z\sim N(0,1)\). Its strict de Almeida–Thouless condition is \(\chi_{\mathrm{AT}}=\beta^2\mathbb{E}[\operatorname{sech}^4(h+\beta\sqrt{q}Z)]<1\). For the SK model, the overlap of two Gibbs replicas \(R_{12}=N^{-1}\sum_i\sigma_i^{(1)}\sigma_i^{(2)}\), with \(\sigma_i^{(a)}\in\{-1,1\}\), satisfies \(\mathbb{E}\langle(R_{12}-q)^2\rangle=O(N^{-1})\) throughout compact subsets with a strict margin below one. The neural adaptation uses local fields \(u_i\), interaction gain \(\beta\), empirical order parameter \(\hat q=N^{-1}\sum_i m_i^2\), and empirical susceptibility \(\hat\chi=\beta^2N^{-1}\sum_i\operatorname{sech}^4(u_i)\), where \(m_i=\tanh(u_i)\). The paper's concentration guarantee is used as a stability hypothesis, not asserted to hold automatically for trained networks.

Implementation notes

Integrate this into a stochastic binary recurrent or state-space layer, not into the optimizer. Let the hidden state have width N and let x_t be the external input. Parameterize an interaction matrix G with entries scaled as G_ij/sqrt(2N), a positive gain beta, and input bias b_t=W_x x_t+b_0. At each time step compute u_t=b_t+beta G s_{t-1}/sqrt(2N), means m_t=tanh(u_t), and stochastic binary states s_t in {-1,+1}^N using P(s_{t,i}=1)=(1+m_{t,i})/2. During training, compute q_hat_t=N^{-1} sum_i m_{t,i}^2 and chi_hat_t=beta^2 N^{-1} sum_i sech^4(u_{t,i}); add the displayed AT penalty averaged over time. Alternatively, use a hard controller after every optimizer step: beta <- beta exp[-eta_beta max(0, chi_hat-(1-delta))], with stop-gradient through chi_hat. For a replica diagnostic, run two stochastic copies with the same G and input sequence, calculate R_t=N^{-1} sum_i s_{t,i}^{(1)}s_{t,i}^{(2)}, and track its variance. The paper supplies the scalar response criterion and predicts overlap concentration only under SK-like mean-field assumptions, so estimate all neural quantities empirically. First test a width-256 tanh or binary vanilla RNN on sequential MNIST or the adding task, comparing unconstrained gain, spectral normalization, and AT regularization at equal parameter count. Use eight stochastic replicas and measure gradient-norm variance, hidden-state Lyapunov growth, replica-overlap variance, task loss, and performance at sequence lengths two and four times the training length. Success means lower long-horizon loss variance and bounded overlap variance without sacrificing short-horizon accuracy; a useful target is at least a twofold reduction in replica-overlap or gradient variance at equal validation loss.

Verification

This idea has not been verified yet.

Verification happens in two stages: Stage 1 — a mechanism check on a toy system confirms the claimed mathematical phenomenon reproduces; Stage 2 — a benchmark implements the idea on a real (small) neural network task and compares it against a tuned baseline over 8 paired seeds with a permutation test.

Artifacts

Artifacts unavailable.