Cross-domain amortized inverse operator
Implementation & benchmark of arXiv:2607.01128 — GAIA: Geometry-Adaptive Operator Learning for Forward and Inverse Problems
Source paper: GAIA: Geometry-Adaptive Operator Learning for Forward and Inverse Problems arXiv:2607.01128 ⓘ · analyzed Aug 30, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Use separate measurement-domain and target-domain token sets so a network can infer a field on one spatial domain from sparse observations on another in one forward pass. The same decoder can answer arbitrary target query points, avoiding an optimization loop for each inverse instance.
Formulas
Mathematical statement
The paper distinguishes the forward operator $\mathcal{S}:\mathcal{A}\to\mathcal{U}$, which maps $a=(c,q,u_b,\chi_D)$ to the full solution $u$, from the inverse setting, where the full field is unavailable and only observables are given through a measurement operator $\mathcal{M}:\mathcal{U}\to\mathcal{Y}$. Here $\mathcal{U}$ is the space of fields on the target domain and $\mathcal{Y}$ is the measurement space. The transferable construction is a learned approximate inverse $\widehat{\mathcal{I}}:\mathcal{Y}\times\mathcal{G}\to\mathcal{U}$, where $\mathcal{G}$ describes source and target geometry. Given measurement tokens $h_j=E_y([y_j,m_j,\chi_{D_s}(y_j)])$, target query tokens $t_i=E_x([x_i,\chi_{D_t}(x_i),\mathrm{geometry}(x_i)])$, and geometry tokens $g_k$, decode with cross-attention from target queries to measurements and geometry: $r_i=\sum_j\beta_{ij}V_yh_j+\sum_k\gamma_{ik}V_gg_k$, with $\beta_{ij}$ and $\gamma_{ik}$ separately normalized softmax weights. Train against the target field using a masked loss over valid target points, so the architecture naturally supports different source and target domains and sparse measurements.
Implementation notes
(1) Integration point: modify an encoder-decoder used for inverse problems, sparse sensing, neural fields, or sensor-to-field reconstruction. The encoder consumes observations $m_j$ at source locations $y_j$; the decoder accepts arbitrary target coordinates $x_i$, including a target domain with a different shape or resolution. Include a binary valid-domain mask and boundary descriptors in both token types. (2) Pseudocode: construct $h_j=E_y([m_j,y_j,\mathrm{source\_geometry}_j])$; construct geometry tokens $g_k$ from boundary samples, normals, signed distance, and global shape pooling; for each target query form $q_i=Q(E_x([x_i,\chi_{D_t}(x_i),\mathrm{target\_geometry}_i]))$; calculate measurement keys and values $(k_j,V_yh_j)$ and geometry keys and values $(\bar k_k,V_gg_k)$; calculate $\beta_{ij}$ and $\gamma_{ik}$ with the two softmax equations; decode $\widehat u(x_i)=D(t_i,\sum_j\beta_{ij}V_yh_j+\sum_k\gamma_{ik}V_gg_k)$. Train with the masked normalized MSE and optionally add a PDE residual only during training. (3) Estimate from data: $\mathcal M$ is represented by the actual sensor sampling pattern, while all attention weights are computed directly; geometry features can be calculated from meshes or point clouds. (4) First cheap experiment: use a 2D Poisson or Darcy inverse task where 16--64 boundary or interior sensors must reconstruct a dense field on unseen shapes. Compare against a neural field conditioned on pooled measurements, a standard cross-attention model without geometry tokens, and per-instance gradient-based inversion. Measure relative $L^2$ reconstruction error, calibration under sensor dropout, inference latency, and the number of optimization steps avoided. Success is lower error on unseen geometries and constant one-pass latency as target resolution increases, with graceful degradation under missing measurements.
Verification
Stage 1 · Toy mechanism gate: Passed ✓
Stage 2 · Mechanism transferred to benchmark: Not reproduced ✗
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 cross-domain amortized inverse operator with separate measurement tokens, arbitrary target queries, source/target geometry conditioning, and cross-attention, plus a pooled-measurement baseline. The mathematical mechanism checks passed: softmax normalization stayed below 3.4e-16 across source-count sweeps, uniform duplicated-token invariance stayed below 1.2e-16, and geometry normalization stayed below 3.4e-16. In the toy reconstruction, the idea achieved lower relative L2 error than the baseline (0.288 vs 0.424), degraded less under 50% sensor dropout (0.667 vs 0.724), and remained stable when target resolution increased from 64 to 256 points (0.286 vs 0.278).
- Agent confidence
- 8/10
- Baseline
- relative L2 0.4238; 50% dropout 0.7242; training 56.1 s
- Idea
- relative L2 0.2875; 50% dropout 0.6665; target resolution 64: 0.2857, 256: 0.2775; training 67.8 s
Limitations: This is a synthetic smooth-field toy problem rather than a Poisson/Darcy PDE or unseen-mesh benchmark. Geometry uses compact analytic descriptors instead of boundary point clouds, the geometry branch has only one token in the implemented toy, and no per-instance gradient-based inversion or calibrated uncertainty was tested.
How to run: python3 experiment.py
Stage 2 — Benchmark vs. tuned baseline (arXiv:2607.01128)
Significantly WORSE than the tuned baseline (+283.9%, p=0.0081, wins 0 / 8 paired seeds).
- Benchmark
- Sequence modeling
- Model
- transformer_tiny
- Paired seeds
- 8
- Baseline mean
- 0.2635
- Idea mean
- 1.0116
- Effect (Δ)
- 0.7481 (+283.9%; negative = idea better)
- Wins
- 0 / 8 paired seeds
- p-value
- 0.0081 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±26.8%
- Mechanism
- Not confirmed ✗
- Practical effect
- Hurts
- Baseline tuning
- swept over 3 configs
Only the registered sequence track was tested; no PDE/Poisson custom track, unseen geometry transfer, sparse sensor reconstruction, target-resolution scaling, dropout calibration, or per-instance optimization baseline was tested.
How to run: python3 bench_experiment.py
Verdict computed by deterministic test code from paired-seed statistics — not by the language model.
Stage 2 — Benchmark latest Failed ✗
Agent confidence: 10/10
Built and evaluated an end-to-end cross-domain-style measurement/target cross-attention model against an independently trained standard transformer on the registered sequence track. The mechanism signature was confirmed from trained-model behavior, but the idea had significantly worse test MSE, so it did not produce a benchmark win.
- Baseline
- sequence transformer_tiny, best lr=0.001, 10 epochs; test MSE mean 0.2635204885; baseline sweep means: lr 0.001=0.2537163, lr 0.003=0.2935706, lr 0.006=0.3902993
- Idea
- cross-domain measurement/target attention with geometry token, best lr=0.003, 10 epochs; test MSE mean 1.0116473064; paired delta +0.7481268179, 0/8 wins, permutation p=0.0081; effective source count 31.9783
Benchmark result
Significantly WORSE than the tuned baseline (+283.9%, p=0.0081, wins 0 / 8 paired seeds).
- Benchmark
- Sequence modeling
- Model
- transformer_tiny
- Paired seeds
- 8
- Baseline mean
- 0.2635
- Idea mean
- 1.0116
- Effect (Δ)
- 0.7481 (+283.9%; negative = idea better)
- Wins
- 0 / 8 paired seeds
- p-value
- 0.0081 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±26.8%
- Mechanism
- Not confirmed ✗
- Practical effect
- Hurts
- Baseline tuning
- swept over 3 configs
Only the registered sequence track was tested; no PDE/Poisson custom track, unseen geometry transfer, sparse sensor reconstruction, target-resolution scaling, dropout calibration, or per-instance optimization baseline was tested.
How to run: python3 bench_experiment.py
Artifacts
- bench_experiment.py 6.7 KB View Raw
- bench_report.json 3.5 KB View Raw
- experiment.py 6.6 KB View Raw
- report.md 1.5 KB View
- report_bench_2026-08-30T145631.md 5.2 KB View
- results.json 1.6 KB View Raw
- signature_observed.json 0.2 KB View Raw