Layered Structural Reachability for Neural States
Implementation & benchmark of arXiv:2607.27852 — On the Strong Structural Controllability of Matrix-Weighted Networks
Source paper: On the Strong Structural Controllability of Matrix-Weighted Networks arXiv:2607.27852 ⓘ · analyzed Aug 31, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Treat the hidden-state Jacobian of an RNN, SSM, or graph neural network as a directed matrix-weighted network and decompose repeated block couplings into scalar interaction layers. Use layer-specific structural controllability to select input, skip, reset, or readout channels that can reach all hidden dimensions, and reject architectures with structurally unreachable states before training.
Formulas
Mathematical statement
For a block state transition A in R^{nd x nd}, partitioned into d x d blocks A_ij, represent each block in a basis of the matrix-weight span as A_ij = sum_k theta_ij^(k) M_k, where M_k are linearly independent d x d basis matrices and theta_ij^(k) are scalar coefficients. Each k defines a scalar directed layer with adjacency W^(k) and Laplacian L^(k) = D^(k) - W^(k). For an input matrix B, structural reachability is tested through the controllability matrix C(A,B) = [B, AB, ..., A^(nd-1)B]. The architecture is structurally controllable when the generic rank of this matrix is nd, meaning full rank for almost every nonzero choice of coefficients. Layer-specific shortest-path distances from input blocks provide a cheap proxy for delayed or weak access to hidden states.
Implementation notes
1. Integration point: apply this before training to an RNN or linear state-space layer with hidden dimension nd, grouped into n vector-valued nodes of width d. For a graph neural network, use graph nodes as the n blocks; for an ordinary RNN, use groups of channels or recurrent modules. Let A be the recurrent/state-transition Jacobian at initialization, or the explicit SSM transition matrix, and let B inject the external input, token embedding, residual input, or teacher-forcing signal. 2. Pseudocode: compute every block A_ij; stack their vectorizations and obtain an SVD basis M_1,...,M_r for the nonzero singular subspace; extract coefficients theta_ij^(k); build each scalar layer W^(k); run directed BFS from input blocks to obtain tau_k; form C(A,B) by repeated matrix-vector products; estimate numerical rank with threshold epsilon times the largest singular value; if rank is deficient, add input or skip edges or penalize small singular values. Use the differentiable surrogate above with s_i denoting singular values and gamma a target margin. 3. Computed from the paper's mechanism: block basis decomposition, scalar interaction layers, structural rank, and layer-specific distances. Estimated empirically: numerical rank, singular-value threshold, and the Jacobian A for nonlinear networks; evaluate A on multiple minibatches because reachability may vary with activation state. 4. First cheap experiment: compare a 2-layer GRU or 4-block linear SSM on Copy, Adding, and sequential MNIST tasks. Keep parameter count fixed; compare random input placement against placement selected by minimizing T_mix and maximizing sampled generic rank. Measure rank deficiency, smallest controllability singular value, gradient norm at horizon H, and task loss. The prediction is that models with sampled generic rank below nd will contain permanently insensitive hidden directions, whereas full-rank models will have nonzero input sensitivity. In linear systems, the measured controllability rank should match the generic-rank prediction in at least 95 percent of random coefficient trials, and the best architecture should approximately minimize the worst-layer distance T_mix.
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.