A pre-training model based on CFD for open-channel velocity field prediction with small sample data

Abstract

Accurately obtaining the distribution of the open-channel velocity field in hydraulic engineering is extremely important, which is helpful for better calculation of open-channel flow and analysis of open-channel water flow characteristics. In recent years, machine learning has been used for open-channel velocity field prediction. However, effective training of data-driven models in machine learning heavily depends on the diversity and quantity of data. In this paper, a CFD-based pre-training neural network model (CFD–PNN) is proposed for accurate open-channel velocity field prediction, allowing the adaption to the task with small sample data. Also, a cross-sectional velocity field prediction method combining the computational fluid dynamics (CFD) and machine learning is established. By comparing CFD–PNN with six other neural network algorithm models and the CFD model, the results show that, in the case of small sample data, the CFD–PNN model can predict a more reasonable open-channel velocity field with higher prediction accuracy than other models. The average error of the velocity calculation for the trapezoidal open-channel cross-section is about 3.62%. Compared with other models, the accuracy is improved by 0.3–2.8%.