Hugging Face's logo

AI4Science-WestlakeU
/
RealPDEBench-models

physics
pde
neural-operator
fluid-dynamics
combustion
benchmark
Model card Files
xet
Community
1
RealPDEBench-models
File size: 3,758 Bytes 
8e1d5ab
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
afdefd8
8e1d5ab
 
afdefd8
 
8e1d5ab
399b694
afdefd8
 
399b694
afdefd8
399b694
 
8e1d5ab
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
 
---
license: cc-by-4.0
tags:
  - physics
  - pde
  - neural-operator
  - fluid-dynamics
  - combustion
  - benchmark
datasets:
  - AI4Science-WestlakeU/RealPDEBench
---

# RealPDEBench Model Checkpoints

Trained model checkpoints for [RealPDEBench](https://github.com/AI4Science-WestlakeU/RealPDEBench), a benchmark for evaluating neural PDE solvers on real-world experimental data.

## Models (10 architectures)

| Model | Type | File Size (per checkpoint) |
|-------|------|---------------------------|
| DPOT-L | Transformer | 2.5-2.6 GB |
| FNO | Spectral | 385M-2.1G |
| Galerkin Transformer | Transformer | 386-642M |
| WDNO | Diffusion | 351M-1.4G |
| DPOT-S | Transformer | 118-159M |
| U-Net | CNN | 88-89M |
| CNO | Hybrid | 31M |
| MWT | Wavelet | 22M |
| Transolver | Transformer | 17M |
| DeepONet | Neural Operator | 14M |

## Scenarios (5)

| Scenario | Description |
|----------|-------------|
| `cylinder` | Flow past a circular cylinder |
| `controlled_cylinder` | Actively controlled cylinder flow |
| `fsi` | Fluid-structure interaction |
| `foil` | Flow past an airfoil |
| `combustion` | Turbulent combustion |

## Training Paradigms

| File | Paradigm |
|------|----------|
| `numerical.pth` | Trained on numerical simulation data only |
| `real.pth` | Trained on real experimental data only |
| `finetune.pth` | Pretrained on numerical, finetuned on real |
| `numerical_base_for_finetune.pth` | Numerical pretrain base (DPOT-S/L only) |

## Directory Structure

```
{scenario}/{model}/{paradigm}.pth
configs/{scenario}/{model}.yaml
```

Example: `cylinder/fno/finetune.pth` + `configs/cylinder/fno.yaml`

## Quick Start

### Install

```bash
git clone https://github.com/AI4Science-WestlakeU/RealPDEBench.git
cd RealPDEBench && pip install -e .
```

### Download a Single Checkpoint

```python
from huggingface_hub import hf_hub_download

path = hf_hub_download(
    repo_id="AI4Science-WestlakeU/RealPDEBench-models",
    filename="cylinder/fno/finetune.pth",
)
```

### Download All Checkpoints for a Scenario

```python
from huggingface_hub import snapshot_download

snapshot_download(
    repo_id="AI4Science-WestlakeU/RealPDEBench-models",
    allow_patterns="cylinder/**",
    local_dir="./checkpoints",
)
```

### Evaluate

```bash
python eval.py --config configs/cylinder/fno.yaml \
    --checkpoint_path ./checkpoints/cylinder/fno/finetune.pth \
    --dataset_type real --test_mode all
```

## Checkpoint Format

```python
checkpoint = torch.load("cylinder/fno/finetune.pth")
# Keys: model_state_dict, train_losses, val_losses,
#        iteration, best_iteration, best_val_loss
```

## DPOT Pretrained Weights

DPOT models require pretrained backbone weights (**not included here**).
Download via:

```bash
# Option 1: Built-in download script
python -m realpdebench.utils.dpot_ckpts_dl

# Option 2: From HuggingFace directly
# https://huggingface.co/hzk17/DPOT
```

## Dataset

The corresponding dataset is hosted at: [AI4Science-WestlakeU/RealPDEBench](https://huggingface.co/datasets/AI4Science-WestlakeU/RealPDEBench)


## Citation

If you find our work and/or our code useful, please cite us via:

```bibtex
@inproceedings{hu2026realpdebench,
      title={RealPDEBench: A Benchmark for Complex Physical Systems with Real-World Data}, 
      author={Peiyan Hu and Haodong Feng and Hongyuan Liu and Tongtong Yan and Wenhao Deng and Tianrun Gao and Rong Zheng and Haoren Zheng and Chenglei Yu and Chuanrui Wang and Kaiwen Li and Zhi-Ming Ma and Dezhi Zhou and Xingcai Lu and Dixia Fan and Tailin Wu},
      booktitle={The Fourteenth International Conference on Learning Representations},
      year={2026},
      url={https://openreview.net/forum?id=y3oHMcoItR},
      note={Oral Presentation}
}
```

## License

CC BY 4.0