LU-Preconditioned Orthogonal Weight Retraction
Implementation & benchmark of arXiv:2607.15561 — RCLUPPr: a new randomized CholeskyQR with LU preconditioning
Source paper: RCLUPPr: a new randomized CholeskyQR with LU preconditioning arXiv:2607.15561 ⓘ · analyzed Aug 30, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Periodically project a rectangular neural-network weight matrix onto an approximately orthonormal-column matrix using LU-preconditioned CholeskyQR rather than ordinary QR or a polar iteration. Pivoted LU handles badly scaled and nearly dependent columns, while Householder orthogonalization of the LU factor produces a triangular preconditioner that makes the subsequent Cholesky step safer in fp16 or bfloat16.
Formulas
Mathematical statement
For a full-column-rank matrix X in R^{m x n}, m >= n, the direct RCLUPPr construction computes a pivoted LU factorization [L,U,P] = LU(X), then [H,S] = HouseholderQR(L), sets Y = S U, forms W = X Y^{-1}, applies [Q,Z] = CholeskyQR2(W), and returns R = ZY. Here P is a row-permutation matrix, L is lower trapezoidal, U and S are triangular, W is preconditioned, Q has approximately orthonormal columns, and X is approximately QR. The preconditioner reduces scale and conditioning problems before forming a Gram matrix. The finite-precision multiplication bound is |AB-fl(AB)| <= gamma_n |A||B|, with gamma_n = n u/(1-nu), where u is unit roundoff. Weyl's inequality, sigma_n(A+B) >= sigma_n(A)-||B||_2, states that a perturbation preserves full column rank when its norm is smaller than the smallest singular value.
Implementation notes
Integrate this as a periodic retraction for a trainable rectangular matrix W in a linear layer, LoRA adapter, orthogonal bottleneck, or projection head, rather than executing it on every minibatch update. Use W in R^{m x n} with m >= n; for a wide layer apply the procedure to W^T. At each retraction, cast W to fp32 for pivoted LU and HouseholderQR, while testing the CholeskyQR stage in the target precision. Pseudocode: read W; compute [L,U,P] = LU(W) with partial pivoting; compute [H,S] = HouseholderQR(L); set Y = S@U; compute Wpre = triangular_solve(Y, W^T)^T instead of explicitly forming Y^{-1}; calculate G = Wpre^T@Wpre; compute C = chol(G); set Q1 = Wpre@C^{-T}; calculate G2 = Q1^T@Q1 and C2 = chol(G2); set Q = Q1@C2^{-T}; replace W by Q. If Cholesky fails, retry with G + epsilon I and record the failure. Initially stop gradients through the projection and re-estimate all factors after each retraction. Measure ||W^T W-I||_F/sqrt(n), reconstruction error, minimum singular value, Cholesky failures, gradient spikes, training loss, and validation accuracy. Start with a 2-layer MLP or Transformer projection of size 1024 x 1024, initialize singular values across eight orders of magnitude, and train on CIFAR-10 or a small language-model benchmark. Compare unconstrained training, standard QR, polar/Newton-Schulz, and this method every 100-1000 steps in bfloat16. Success is lower orthogonality error and zero numerical breakdowns at equal validation accuracy and similar wall-clock cost.
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.