Streaming TT-ALS adapter
Implementation & benchmark of arXiv:2606.31061 — Online TT-ALS for Streaming Tensor Decomposition with Incremental Orthogonalization
Source paper: Online TT-ALS for Streaming Tensor Decomposition with Incremental Orthogonalization arXiv:2606.31061 ⓘ · analyzed Aug 29, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Parameterize a frozen-feature neural projection as a tensor-train operator and update one TT core at a time by an exact least-squares solve on each incoming minibatch. After every core solve, QR-orthogonalize its matricization and absorb the triangular factor into the next core, preserving the represented operator while controlling conditioning. This creates a deterministic, low-memory alternative to Adam-trained LoRA for regression heads, MLP projections, or linear attention projections.
Formulas
Mathematical statement
The paper represents a streaming tensor as \(\mathcal{X}[t]=\mathcal{X}[t-1]\boxplus\overline{\mathcal{X}}_t\), where \(\overline{\mathcal{X}}_t\) is the new tensor slice. A TT operator consists of cores \(\mathcal{G}_k\), with left interface \(\mathbf{A}_k[t]\) and right interface \(\mathbf{B}_k[t-1]\); with all cores except \(\mathcal{G}_k\) fixed, the local objective is \(f(\mathcal{G}_k)=\tfrac12\|\widehat{\mathcal{X}}_t-\mathbf{A}_k[t]\times_k^1\mathcal{G}_k\times_{k+1}^1\mathbf{B}_k[t-1]\|_F^2\). Here \(\widehat{\mathcal{X}}_t\) is the current minibatch target tensor, \(\mathbf{A}_k[t]\) contracts all cores to the left of core \(k\), and \(\mathbf{B}_k[t-1]\) contracts all cores to its right. The updated core \(\mathcal{G}_k^\star\) satisfies \(\partial f/\partial\mathcal{G}_k=0\). For any perturbation \(\mathcal{G}_k=\mathcal{G}_k^\star+\Delta\mathcal{G}\), define \(\mathcal{R}=\widehat{\mathcal{X}}_t-\mathbf{A}_k[t]\times_k^1\mathcal{G}_k^\star\times_{k+1}^1\mathbf{B}_k[t-1]\) and \(\mathcal{D}=\mathbf{A}_k[t]\times_k^1\Delta\mathcal{G}\times_{k+1}^1\mathbf{B}_k[t-1]\). The paper obtains \(f(\mathcal{G}_k)=f(\mathcal{G}_k^\star)-\langle\mathcal{R},\mathcal{D}\rangle+\tfrac12\|\mathcal{D}\|_F^2\); stationarity makes the cross term zero, proving local objective decrease. For the first core, the exact update simplifies to \(\mathcal{G}_1[t]=\widehat{\mathcal{X}}_t\times\mathbf{B}_1[t-1]^\top\). Its matricization is factorized as \(\mathbf{G}_1[t]=\mathbf{Q}\mathbf{R}\), with \(\mathbf{Q}^\top\mathbf{Q}=\mathbf{I}\); reshape \(\mathbf{Q}\) into the core and absorb \(\mathbf{R}\) into the next core. The neural adaptation solves the same least-squares problem for a tensorized weight operator mapping feature tensor \(x\) to target \(y\).
Implementation notes
1. Integration point: replace a frozen-feature linear layer or MLP projection \(y=Wx\) by a TT-matrix operator. Factor input and output dimensions into mode sizes \(I_1\cdots I_n\) and \(J_1\cdots J_n\), and represent \(W\) with cores \(G_k\in\mathbb{R}^{r_{k-1}\times I_k\times J_k\times r_k}\), where \(r_0=r_n=1\). Keep the backbone frozen and train only the TT cores on minibatches \((x_b,y_b)\). 2. Pseudocode: initialize TT cores and cache right contractions \(B_k\); for each minibatch, contract all cores except \(G_k\) to form design operator \(Z_k\), solve \(\operatorname{vec}(G_k^\star)=\arg\min_g\|Z_kg-y\|_2^2+\lambda\|g\|_2^2\) using Cholesky or conjugate gradients; reshape \(G_k^\star\); matricize it; compute thin QR \((Q,R)=\operatorname{qr}(G_{k,(12)})\); replace the core by reshaped \(Q\), and multiply \(R\) into the next core along its left-rank axis. Sweep \(k=1,\ldots,n\), then optionally sweep backward. 3. The paper supplies the local least-squares structure, QR gauge transformation, and monotone local-objective argument. Estimate \(\lambda\), minibatch covariance, and rank from validation data; use a rolling buffer or exponential weighting if the stream is nonstationary. Monitor residual before and after each update and reject a solve if approximate contractions increase the residual. 4. First experiment: use CIFAR-10 or ImageNet-100 with a frozen ViT or MLP feature extractor and replace its classifier or one projection layer. Compare TT-ALS against equal-parameter LoRA trained by AdamW and full-layer SGD. Begin with rank 4--8 and 4--6 modes. Measure wall-clock time, peak memory, validation accuracy, and loss after equal numbers of examples. The expected signal is faster early loss descent and lower optimizer memory, with equal accuracy at less than half the update time.
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.