Seven-Factor Stochastic Transition Layer
Implementation & benchmark of arXiv:2608.08242 — Bang--bang representation of $3\times 3$ embeddable stochastic matrices
Source paper: Bang--bang representation of $3\times 3$ embeddable stochastic matrices arXiv:2608.08242 ⓘ · analyzed Sep 1, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Parameterize a learned 3-state transition operator as a product of at most seven elementary row-stochastic matrices rather than learning its nine entries independently. Each factor performs one convex pull-in of row i toward row j, so every intermediate and final matrix remains row-stochastic and the layer has a sparse, bounded-depth interpretation.
Formulas
Mathematical statement
The paper defines the elementary row-stochastic matrix K_{ij}(t)=I+t(E_{ij}-E_{ii}), where I is the 3x3 identity, E_{ij} has a one in position (i,j) and zeros elsewhere, i and j are distinct states, and 0<=t<1. Left multiplication replaces row i of a matrix P by (1-t)P_i+tP_j while leaving all other rows unchanged. Finite products form R_n={K_{i_nj_n}(s_n)...K_{i_1j_1}(s_1): i_l != j_l, 0<=s_l<1}. The paper proves that every embeddable 3x3 row-stochastic matrix can be represented by a product with n<=7, and that seven is sharp. The neural adaptation learns P_theta=K_{i_7j_7}(s_7)...K_{i_1j_1}(s_1), with s_l=sigmoid(alpha_l) or s_l=(1-epsilon)sigmoid(alpha_l) to ensure s_l<1. The neural module inherits nonnegative entries and unit row sums because each factor has those properties and matrix products preserve them.
Implementation notes
Integrate this as a transition operator in a small RNN, a neural Markov model, a graph message-passing layer with three node types, or an MoE router over exactly three experts. Use seven fixed factor slots. Each slot has an ordered pair (i_l,j_l) selected from the six possibilities and one trainable scalar alpha_l. For a first MVP, choose the ordered pairs by enumerating or randomly sampling candidate sequences, then optimize only the alpha values. Compute s_l=(1-epsilon)*sigmoid(alpha_l), initialize P=I, and for l=1,...,7 construct K=I, set K[i_l,i_l]=1-s_l and K[i_l,j_l]=s_l, then update P=K@P. Apply y=x@P to row distributions or use P as the recurrent transition. A differentiable variant can learn pair logits beta_l and use K_l=sum_{i != j} softmax(beta_l)_{ij}K_{ij}(s_l); this convex combination remains row-stochastic. The mathematics gives exact nonnegativity, unit row sums, and the bounded seven-factor representation; the implementation should estimate no theorem-specific quantities. Monitor row-sum error, minimum matrix entry, singular values, gradient norms, and long-horizon rollout stability. First test on a synthetic 3-state sequence-prediction task and a tiny GRU whose explicit 3-state transition is replaced by this layer. Compare with a directly learned row-softmax 3x3 matrix, a single-factor model, and a dense unconstrained matrix. Success means equal or better validation negative log-likelihood, zero stochasticity violations, and improved long-horizon calibration or stability at comparable parameter count. Also report whether seven factors actually improve over fewer factors, since the paper's bound is worst-case rather than an assertion that all matrices need seven.
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.