arXiv:2602.20714cs.LGcs.CE2026-02被引 1

构建首个大规模3D水力模型数据集,加速钢琴键堰设计优化。

WeirNet: A Large-Scale 3D CFD Benchmark for Geometric Surrogate Modeling of Piano Key Weirs

  • 基于OpenFOAM生成3794种几何变体,覆盖19种工况,共7万余次仿真。
  • 参数化模型预测精度最优,点云/网格模型实现无需参数化的快速推理。
  • 公开数据集与评估流程,支持高效设计探索,适合水利结构优化研究者。

钢琴键堰(PKW)的水力性能预测因三维几何和运行条件复杂而困难。代理模型可加速水工结构设计,但受限于缺乏大规模、标注完善的联合数据集。本文提出WeirNet,一个面向PKW几何代理建模的大规模3D CFD基准数据集。WeirNet包含3,794个参数化、可行性约束的矩形与梯形PKW几何,每种在19种流量工况下通过一致的自由表面OpenFOAM流程模拟,共完成71,387次仿真,附完整泄流系数标签。数据以参数描述、封闭表面网格和高分辨率点云多模态形式发布,并提供标准任务及分布内/外划分。对代表性代理模型进行泄流系数预测基准测试:基于参数描述的树模型整体表现最佳,点云与网格模型亦具竞争力,且无需依赖参数化表示。所有代理模型单样本推理仅需毫秒级时间,相比CFD提升数个数量级。分布外结果表明几何偏移是主要失效模式,而非未见流量值;数据效率实验显示训练数据超过60%后收益递减。通过公开数据集、仿真设置与评估流程,WeirNet建立了可复现的数据驱动水力建模框架,助力早期水利规划中对PKW设计的快速探索。

原文摘要 · Abstract (English)

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.

水力模型代理建模数据集

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。