Bundle-Glued Neural Field
Implementation & benchmark of arXiv:2608.02108 — Finite element discretization of Yang--Mills connections
Source paper: Finite element discretization of Yang--Mills connections arXiv:2608.02108 ⓘ · analyzed Aug 31, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Represent a field on a manifold with one neural network per chart, while enforcing the exact transition law between chart outputs on overlaps. This avoids the artificial requirement that one coordinate frame work globally and should improve learning on spherical, periodic, or otherwise topologically nontrivial domains. Use an augmented Lagrangian rather than only a pointwise penalty so chart compatibility is enforced strongly without requiring identical local parameterizations.
Formulas
Mathematical statement
The paper defines local sections \(\sigma_i\) and transition functions \(g_{ij}:U_i\cap U_j\to G\) by \(\sigma_j(x)=\sigma_i(x)\cdot g_{ij}(x)\), where \(U_i,U_j\) are overlapping charts, \(G\) is the structure group, and \(x\) is an overlap point. A connection represented in two charts obeys the affine gluing equation \(\omega_j-\Ad_{g_{ij}^{-1}}\omega_i=g_{ij}^{\ast}\theta\), where \(\omega_i,\omega_j\) are local Lie-algebra-valued 1-forms, \(\Ad_{g_{ij}^{-1}}\) is the adjoint action of the inverse transition, and \(g_{ij}^{\ast}\theta\) is the pullback of the Maurer--Cartan form \(\theta\), producing the inhomogeneous gauge term. The reverse-orientation equation is \(\omega_i-\Ad_{g_{ji}^{-1}}\omega_j=g_{ji}^{\ast}\theta\). The paper treats local fields as independent broken finite-element fields and imposes this jump condition weakly on facets using Lagrange multipliers. For a general neural feature representation \(\rho:G\to GL(d)\), replace \(\Ad\) by \(\rho\), let \(y_i(x),y_j(x)\in\mathbb{R}^d\) be local network outputs, and let \(b_{ij}(x)\) be a known transition-dependent offset. Define \(r_{ij}(x)=y_j(x)-\rho(g_{ij}(x)^{-1})y_i(x)-b_{ij}(x)\). Train with the augmented Lagrangian \(\mathcal{L}_{\mathrm{aug}}=\mathcal{L}_{\mathrm{task}}+\sum_{(i,j)}\int_{U_i\cap U_j}\langle\lambda_{ij}(x),r_{ij}(x)\rangle+(\beta/2)\|r_{ij}(x)\|^2\,d\mu(x)\), where \(\lambda_{ij}\) is a dual multiplier, \(\beta>0\) is a penalty coefficient, and \(d\mu\) is the overlap measure. For connection 1-forms, evaluate the residual componentwise in a local tangent basis or output coefficients relative to a local coframe.
Implementation notes
(1) Exact integration point. Choose a manifold with an explicit overlapping atlas, such as the sphere covered by two stereographic charts, a torus, or a periodic domain. Instantiate either one shared MLP with chart embeddings or separate local MLP heads \(f_i\); each head receives chart coordinates \(u_i(x)\) and outputs \(y_i(x)\in\mathbb{R}^d\). Insert the compatibility module immediately after the local heads and evaluate it at sampled points in chart overlaps. For a scalar trivial bundle use \(\rho(g)=1\) and \(b_{ij}=0\). For vector or connection-like outputs use a supplied matrix representation \(\rho(g_{ij}^{-1})\); for the paper's connection law use \(\rho=\Ad\) and \(b_{ij}=g_{ij}^{\ast}\theta\).
(2) Pseudocode. Sample task points \(x\) and overlap points \(x_{ij}\). Compute \(y_i=f_i(u_i(x))\), \(y_j=f_j(u_j(x))\), then compute \(A_{ij}=\rho(g_{ij}(x)^{-1})\) and \(r_{ij}=y_j-A_{ij}y_i-b_{ij}(x)\). Minimize \(L_{task}+\sum_{ij}[\lambda_{ij}^{\mathsf T}r_{ij}+(\beta/2)\|r_{ij}\|^2]\). After each primal optimizer step, update dual variables by \(\lambda_{ij}\leftarrow\lambda_{ij}+\eta_\lambda r_{ij}\), where \(\eta_\lambda\) is a dual step size. If memory is limited, maintain one multiplier vector per overlap sample batch instead of a persistent multiplier at every point. Increase \(\beta\) geometrically only when validation jump error stops decreasing.
(3) Mathematical versus empirical quantities. The atlas, transition functions \(g_{ij}\), representation matrices \(A_{ij}\), and offset \(b_{ij}\) are computed analytically from the chosen manifold and bundle. Estimate the overlap integral by Monte Carlo or quadrature. The task loss, \(\beta\), primal learning rate, dual learning rate, and overlap sampling density are empirical choices. Check the cocycle identity numerically by comparing direct and two-hop transformations, \(A_{ik}\) versus \(A_{jk}A_{ij}\), before training. For a genuine 1-form output, obtain local tangent bases from chart Jacobians and apply the compatibility residual to each coefficient vector.
(4) First cheap experiment. Use a two-chart stereographic atlas on \(S^2\) and train a small four-layer MLP to regress a known smooth tangent vector field from randomly sampled sphere points. Compare (a) one global latitude-longitude MLP, (b) two independent chart MLPs with no gluing, (c) two chart MLPs with a fixed penalty \(\|r\|^2\), and (d) the proposed augmented-Lagrangian model. Hold total parameter count and sample count fixed. Measure task MSE, maximum and mean overlap residual, sensitivity near chart boundaries, and accuracy under random sphere rotations. Success is lower boundary error and more uniform rotated-sphere accuracy than the global-coordinate baseline at equal parameter count; a useful target is at least a tenfold reduction in overlap residual at matched task MSE with no meaningful increase in inference cost beyond evaluating the active chart head. A second test can use the Hopf-fibration transition law, where the nonzero affine term tests whether the network learns genuinely gauge-consistent outputs rather than merely matching duplicated values.
Verification
Stage 1 · Toy mechanism gate: Passed ✓
Stage 2 · Mechanism transferred to benchmark: Confirmed ✓
Stage 2 · Practical benchmark result: Significantly worse ✗
Methodology: 8 paired seeds, permutation test (p<0.05) against a learning-rate-tuned baseline. How verification works
Stage 1 — Mechanism check agent confidence 8/10
Built a two-chart periodic neural field with independent local MLPs, overlap residuals, fixed quadratic penalty, and augmented-Lagrangian dual updates. The toy math verified all three predictions: penalty residual 2/(1+beta), AL contraction |1-eta/(1+beta)|, and one-step annihilation at eta=1+beta, with relative errors at machine precision. In the neural comparison, AL reduced boundary residual to 0.000150 versus 0.000190 independent and 0.002876 penalty, but its mean overlap residual 0.009265 was worse than independent 0.001583 and penalty 0.005399; therefore the mechanism is confirmed, but a broad accuracy win is not established.
- Agent confidence
- 8/10
- Baseline
- Global MLP task MSE=1.276e-4 (2.15 s); independent two-chart MLP task MSE=1.987e-6, overlap residual=1.583e-3, boundary residual=1.900e-4 (6.23 s). Fixed penalty task MSE=3.912e-4, overlap residual=5.399e-3, boundary residual=2.876e-3 (8.46 s).
- Idea
- Augmented-Lagrangian two-chart model with beta=4 and eta_lambda=0.02: task MSE=2.142e-4, overlap residual=9.265e-3, boundary residual=1.501e-4 (8.22 s). Exact sanity sweeps matched predictions: residual scaling and contraction relative errors <=1.6e-16; one-step residual was 0 for beta=0.25,1,4,16.
Limitations: The neural test used a scalar trivial bundle on a periodic circle with A=1 and b=0, not the sphere, connection 1-forms, nonzero affine Maurer-Cartan term, or nontrivial structure-group representation. It used one fixed seed and dataset, did not test random rotations, generalization, multiple seeds, adaptive beta, equal parameter counts, or rigorous equal-FLOP accounting. The independent model was already highly accurate on this easy target, limiting room for improvement.
How to run: python3 bundle_glued_experiment.py
Stage 2 — Benchmark vs. tuned baseline (arXiv:2608.02108)
Significantly WORSE than the tuned baseline (+24107.2%, p=0.0081, wins 0 / 8 paired seeds).
- Benchmark
- Periodic_atlas_regression
- Model
- atlas_mlp
- Paired seeds
- 8
- Baseline mean
- 0.0003
- Idea mean
- 0
- Effect (Δ)
- 0.0661 (+24107.2%; negative = idea better)
- Wins
- 0 / 8 paired seeds
- p-value
- 0.0081 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±8547.5%
- Mechanism
- Confirmed ✓
- Practical effect
- Hurts
- Baseline tuning
- swept over 3 configs
The custom task is a small one-dimensional periodic atlas regression rather than a sphere, nontrivial representation, connection one-form, or nonzero affine Maurer-Cartan transition. The baseline and idea use a compact custom two-head MLP and 30 epochs; no larger manifold, higher-dimensional field, adaptive beta schedule, rotation robustness, or inference-cost study was tested. The reported baseline signature contains the trained baseline model, while the top-level mechanism confirmation records only the residual-reduction prediction; the nested per-model confirmed flags are threshold checks rather than the primary decision.
How to run: python3 bench_bundle_glued.py
Verdict computed by deterministic test code from paired-seed statistics — not by the language model.
Stage 2 — Benchmark latest Failed ✗
Agent confidence: 9/10
Built and evaluated a custom periodic-atlas regression benchmark because no built-in track contains overlapping-chart compatibility structure. The augmented-Lagrangian system reduced overlap residual from 0.3508 to 0.1208 in the reported trained-model signature, but test MSE increased from 0.0002744 to 0.06614 mean, with paired delta +0.06614 and permutation p=0.0081. The mechanism's residual-reduction prediction was observed, but the standard task metric became significantly worse, so the idea is not a benchmark win.
- Baseline
- Best lr=0.012, 30 epochs; 8-seed test MSE mean=0.0002743525, std=0.0001253615; sweep means: lr 0.003=0.00483641, 0.006=0.00073137, 0.012=0.00032798.
- Idea
- Best beta=4.0, eta=0.02, lr=0.012, 30 epochs; 8-seed test MSE mean=0.0664131030, with per-seed values [0.04835226, 0.05208624, 0.10922238, 0.11329208, 0.05160725, 0.04528246, 0.05048604, 0.06097611]. Paired delta=+0.06613875, p=0.0081, 0/8 wins. Trained-model overlap absolute residual signature: baseline=0.3507734, idea=0.1208182; residual prediction confirmed, task outcome negative.
Benchmark result
Significantly WORSE than the tuned baseline (+24107.2%, p=0.0081, wins 0 / 8 paired seeds).
- Benchmark
- Periodic_atlas_regression
- Model
- atlas_mlp
- Paired seeds
- 8
- Baseline mean
- 0.0003
- Idea mean
- 0
- Effect (Δ)
- 0.0661 (+24107.2%; negative = idea better)
- Wins
- 0 / 8 paired seeds
- p-value
- 0.0081 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±8547.5%
- Mechanism
- Confirmed ✓
- Practical effect
- Hurts
- Baseline tuning
- swept over 3 configs
The custom task is a small one-dimensional periodic atlas regression rather than a sphere, nontrivial representation, connection one-form, or nonzero affine Maurer-Cartan transition. The baseline and idea use a compact custom two-head MLP and 30 epochs; no larger manifold, higher-dimensional field, adaptive beta schedule, rotation robustness, or inference-cost study was tested. The reported baseline signature contains the trained baseline model, while the top-level mechanism confirmation records only the residual-reduction prediction; the nested per-model confirmed flags are threshold checks rather than the primary decision.
How to run: python3 bench_bundle_glued.py
Artifacts
- bench_bundle_glued.py 5.5 KB View Raw
- bench_report.json 3.3 KB View Raw
- bundle_glued_experiment.py 6.4 KB View Raw
- periodic_atlas_track.py 0.7 KB View Raw
- report.md 2.0 KB View
- report_bench_2026-08-31T201257.md 5.8 KB View
- results.json 2.7 KB View Raw
- sweep_tmp.py 6.3 KB View Raw