pdebench-reval-26 commited on
Upload folder using huggingface_hub
Browse files- README.md +56 -0
- test_01_best_model.pt +3 -0
- test_02_best_model.pt +3 -0
- test_03_best_model.pt +3 -0
- test_04_best_model.pt +3 -0
- test_05_best_model.pt +3 -0
- test_06_best_model.pt +3 -0
- test_07_best_model.pt +3 -0
- test_08_best_model.pt +3 -0
- test_09_best_model.pt +3 -0
- test_10_best_model.pt +3 -0
- test_11_best_model.pt +3 -0
- test_13_best_model.pt +3 -0
- test_16_best_model.pt +3 -0
- test_17_best_model.pt +3 -0
- test_19_best_model.pt +3 -0
- test_20_best_model.pt +3 -0
- test_21_best_model.pt +3 -0
- test_22_best_model.pt +3 -0
- test_23_best_model.pt +3 -0
- test_24_best_model.pt +3 -0
- test_25_best_model.pt +3 -0
- test_26_best_model.pt +3 -0
- test_27_best_model.pt +3 -0
- test_29_best_model.pt +3 -0
README.md
ADDED
|
@@ -0,0 +1,56 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
tags:
|
| 4 |
+
- fno
|
| 5 |
+
- neural-operator
|
| 6 |
+
- pdebench
|
| 7 |
+
- pde
|
| 8 |
+
- scientific-ml
|
| 9 |
+
---
|
| 10 |
+
|
| 11 |
+
# PDEBench FNO Re-evaluation: Model Weights
|
| 12 |
+
|
| 13 |
+
Pre-trained model checkpoints from "The Unrealized Potential of Fourier Neural Operators: A Systematic Re-evaluation of PDEBench Baselines" (NeurIPS 2026 E&D Track submission).
|
| 14 |
+
|
| 15 |
+
## Files
|
| 16 |
+
|
| 17 |
+
| File | Test | PDE | nRMSE | Improvement |
|
| 18 |
+
|------|------|-----|-------|-------------|
|
| 19 |
+
| `test_01_best_model.pt` | 1 | Advection β=1.0 | 4.68e-3 | 2.07× |
|
| 20 |
+
| `test_02_best_model.pt` | 2 | Burgers ν=0.1 | 1.41e-3 | 2.05× |
|
| 21 |
+
| `test_03_best_model.pt` | 3 | 1D Diff-React ν=0.5 | 1.23e-3 | 1.13× |
|
| 22 |
+
| `test_04_best_model.pt` | 4 | Burgers ν=0.001 | 1.08e-2 | 2.69× |
|
| 23 |
+
| `test_05_best_model.pt` | 5 | 1D Comp NS η=ζ=0.01 | 1.56e-2 | 6.08× |
|
| 24 |
+
| `test_06_best_model.pt` | 6 | 1D Comp NS Shock | 1.32e-2 | 3.57× |
|
| 25 |
+
| `test_07_best_model.pt` | 7 | Advection β=0.1 | 3.29e-3 | 2.34× |
|
| 26 |
+
| `test_08_best_model.pt` | 8 | Advection β=0.4 | 4.60e-3 | 2.18× |
|
| 27 |
+
| `test_09_best_model.pt` | 9 | Advection β=4.0 | 4.65e-3 | 1.44× |
|
| 28 |
+
| `test_10_best_model.pt` | 10 | Burgers ν=0.01 | 4.16e-3 | 1.87× |
|
| 29 |
+
| `test_11_best_model.pt` | 11 | Burgers ν=1.0 | 4.18e-3 | 0.96× (miss) |
|
| 30 |
+
| `test_13_best_model.pt` | 13 | 1D Diff-React ν=2.0 | 3.45e-4 | 2.03× |
|
| 31 |
+
| `test_16_best_model.pt` | 16 | Diff-Sorp | 9.95e-4 | 1.71× |
|
| 32 |
+
| `test_17_best_model.pt` | 17 | 1D Comp NS η=ζ=0.1 | 6.76e-3 | 10.05× |
|
| 33 |
+
| `test_19_best_model.pt` | 19 | 1D Comp NS Inv Rand | 3.00e-2 | 4.00× |
|
| 34 |
+
| `test_20_best_model.pt` | 20 | 1D Comp NS Inv Outg | 2.36e-1 | 28.44× |
|
| 35 |
+
| `test_21_best_model.pt` | 21 | Darcy β=0.01 | 2.67e-1 | 9.36× |
|
| 36 |
+
| `test_22_best_model.pt` | 22 | Darcy β=0.1 | 1.18e-1 | 1.87× |
|
| 37 |
+
| `test_23_best_model.pt` | 23 | Darcy β=1.0 | 2.77e-2 | 2.31× |
|
| 38 |
+
| `test_24_best_model.pt` | 24 | Darcy β=10.0 | 1.15e-2 | 1.04× |
|
| 39 |
+
| `test_25_best_model.pt` | 25 | Darcy β=100.0 | 9.50e-3 | 0.54× (miss) |
|
| 40 |
+
| `test_26_best_model.pt` | 26 | 2D Diff-React | 2.49e-3 | 48.21× |
|
| 41 |
+
| `test_27_best_model.pt` | 27 | 2D SWE | 1.90e-3 | 2.31× |
|
| 42 |
+
| `test_29_best_model.pt` | 29 | 2D Comp CFD | 3.98e-3 | 42.75× |
|
| 43 |
+
|
| 44 |
+
## Usage
|
| 45 |
+
|
| 46 |
+
```python
|
| 47 |
+
import torch
|
| 48 |
+
from standalone.models import FNO1d_AR # from the code repo
|
| 49 |
+
|
| 50 |
+
model = FNO1d_AR(nc=1, modes=12, width=32, init_step=5, n_layers=4)
|
| 51 |
+
model.load_state_dict(torch.load("test_13_best_model.pt", weights_only=True))
|
| 52 |
+
```
|
| 53 |
+
|
| 54 |
+
## Baselines
|
| 55 |
+
|
| 56 |
+
All improvement factors computed against PDEBench FNO (arXiv:2210.07182v7).
|
test_01_best_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1d0ab5c9d6742e36601afdb687b7f5de9df21ea5d5b91e8e448057ef985624c9
|
| 3 |
+
size 2207317
|
test_02_best_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a8d44248e4dd90e0bc701c4543b88a050a1e8afae1a343210dcca372852ce66a
|
| 3 |
+
size 2207317
|
test_03_best_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0df1dcaf1c02fbdbc111a335dad21a5827f9e206f0146ecc7096ca27c9a451dc
|
| 3 |
+
size 9647701
|
test_04_best_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5bc5fe7dd2be72ad076ef9135297a752e5d18bc5a9634d64f0111e3404cade97
|
| 3 |
+
size 8962082
|
test_05_best_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a8e4311a1dbfa7defd3c54de11d9d7ac5bbe706a67dd71dcdcade494ddc47a92
|
| 3 |
+
size 4773922
|
test_06_best_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e555e48e05c4d093094e0fa79dee69a63caae42a173097d29151a75272a2401f
|
| 3 |
+
size 4773922
|
test_07_best_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:af743a6890f9596ff7d98c85b4b26753d269da35a058415c745a7b2a2533d3e1
|
| 3 |
+
size 2206818
|
test_08_best_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:93edba9b1ad1c7af95062e91ca3edba04f9925cca9030895226ece550397b91e
|
| 3 |
+
size 2206818
|
test_09_best_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:004e9eecf1aa1d1249fdcb19c4b56e599b4b8a71a6b9516f5a33eb0aaf8e210e
|
| 3 |
+
size 2206818
|
test_10_best_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9887d8d4f098fc60ac758e9ca3d256370fd91ffb4a1373bf58c13a1e3547926d
|
| 3 |
+
size 10684962
|
test_11_best_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:afba6510ff0ae81d5b29e36be32406cfe55b427487f48635c0d096c2500dfef4
|
| 3 |
+
size 10684962
|
test_13_best_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d5e0999f1e32184bead1cd14cd4e8333687abd0be4421ed80cef064a4699c715
|
| 3 |
+
size 10684962
|
test_16_best_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8bb180db38f2872f9e8b1405dbd1886d9be58278aedd34e39cd5b7916c449156
|
| 3 |
+
size 10684962
|
test_17_best_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e0105ccd3b7d9b991bc19b167c47449a1798f13290043f5f432bfe1c4d6a2ae8
|
| 3 |
+
size 4773922
|
test_19_best_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0267e731e81763c9bfe07b0acf6af03c794f258437ba1c957f0a125204f78281
|
| 3 |
+
size 4773922
|
test_20_best_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c2731361bbe84d6f494bfc5cc7827c4eb44c0e15d14b0e76190f68386e9ed72c
|
| 3 |
+
size 4773922
|
test_21_best_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a502d07bd536bad1709432a316e5ce6a87dd8ae32e3396edcdadb6e4ecb0eab1
|
| 3 |
+
size 9893794
|
test_22_best_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2078e36d9a78138a0f3f13525ca18df6ecaae51a5626db40040fd0e009321fb9
|
| 3 |
+
size 9893794
|
test_23_best_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:124adae33d13afb66e3bcc07663c327d51f8ff18ccccb52ce4af27b21574395b
|
| 3 |
+
size 9893794
|
test_24_best_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:927f2e19dcd4e10404b412d0bc223eb7089ae0d5ce2390753916563b5e999f10
|
| 3 |
+
size 9893794
|
test_25_best_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3256cbc0f4e28a1df3e22831862cf6da0f9c7257e3a0bd41cb00608aa6b31aa7
|
| 3 |
+
size 39501218
|
test_26_best_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a41f4ef75ea0e8c4d48a0d5cb4204e657c0dee153edb963f336693542f44fbd8
|
| 3 |
+
size 9895458
|
test_27_best_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cbc9c4d98c5b70edcfdaf66ec111f9348fd32cdae13f3558b4f090050a39f377
|
| 3 |
+
size 9894306
|
test_29_best_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d6b9bcdc60057336d977ebdcd7489335865dc09fe72c6ec49d9260711728d9e2
|
| 3 |
+
size 9897570
|