Markov-Increment Window Encoder
Implementation & benchmark of arXiv:2607.25949 — Block-hierarchical covariance decompositions for finite-block additive functionals
Source paper: Block-hierarchical covariance decompositions for finite-block additive functionals arXiv:2607.25949 ⓘ · analyzed Aug 31, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Replace a collection of overlapping sliding-window features with approximately orthogonal incremental features: the length-m feature contains information not predictable from shorter consecutive windows. Use the paper's transition-operator Toeplitz precision matrix to decorrelate the resulting sequence of window features before attention, suppressing duplicated local evidence and improving conditioning.
Formulas
Mathematical statement
For a stationary Markov chain X_t with transition operator K on one-coordinate square-integrable functions, the lifted k-window process is Y_t^(k)=(X_t,...,X_{t+k-1}). The paper defines conditional-expectation projections L_m psi=E(psi|F_L) and R_m psi=E(psi|F_R), where F_L=sigma(X_1,...,X_{m-1}) and F_R=sigma(X_2,...,X_m); the shared center field is F_C=sigma(X_2,...,X_{m-1}). The incremental subspace at window length m is the orthogonal complement of functions measurable with respect to shorter consecutive blocks. In implementation, this is approximated by residualizing a length-m feature against predictors from lengths 1,...,m-1. For reversible chains, correlations between one-coordinate features at offsets i and j are K^{|i-j|}, so the block covariance operator is T_k(K)=[K^{|i-j|}]_{i,j=1}^k. If I-K^2 is invertible, the inverse is tridiagonal: (I-K^2)^(-1) times a matrix with endpoint diagonals I, interior diagonals I+K^2, and neighboring off-diagonals -K. This explicit local precision filter is the transferable computational structure.
Implementation notes
Integrate the method before a transformer attention block, operating on local window embeddings rather than raw tokens. Choose window lengths m=1,...,k, with k=4 or 8. Let phi_m(t) be a d-dimensional embedding of tokens X_t,...,X_{t+m-1}, produced by a shared-weight MLP, depthwise one-dimensional convolution, or small local attention module. For each m>1, train predictors g_{m,r} from shorter windows X_{t:t+r-1} to phi_m(t), for every r<m, and form the residual e_m(t)=phi_m(t)-sum_{r<m} alpha_{m,r}g_{m,r}(X_{t:t+r-1}); implement the sum using concatenation followed by a linear projection. This approximates projection onto the incremental component. Stack e_1(t),...,e_k(t) into z(t). Estimate a one-step feature transition operator K by ridge regression, K=argmin_A sum_t ||u_{t+1}-Au_t||^2+lambda||A||_F^2, where u_t is a low-dimensional projected token feature. Apply the displayed tridiagonal precision transform to the length-indexed feature stack: q=(I-K^2)^(-1)Bz, where B has endpoint diagonal I, interior diagonal I+K^2, and adjacent off-diagonal -K. To avoid an expensive matrix square root, begin with this precision-style preconditioner rather than explicitly computing T_k(K)^(-1/2). Maintain K and the shorter-window predictors with exponential moving averages and stop gradients through them during the first experiment. The paper determines the Toeplitz structure and tridiagonal coefficients; K, predictor parameters, ridge coefficient, and residual dimensions are estimated empirically. Test first on WikiText-103 or a synthetic reversible Markov token generator with a 2-layer, 256-dimensional transformer. Compare against standard overlapping convolutional/window embeddings at equal parameter count and FLOPs. Measure validation perplexity, loss descent speed, gradient covariance condition number, attention entropy, and feature cross-covariance. A successful result is lower perplexity or faster loss descent at equal compute, at least a 20% reduction in cross-window feature covariance, and no increase in activation memory.
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.