Reconstruction and analysis of negatively buoyant jets with interpretable machine learning

In this paper, negatively inclined buoyant jets, which appear during the discharge of wastewater from processes such as desalination, are observed. A detailed numerical investigation is necessary to minimize harmful effects and assess environmental impact. Selecting appropriate geometry and working conditions for minimizing such effects often requires numerous experiments and numerical simulations. For this reason, the application of machine learning models is proposed. Several models including Support Vector Regression, Artificial Neural Networks, Random Forests, XGBoost, CatBoost and LightGBM were trained. The dataset was built with numerous OpenFOAM simulations, validated by experimental data from previous research. The average prediction of ML models has R2 0.94±0.05, RMSE 0.42±0.14 and RRSE 0.24 ± 0.09, whereas the best prediction was obtained by Artificial Neural Network with R2 0.98, RMSE 0.28 and RRSE 0.16. To understand the influence of input parameters on the geometrical characteristics of inclined buoyant jets, the SHAP feature interpretation method was used. [Display omitted] •ML is used on inclined buoyant jets with different geometrical characteristics.•ANN model has shown the best performance.•The SHAP method for explainability and feature interpretation is applied.•Effluent velocity is shown to be the most significant parameter.•The Coanda effect and shallow water effect have been investigated.