Unverified 2026

LU-Preconditioned Orthogonal Weight Retraction

Implementation & benchmark of arXiv:2607.15561 — RCLUPPr: a new randomized CholeskyQR with LU preconditioning

Usefulness5/10
Difficulty6/10
Novelty5/10

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

$$[L,U,P]=\operatorname{LU}(X),\qquad [H,S]=\operatorname{HouseholderQR}(L),\qquad Y=SU,\qquad W=XY^{-1},$$ $$[Q,Z]=\operatorname{CholeskyQR2}(W),\qquad R=ZY.$$
$$A=\Omega X,\qquad [L,U,P]=\operatorname{LU}(A),\qquad [H,S]=\operatorname{HouseholderQR}(L),\qquad Y=SU,$$ $$W=XY^{-1},\qquad [Q,Z]=\operatorname{CholeskyQR2}(W),\qquad R=ZY.$$
$$|AB-\operatorname{fl}(AB)|\leq \gamma_n |A||B|,\qquad \gamma_n:=\frac{nu}{1-nu}\leq 1.02nu.$$
$$\sigma_n(A+B)\geq\sigma_n(A)-\|B\|_2.$$

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.