- Research Article
- 10.1088/1873-7005/ae1ab2
A holistic machine learning-lattice Boltzmann approach for data prediction in two-dimensional cavity flows in a square domain
- Nov 14, 2025
- Fluid Dynamics Research
- Sasithradevi Anbalagan + 3 more +3
This study investigates the prediction of two-dimensional lid-driven cavity fluid flow in a square domain using supervised machine learning techniques. The classical problem of lid-driven cavity flow is utilized to demonstrate the integration of the mesoscopic Lattice Boltzmann Method (LBM) with machine learning algorithms. The LBM is employed to generate macroscopic velocity and pressure data for varying Reynolds numbers (Re = 2000, 3200, 5000, and 7500). Machine learning-based regression models are then applied to forecast key macroscopic variables, including density, pressure, and velocity based on directional coordinate measurements. For all datasets, the Random Forest regressor consistently predicted velocity with high precision, with R² scores of 0.9998 (Re = 2000), 0.9997 (Re = 3200), 0.9996 (Re = 5000), and 0.9996 (Re = 7500), and RMSE values ranging between 0.0032–0.0042. For pressure prediction, XGBoost outperformed others with an R² score of 0.9250 and RMSE of 1.04×10⁻⁴, while Random Forest followed closely with R² = 0.9063 and RMSE = 1.02×10⁻⁴. On the other hand, Decision Tree (R² = 0.112) and Linear Regression (R² = 0.064) performed badly in terms of prediction, underscoring the superiority of ensemble-based approaches. The study further demonstrates that ensemble-based approaches like Random Forest and XGBoost outperform traditional regression methods, showing significant capability for accelerating the parametric research and enhancing design optimization in engineering domain.
Read more