Tropical support-restricted PINN
Implementation & benchmark of arXiv:2607.00237 — Tropical Geometry as a Restricted Architecture for Physics-Informed Neural Networks: Applications in Nonlinear Fluid-Structure Examples
Source paper: Tropical Geometry as a Restricted Architecture for Physics-Informed Neural Networks: Applications in Nonlinear Fluid-Structure Examples arXiv:2607.00237 ⓘ · analyzed Aug 29, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Represent the PINN solution in a restricted polynomial or Taylor basis whose exponent set is supplied by tropical support analysis, instead of asking an MLP to discover the local series structure from scratch. The restriction removes coefficients that cannot occur in the formal solution, reducing trainable degrees of freedom and preventing spurious low-order or singular terms.
Formulas
Mathematical statement
The paper defines the Boolean semiring \(\mathbb{B}=\{0_{\mathbb{B}}<1_{\mathbb{B}}\}\), with Boolean addition \(x+y=\max\{x,y\}\), so addition records whether a term is present. For a formal series \(\Phi(t)=\sum_{i=0}^{\infty}a_i t^i\in\mathbb{C}[[t]]\), its tropicalization or support is \(\operatorname{supp}(\Phi)=\sum_i v_0(a_i)t^i\), where \(v_0(0)=0_{\mathbb{B}}\) and \(v_0(a)=1_{\mathbb{B}}\) for \(a\neq0\). Its order is \(\operatorname{ord}(\Phi)=\min\{i:a_i\neq0\}\), with \(\operatorname{ord}(0)=\infty\). Let \(A\subseteq\{0,\ldots,K\}\) be the admissible exponent set obtained by tropicalizing the differential-equation residual and retaining exponents compatible with the formal solution; define the restricted ansatz \(u_\theta(t)=\sum_{i\in A}c_i(\theta)t^i\). The mathematical asset is structural sparsity: coefficients outside \(A\) are exactly zero by construction, and the smallest active exponent is controlled by the order constraint rather than learned through a badly conditioned residual loss.
Implementation notes
Integrate this at the PINN output layer, not inside the residual network. First choose a local coordinate \(t\), such as time for Van der Pol or a spatial coordinate near a boundary or shock for Burgers, and a truncation order \(K\). Construct the candidate basis \(\{t^0,\ldots,t^K\}\). From the polynomial differential equation, enumerate candidate formal-series coefficients: initialize a Boolean support mask \(s_i=1\), substitute \(u=\sum_{i=0}^{K}c_i t^i\) into the differential operator, apply derivative shifts \(t^i\mapsto i t^{i-1}\), and apply polynomial products by coefficient convolution. Use the resulting coefficient equations to identify exponents whose coefficients are forced to zero; define \(A=\{i:s_i=1\}\) before training. Use an MLP \(h_\theta(t)\in\mathbb{R}^{|A|}\) and map its outputs to \(u_\theta(t)=\sum_{j=1}^{|A|}h_{\theta,j}(t)t^{A_j}\). A simpler first implementation uses one trainable coefficient per allowed exponent, while a nonlinear coefficient head can be tested afterward. Compute derivatives analytically or with autodiff and minimize the PDE residual and initial or boundary loss in the displayed objective. Enforce a prescribed order by excluding all exponents below the required \(m\), or initialize the first active coefficient separately when it must be nonzero. The paper supplies the Boolean support and order machinery; empirical estimation is only needed when symbolic elimination is too expensive. Begin with a two-layer tanh PINN on one-dimensional Van der Pol and Burgers, using \(K=20\), and compare a dense basis with the tropical mask at equal hidden width and collocation count. Measure residual loss versus optimizer steps, final relative solution error, number of stalled runs, and error near stiff or singular regions. The hypothesis is supported if the masked model reaches a fixed error in fewer steps, has lower run-to-run variance, or achieves lower local error with equal FLOPs.
Verification
Stage 1 · Toy mechanism gate: Passed ✓
Stage 2 · Mechanism transferred to benchmark: Confirmed ✓
Stage 2 · Practical benchmark result: Significant win ✓
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 coefficient-based tropical support-restricted PINN for u'=1+u^2, u(0)=0, using the formally derived odd-power support of tan(t). The recurrence check exactly produced zero even coefficients, and across three fixed seeds the restricted model used 8 rather than 16 coefficients, reduced mean residual loss from 4.55e-6 to 8.42e-9, and reduced relative solution error from 2.25e-4 to 7.86e-6 at equal optimization steps. This is a clear toy-system signal for the claimed structural-sparsity effect, but it is not evidence yet for general PDEs or MLP coefficient heads.
- Agent confidence
- 8/10
- Baseline
- Dense exponents 0..15; 16 parameters; mean final residual 4.554e-6; mean relative tan(t) error 2.246e-4 over 3 seeds.
- Idea
- Odd exponents {1,3,...,15}; 8 parameters; mean final residual 8.419e-9; mean relative error 7.865e-6 over 3 seeds; exact formal-series check max even coefficient 0.0.
Limitations: Only a one-dimensional Riccati ODE on t in [0,1] was tested, with one trainable scalar coefficient per basis exponent rather than the proposed nonlinear MLP coefficient head. No Van der Pol, Burgers, wall-clock speed, FLOP-normalized comparison, or broader support-analysis implementation was tested.
How to run: python3 run_experiment.py
Stage 2 — Benchmark vs. tuned baseline (arXiv:2607.00237)
Beats the tuned baseline by 95.2% (p=0.0081, wins 8 / 8 paired seeds; metric: lower is better).
- Benchmark
- Tropical_bvp_local
- Model
- mlp_tiny
- Paired seeds
- 8
- Baseline mean
- 0.0009
- Idea mean
- 0
- Effect (Δ)
- -0.0009 (−95.2%; negative = idea better)
- Wins
- 8 / 8 paired seeds
- p-value
- 0.0081 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- n/a
- Mechanism
- Confirmed ✓
- Practical effect
- Helps
- Baseline tuning
- swept over 3 configs
No registered PDE track was available, so a local manufactured 1D BVP custom track was used. The benchmark used supervised solution regression rather than autodiff PDE-residual training, one sine solution and one truncation order, and did not measure wall-clock or FLOP-normalized speed.
How to run: python3 stage2_bench.py
Verdict computed by deterministic test code from paired-seed statistics — not by the language model.
Stage 2 — Benchmark latest Worked ✓
Agent confidence: 8/10
The tropical support-restricted model significantly outperformed the dense polynomial-output baseline on the matched custom BVP track. Mean test MSE was 4.335e-05 versus 9.053e-04 over 8 paired seeds, with delta_mean=-8.619e-04 and permutation p=0.0081. The trained-model mechanism signature confirmed baseline off-support contribution and zero restricted off-support contribution.
- Baseline
- Dense exponents 0..15, best lr=0.01, mean test MSE 0.000905269, std 0.000435092, 8 seeds.
- Idea
- Odd exponents {1,3,5,7,9,11,13,15}, best lr=0.01, mean test MSE 0.000043348, std 0.000034194, 8 seeds.
- Idea wins
- 8
- N pairs
- 8
- Paired delta mean
- -0.000861921
- Permutation p value
- 0.0081
Benchmark result
Beats the tuned baseline by 95.2% (p=0.0081, wins 8 / 8 paired seeds; metric: lower is better).
- Benchmark
- Tropical_bvp_local
- Model
- mlp_tiny
- Paired seeds
- 8
- Baseline mean
- 0.0009
- Idea mean
- 0
- Effect (Δ)
- -0.0009 (−95.2%; negative = idea better)
- Wins
- 8 / 8 paired seeds
- p-value
- 0.0081 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- n/a
- Mechanism
- Confirmed ✓
- Practical effect
- Helps
- Baseline tuning
- swept over 3 configs
No registered PDE track was available, so a local manufactured 1D BVP custom track was used. The benchmark used supervised solution regression rather than autodiff PDE-residual training, one sine solution and one truncation order, and did not measure wall-clock or FLOP-normalized speed.
How to run: python3 stage2_bench.py
Artifacts
- bench_report.json 4.2 KB View Raw
- report.md 1.5 KB View
- report_bench_2026-09-03T232251.md 4.0 KB View
- results.json 3.2 KB View Raw
- run_experiment.py 4.1 KB View Raw
- stage2_bench.py 5.2 KB View Raw
- tropical_bvp_track.py 0.7 KB View Raw