Residual-greedy latent basis expansion
Implementation & benchmark of arXiv:2609.02578 — Reduced order model for parametric Boltzmann equation and its application to inverse problems
Source paper: Reduced order model for parametric Boltzmann equation and its application to inverse problems arXiv:2609.02578 ⓘ · analyzed Sep 3, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Construct a compact latent basis for a parameter-conditioned neural operator by repeatedly adding the parameter sample with the largest normalized prediction residual. This replaces uniform parameter coverage with adaptive basis growth and supplies a stopping rule for when the latent approximation is sufficiently accurate.
Formulas
Mathematical statement
The paper constructs V_N=span{f(.;mu_i)} from parameterized solution snapshots and selects new snapshots by a residual-based greedy rule. Let U_N=[u_1,...,u_N] be orthonormal basis vectors, c_N(mu) reduced coordinates, and y_N(mu)=U_N c_N(mu) the reduced prediction. Let F_theta(y;mu) be a model or governing-equation residual, b(mu) a scale vector, and epsilon>0 a numerical safeguard. Define rho_N(mu)=||F_theta(y_N(mu);mu)||/max(||b(mu)||,epsilon). Choose mu_{N+1}=argmax_{mu in P_train}rho_N(mu), append its snapshot s(mu_{N+1}), and orthogonalize it with u_{N+1}=(I-U_NU_N^T)s/||(I-U_NU_N^T)s||. Stop when the maximum residual is at most tau. The transferable property is adaptive approximation of a parameterized solution manifold using residual information rather than random sampling alone.
Implementation notes
(1) Integration point: use this procedure for adaptive data selection and latent-basis construction in a parametric neural operator or differentiable simulator. A snapshot s(mu) may be an encoder embedding, decoder output, hidden rollout state, or full predicted field compressed by randomized SVD. Define F_theta as one-step prediction error on a target batch; if a governing equation is available, concatenate its discretized residual. Do not use training loss alone, because it may miss poorly represented parameter regions.
(2) Pseudocode:
P = candidate_parameter_samples
choose mu_1; S = [snapshot(mu_1)]; U = orthonormalize(S)
repeat:
for mu in P:
z = snapshot_or_encoder(mu)
c = U.T @ z
yN = U @ c
rho[mu] = norm(residual(theta, yN, mu)) / max(norm(scale(mu)), eps)
mu_new = argmax(rho)
if rho[mu_new] <= tau: break
s = snapshot(mu_new)
s = s - U @ (U.T @ s)
U = concatenate(U, s / (norm(s)+eps))
optionally fine-tune theta on selected high-rho samples
(3) Orthonormalization, residual scoring, greedy argmax selection, and the threshold stopping rule come from the reduced-basis idea. The snapshot type, residual normalization, candidate pool, and threshold tau must be selected empirically. Compute scores in batches and validate them on a disjoint parameter set. (4) First cheap experiment: use a small Fourier neural operator or MLP neural operator on a 1D parametric advection-diffusion family with two to four parameters. Compare random, Latin-hypercube, and residual-greedy sampling using the same number of expensive training solves. Report test error versus snapshot count, worst-case error over parameters, and cost to reach a target error. Success is a 2x reduction in required snapshots or lower worst-case error at equal data-generation cost; for inverse estimation also compare observable misfit J(mu) and recovered-parameter error.
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.