Sobolev-Calibrated Frozen Sigmoid Features
Implementation & benchmark of arXiv:2608.19797 — Optimal Sobolev Approximation by Deterministic and Random Shallow Sigmoidal Networks
Source paper: Optimal Sobolev Approximation by Deterministic and Random Shallow Sigmoidal Networks arXiv:2608.19797 ⓘ · analyzed Sep 1, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Replace a trainable shallow MLP hidden layer by a frozen bank of smooth sigmoid ridge functions and train only a linear output head. Choose the feature count and parameter sampling regime using the theorem's explicit dependence on input dimension d, target regularity k, evaluation norm m, and confidence delta. The construction is especially appropriate for smooth regression, scientific surrogate models, and PINNs, where derivatives of the network output are part of the loss.
Formulas
Mathematical statement
The central approximation theorem states that for a bounded domain \(\Omega\subset\mathbb{R}^d\), every target \(u\in H^k(\Omega)\), and every \(m\in\{0,\ldots,k\}\), a deterministic dictionary with \(M\) smooth sigmoidal ridge features contains an approximant \(v_M^{\rm det}\) satisfying \(\|u-v_M^{\rm det}\|_{H^m(\Omega)}\lesssim M^{-(k-m)/d}\|u\|_{H^k(\Omega)}\). Here \(H^r(\Omega)\) is the Sobolev space of functions with square-integrable weak derivatives through order \(r\), \(d\) is input dimension, \(k\) is target smoothness, and \(m\) is the error norm order. For independently sampled hidden parameters from a density \(q\) bounded below on the prescribed direction-offset parameter domain, with probability at least \(1-\delta\), the random dictionary satisfies \(\|u-v_M^{\rm ran}\|_{H^m(\Omega)}\lesssim [M/\log(M/\delta)]^{-(k-m)/d}\|u\|_{H^k(\Omega)}\). The ridge dictionary is \(v_M(x)=\sum_{j=1}^M a_j\,\sigma(w_j^\top x+b_j)\), where \(\sigma\) is a smooth sigmoid such as \(\tanh\), logistic sigmoid, or \(\operatorname{erf}\). The theorem assumes rapid decay of derivatives of \(\sigma\). The transferable property is dimension- and regularity-explicit approximation with only output coefficients optimized.
Implementation notes
(1) Integration point: use this as the first hidden layer of a shallow MLP, a frozen coordinate encoder for a PINN, or a frozen feature map before a small trainable linear head. Normalize every input coordinate to a bounded domain \(\Omega\), sample or deterministically construct \((w_j,b_j)\), compute \(\phi_j(x)=\sigma(w_j^\top x+b_j)\), and train only coefficients \(a_j\). For a PINN, differentiate \(\phi_j\) analytically or with autodiff and include PDE residuals in the head-training loss. (2) Pseudocode: x <- normalized inputs; for j=1..M: w[j],b[j] <- sample q(w,b); Phi[j] <- tanh(w[j]^T x+b[j]); yhat <- Phi @ a; loss <- mean((yhat-y)^2) + lambda * Sobolev_or_PDE_residual(Phi @ a); update a only. If targeting an \(H^m\) metric, add sampled derivative terms \(\sum_{|\alpha|\le m}\|D^\alpha yhat-D^\alpha y\|_2^2\). (3) Mathematics versus estimates: the exponent \((k-m)/d\) and random-feature logarithmic factor come from the paper. Estimate unknown target smoothness \(k\) by fitting validation-error slopes versus \(M\), or use known PDE regularity. The sampling density must be nonzero throughout the chosen direction-offset box; uniform sampling is the simplest valid choice. For a deterministic baseline, place directions on a spherical grid and offsets on a uniform grid, then compare against i.i.d. parameters. (4) First cheap experiment: fit smooth 2D functions such as \(u(x,y)=\sin(3x)\cos(2y)+e^{x+y}\) and a 2D Poisson solution using frozen tanh features, with \(M\in\{64,128,256,512,1024\}\). Compare against a trainable one-hidden-layer tanh MLP and ordinary random Fourier features at equal feature counts and optimizer steps. Plot \(L^2\), gradient, and Hessian errors against \(M\) on log-log axes. Success means a slope near \(-(k-m)/d\), stable performance across random seeds, and lower wall-clock optimization cost because only the output head is trained; for a PINN, additionally require lower PDE residual at equal training time.
Verification
Stage 1 · Toy mechanism gate: Failed ✗
Stage 2 · Mechanism transferred to benchmark: Not tested
Stage 2 · Practical benchmark result: Not run
Methodology: Toy-system gate first; the benchmark stage runs only after a pass. How verification works
Stage 1 — Mechanism check agent confidence 8/10
Built frozen tanh Sobolev features with analytic first derivatives, Sobolev-weighted linear least squares, random/grid parameter sampling, and numerical evaluation. The derivative identity was verified to about 6e-9 error, and 2D errors decreased with feature count, but observed slopes (-0.714 for values and -0.392 for gradients) did not adequately confirm the predicted d=2,k=2 rates (-1 and -0.5); random-versus-grid behavior was also non-monotonic rather than a clean logarithmic penalty. The mechanism shows a promising approximation signal, but the required quantitative predictions were not sufficiently confirmed, so this is not counted as a win.
- Agent confidence
- 8/10
- Baseline
- The earlier 1D trainable tanh baseline at M=64 had value RMSE 0.1195, gradient RMSE 0.8845, Hessian RMSE 10.25, and about 17.38 seconds per seed; this comparison is not treated as decisive because the 1D frozen fit became nearly exactly representable.
- Idea
- In the controlled 2D sweep, random frozen Sobolev features improved from value/gradient RMSE 0.757/3.689 at M=8 to 0.07396/0.9648 at M=192; fitted slopes were -0.714/-0.392 versus predicted -1/-0.5.
Limitations: No PDE/PINN experiment, CUDA implementation, higher-dimensional scaling study, or robust trainable 2D baseline was tested. The theorem provides upper bounds rather than guaranteed empirical slopes, and the grid construction used a truncated tensor parameter grid that is not an optimized deterministic dictionary.
How to run: python3 sobolev_2d_check.py
Verdict computed by deterministic test code from paired-seed statistics — not by the language model.
Artifacts
- report.md 1.8 KB View
- results.json 4.7 KB View Raw
- results_2d.json 5.3 KB View Raw
- sobolev_2d_check.py 3.4 KB View Raw
- sobolev_features.py 6.0 KB View Raw