✗ Failed on benchmark 2026

Cross-domain amortized inverse operator

Implementation & benchmark of arXiv:2607.01128 — GAIA: Geometry-Adaptive Operator Learning for Forward and Inverse Problems

Usefulness7/10
Difficulty5/10
Novelty6/10

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

$$\mathcal{M}:\mathcal{U}\to\mathcal{Y},\qquad y=\mathcal{M}(u),$$
$$\widehat{\mathcal{I}}:\mathcal{Y}\times\mathcal{G}\to\mathcal{U},\qquad \widehat u=\widehat{\mathcal{I}}(y,\mathcal{G}),$$
$$\beta_{ij}=\frac{\exp(q_i^{\mathsf T}k_j/\sqrt r)}{\sum_{\ell=1}^{N_s}\exp(q_i^{\mathsf T}k_\ell/\sqrt r)},\qquad \gamma_{ik}=\frac{\exp(q_i^{\mathsf T}\bar k_k/\sqrt r)}{\sum_{\ell=1}^{K}\exp(q_i^{\mathsf T}\bar k_\ell/\sqrt r)},$$
$$\widehat u(x_i)=D\!\left(t_i,\sum_{j=1}^{N_s}\beta_{ij}V_yh_j+\sum_{k=1}^{K}\gamma_{ik}V_gg_k\right),\qquad \mathcal{L}=\frac{\sum_i\chi_{D_t}(x_i)\lVert \widehat u(x_i)-u(x_i)\rVert_2^2}{\sum_i\chi_{D_t}(x_i)}.$$

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

Failed on benchmark

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
Limitations:

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
Limitations:

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

Implementation overview ⬇ Download all as ZIP 7 files · code, reports and structured results