WeirNet: A Large-Scale 3D CFD Benchmark for Geometric Surrogate Modeling of Piano Key Weirs
Organizations: Hydraulic Engineering Section, Civil Engineering, Helmut-Schmidt-University, Hamburg, Germany · Institute for Artificial Intelligence, Helmut-Schmidt-University, Hamburg, Germany · Professorship of Computer Science in Mechanical Engineering, Helmut-Schmidt-University, Hamburg, Germany · Federal Waterways Engineering and Research Institute, Karlsruhe, Germany ·
Abstract
Reliable prediction of hydraulic performance is challenging for Piano Key Weir (PKW) design because discharge capacity depends on three-dimensional geometry and operating conditions. Surrogate models can accelerate hydraulic-structure design, but progress is limited by scarce large, well-documented datasets that jointly capture geometric variation, operating conditions, and functional performance. This study presents WeirNet, a large 3D CFD benchmark dataset for geometric surrogate modeling of PKWs. WeirNet contains 3,794 parametric, feasibility-constrained rectangular and trapezoidal PKW geometries, each scheduled at 19 discharge conditions using a consistent free-surface OpenFOAM workflow, resulting in 71,387 completed simulations that form the benchmark and with complete discharge coefficient labels. The dataset is released as multiple modalities compact parametric descriptors, watertight surface meshes and high-resolution point clouds together with standardized tasks and in-distribution and out-of-distribution splits. Representative surrogate families are benchmarked for discharge coefficient prediction. Tree-based regressors on parametric descriptors achieve the best overall accuracy, while point- and mesh-based models remain competitive and offer parameterization-agnostic inference. All surrogates evaluate in milliseconds per sample, providing orders-of-magnitude speedups over CFD runtimes. Out-of-distribution results identify geometry shift as the dominant failure mode compared to unseen discharge values, and data-efficiency experiments show diminishing returns beyond roughly 60% of the training data. By publicly releasing the dataset together with simulation setups and evaluation pipelines, WeirNet establishes a reproducible framework for data-driven hydraulic modeling and enables faster exploration of PKW designs during the early stages of hydraulic planning.
Figures & tables
| Parameters | Definition |
| Upstream-downstream length of the PKW (total weir length) | |
| Upstream (outlet key) overhang crest length | |
| Downstream (inlet key) overhang crest length | |
| Base length | |
| Height of the inlet entrance measured from the PKW crest (including possible parapet walls) | |
| Height of the outlet entrance measured from the PKW crest (including possible parapet walls) |
| Study | Size | SpD | Dim. | Struct. type | Rep. | Outputs | Open | |
| [sims/cases] | [–] | [–] | (P/M/PC) | |||||
| Studies on PKW discharge coefficients | ||||||||
| Bansal et al. [ 4 ] | 162 | 9 | 18 | 3D | PKW key width ratio | ✓/ – / – | , Energy dissipation (exp.) | – |
| Iqbal & Ghani. [ 30 ] | 88 | 11 | 8 | 3D | PKW key width ratio & slope ratio | ✓/ – / – | (exp.) | – |
| Studies on Weir geometry datasets | ||||||||
| Eilermann et al. [ 14 ] | 15 | 15 | 1 | 2D | Weir geometries | ✓/ – / – | , free surface, | ✓ |
| Parameters | Definition |
| Inlet key width (from sidewall to sidewall) on the upstream side *1 | |
| Inlet key width (from sidewall to sidewall) on the downstream side *2 | |
| Outlet key width (from sidewall to sidewall) on the upstream side *2 | |
| Outlet key width (from sidewall to sidewall) on the downstream side *1 | |
| Ratio of the downstream (inlet key) overhang crest length to base length | |
| Ratio of the upstream (outlet key) overhang crest length to base length |
| Parameters | Min | Max | Constant |
| 1000 mm | |||
| 330 mm | |||
| 0.25 | 1 | ||
| 3 | |||
| Parameter | Computational Relationship |
| [l/s] | ||||
| 50 | 80 | 120 | 160 | 200 |
| 55 | 90 | 130 | 170 | 225 |
| 60 | 100 | 140 | 180 | 250 |
| 70 | 110 | 150 | 190 | |
| Model | MSE (↓) | (↑) | MAE (↓) | Max AE (↓) | Time/epoch (↓) | Inference Time (↓) | #Params |
| PointNet [ 57 ] | 15.30 | 98.96 | 9.06 | 1.67 | 10 s | 0.49 ms | 5,435,009 |
| Mesh-GCN [ 36 ] | 241.60 | 64.33 | 36.88 | 2.99 | 24 s | 0.90 ms | 100,481 |
| RegDGCNN [ 16 ] | 97.40 | 83.11 | 22.01 | 1.90 | 1054 s | 53.15 ms | 4,482,657 |
| Model | MSE (↓) | (↑) | MAE (↓) | Max AE (↓) |
| RandomForest [ 9 ] | 5.20 | 99.67 | 3.94 | 2.90 |
| XGBoost [ 10 ] | 6.80 | 99.58 | 5.53 | 1.44 |
| LightGBM [ 34 ] | 7.50 | 99.54 | 6.00 | 1.05 |
| GradientBoosting [ 21 ] | 119.50 | 92.63 | 19.61 | 3.54 |
| Model | Split | MSE (↓) | (↑) | MAE (↓) | Max MAE (↓) |
| RandomForest [ 9 ] | 1628.40 | -19.43 | 94.34 | 5.27 | |
| XGBoost [ 10 ] | 1350.50 | 0.95 | 86.13 | 4.78 | |
| LightGBM [ 34 ] | 1242.40 | 8.88 | 83.35 | 4.80 | |
| GradientBoosting [ 21 ] | 1582.60 | -16.07 | 94.28 | 5.19 | |
| RandomForest [ 9 ] | 87.20 | 87.59 | 19.11 | 4.93 | |
| XGBoost [ 10 ] | 37.10 | 94.72 | 14.53 | 1.57 |
Surrogate Modeling
Surrogate modeling builds faster or more interpretable approximations of expensive simulations and black-box models.
See all 226 papers