JCSIS
JCSIS

Machine Learning-Driven Energy Optimization in Multi-Stage Flash Desalination Plants Using Gradient Boosting Algorithms.

Lima Hongou
Volume 5

Abstract

Multi-Stage Flash (MSF) desalination remains one of the most established thermal desalination technologies for large-scale freshwater production, particularly in regions with high seawater salinity, limited freshwater resources, and access to thermal energy from power generation or industrial processes. Despite its operational robustness and ability to handle challenging feedwater conditions, MSF desalination is energy-intensive due to the substantial thermal energy required for brine heating and the electrical energy consumed by pumping, circulation, and auxiliary systems. Inefficient operating conditions, suboptimal brine recirculation, scaling formation, heat-transfer degradation, and variations in seawater temperature can significantly increase specific energy consumption and reduce plant productivity. Conventional optimization strategies often depend on simplified thermodynamic models, fixed operating rules, and manual operator experience, which may be insufficient to capture nonlinear interactions among process variables in complex multi-stage desalination plants. To address these limitations, this paper proposes a machine learning-driven energy optimization framework for MSF desalination plants using gradient boosting algorithms.

The proposed framework utilizes historical and real-time operational data collected from MSF plant monitoring systems, including top brine temperature, seawater intake temperature, brine heater outlet temperature, recycle brine flow rate, distillate production rate, stage pressure, stage temperature, flashing chamber performance, blowdown salinity, steam flow rate, cooling water flow rate, condenser performance, pump power consumption, heat-transfer coefficients, antiscalant dosing records, and cleaning history. A comprehensive preprocessing pipeline is applied to handle missing sensor readings, remove abnormal operating points, normalize numerical variables, align multi-rate process measurements, and construct energy-performance indicators such as gain output ratio, performance ratio, specific thermal energy consumption, specific electrical energy consumption, and total specific energy consumption.

Gradient boosting algorithms, including XGBoost, LightGBM, CatBoost, and Gradient Boosting Regression Trees, are employed to model the nonlinear relationship between plant operating conditions and energy efficiency outcomes. These models are trained to predict specific energy consumption, freshwater production rate, thermal efficiency, and operational performance under varying feedwater and process conditions. Feature selection, correlation analysis, Bayesian hyperparameter tuning, and cross-validation are incorporated to improve model generalization and reduce overfitting. The trained predictive models are then integrated into an optimization layer that identifies energy-efficient operating setpoints while satisfying technical constraints related to top brine temperature limits, allowable salinity, stage pressure boundaries, scaling risk, minimum production demand, steam availability, and pump operating limits.

The proposed framework can be evaluated using real-world MSF plant datasets, pilot-scale desalination records, or high-fidelity process simulation data. Model performance is assessed using Mean Absolute Error, Root Mean Square Error, Mean Absolute Percentage Error, coefficient of determination, prediction stability, and computational efficiency. Energy optimization performance is evaluated through reduction in specific energy consumption, improvement in gain output ratio, increase in distillate production efficiency, reduction in steam usage, pump energy savings, and maintenance of safe operating conditions. Comparative analysis is conducted against conventional regression models, artificial neural networks, support vector regression, random forest, rule-based plant optimization, and thermodynamic baseline models.

Experimental evaluation is expected to demonstrate that gradient boosting-based models provide high predictive accuracy and strong interpretability for MSF energy optimization. Their ability to capture nonlinear interactions among thermal, hydraulic, and salinity-related variables allows the framework to identify operating regimes that reduce unnecessary energy consumption without compromising water production or equipment safety. To improve transparency and operator trust, explainability techniques such as SHAP, feature importance analysis, partial dependence plots, and sensitivity analysis are incorporated. These tools help identify the most influential energy drivers, including top brine temperature, steam flow rate, recycle brine flow, seawater intake temperature, condenser performance, and stage pressure distribution.

The proposed machine learning-driven optimization framework offers a practical, scalable, and data-driven solution for improving energy efficiency in MSF desalination plants. By combining gradient boosting prediction models with operational constraint-aware optimization, the system can support real-time decision-making, reduce thermal and electrical energy consumption, improve plant productivity, and lower operational costs. Its integration into supervisory control and plant energy management systems can assist operators in achieving more sustainable desalination performance, particularly in water-stressed regions where energy-efficient freshwater production is essential.

 


Keywords


References

  • [1] El-Kenawy, E. S. M., Khodadadi, N., Mirjalili, S., Abdelhamid, A. A., Eid, M. M., & Ibrahim, A. (2024). Greylag goose optimizati on:
  • [2] Abdollahzadeh, B., Khodadadi, N., Barshandeh, S., Trojovský, P., Gharehchopogh, F. S., El -kenawy, E. S. M., ... & Mirjalili, S. (2024).
  • [3] El-Kenawy, E. S. M., Ibrahim, A., Mirjalili, S., Eid, M. M., & Hussein, S. E. (2020). Novel feature selection and voting classif ier
  • [4] El-Kenawy, E. S. M., Eid, M. M., Saber, M., & Ibrahim, A. (2020). MbGWO -SFS: Modified binary grey wolf optimizer based on
  • [5] El-Kenawy, E. S., & Eid, M. (2020). Hybrid gray wolf and particle swarm optimization for feature selection. Int. J. Innov. Compu t. Inf.
  • [6] El-Kenawy, E. S. M., Mirjalili, S., Ibrahim, A., Alrahmawy, M., El -Said, M., Zaki, R. M., & Eid, M. M. (2021). Advanced meta -
  • [7] Khodadadi, N., Abualigah, L., El -Kenawy, E. S. M., Snasel, V., & Mirjalili, S. (2022). An archive -based multi -objective arithmetic
  • [8] Ibrahim, A., Mirjalili, S., El -Said, M., Ghoneim, S. S., Al -Harthi, M. M., Ibrahim, T. F., & El -Kenawy, E. S. M. (2021). Wind speed
  • [9] Abdelhamid, A. A., El -Kenawy, E. S. M., Alotaibi, B., Amer, G. M., Abdelkader, M. Y., Ibrahim, A., & Eid, M. M. (2022). Robust
  • [10] El-Kenawy, E. S. M., Mirjalili, S., Alassery, F., Zhang, Y. D., Eid, M. M., El -Mashad, S. Y., ... & Abdelhamid, A. A. (2022). Novel
  • [11] Hassib, E. M., El -Desouky, A. I., Labib, L. M., & El -Kenawy, E. S. M. (2020). WOA+ BRNN: An imbalanced big data classification
  • [12] Eid, M. M., El -kenawy , E. S. M., & Ibrahim, A. (2021, March). A binary sine cosine -modified whale optimization algorithm for feature
  • [13] El-Sayed Towfek, M. (2018). El -kenawy. Trust Model for Dependable File Exchange in Cloud Computing. International Journal of
  • [14] Abdelhamid, A. A., Towfek, S. K., Khodadadi, N., Alhussan, A. A., Khafaga, D. S., Eid, M. M., & Ibrahim, A. (2023). Waterwhee l plant
  • [15] El-Kenawy, E. S. M., Abdelhamid, A. A., Ibrahim, A., Mirjalili, S., Khodadad, N., Alduailij, M. A., ... & Khafaga, D. S. (2023). Al-
  • [16] Alhussan, A. A., Abdelhamid, A. A., El -Kenawy, E. S. M., Ibrahim, A., Eid, M. M., Khafaga, D. S., & Ahmed, A. E. (2023). A binary
  • [17] Alhussan, A. A., Khafaga, D. S., Abotaleb, M., Mishra, P., & El -Kenawy, E. S. M. (2024). Global Potato Production Forecasting Based
  • [18] Towfek, S. K., & Alhussan, A. A. (2024). Potato Production Forecasting Based on Balance Dynamic Biruni Earth Radius Algorithm for
  • [19] Mishra, P., Alhussan, A. A., Khafaga, D. S., Lal, P., Ray, S., Abotaleb, M., ... & El -Kenawy, E. S. M. (2024). Forecasting production of
  • [20] El-Kenawy, E. S. M., Mirjalili, S., Abdelhamid, A. A., Ibrahim, A., Khodadadi, N., & Eid, M. M. (2022). Meta -heuristic optimization
  • [21] Abdelhamid, A. A., El -Kenawy, E. S. M., Khodadadi, N., Mirjalili, S., Khafaga, D. S., Alharbi, A. H., ... & Saber, M. (2022).
  • [22] Alharbi, A. H., Towfek, S. K., Abdelhamid, A. A., Ibrahim, A., Eid, M. M., & Khafaga, D. S. & Saber, M.(2023). Diagnosis of
  • [23] Alhussan, A. A., & Towfek, S. K. (2024). 5G Resource Allocation Using Feature Selection and Greylag Goose Optimization Algori thm.
  • [24] Towfek, S. K., & Alhussan, A. A. (2024). Potato Production Forecasting Based on Balance Dynamic Biruni Earth Radius Algorithm for
  • [25] Abdelhamid, A. A., Alhussan, A. A., Qenawy, A. S. T., Osman, A. M., Elshewey, A. M., & Eed, M. (2024). Potato harvesting pred iction
  • [26] Eed, M., Alhussan, A. A., Qenawy, A. S. T., Osman, A. M., Elshewey, A. M., & Arnous, R. (2024). Potato Consumption Forecastin g
  • [27] Mahmood, S., Sun, H., Iqbal, A., Alhussan, A. A., & El -kenawy, E. S. M. (2024). Green finance, sustainable infrastructure, and green
  • [28] El-Kenawy, E. S. M., Alhussan, A. A., Khodadadi, N., Mirjalili, S., & Eid, M. M. (2024). Predicting potato crop yield with machi ne
  • [29] Radwan, M., Alhussan, A. A., Ibrahim, A., & Tawfeek, S. M. (2025). Potato leaf disease classification using optimized machine learning
  • [30] Ghasemi, M., Khodadadi, N., Trojovský, P., Li, L., Mansor, Z., Abualigah, L., ... & El -Kenawy, E. S. M. (2025). Kirchhoff’s law
  • [31] Yassen, M. A., El -Kenawy, E. S. M., Abdel -Fattah, M. G., Ismail, I., & Mostafa, H. E. D. S. (2025). Explainable artificial intelligence
  • [32] Radwan, M., Alhussan, A. A., Ibrahim, A., & Tawfeek, S. M. (2025). Potato leaf disease classification using optimized machine learning
  • [33] Mozhdehi, A. T., Khodadadi, N., Aboutalebi, M., El -Kenawy, E. S. M., Hussien, A. G., Zhao, W., ... & Mirjalili, S. (2025). Divine
  • [34] El-kenawy, E. S. M., Alhussan, A. A., Mattar, E. A., & Radwan, M. (2026). Feature selection and hyperparameter tuning in transfo rmer -
  • [35] Chen, L., Xu, C., Lim, W. H., Sharma, A., Tiang, S. S., Chong, K. S., ... & Khafaga, D. S. (2025). Transparent and reliable c onstruction
  • [36] Khodadadi, N., Towfek, S. K., Zaki, A. M., Alharbi, A. H., Khodadadi, E., Khafaga, D. S., ... & Eid, M. M. (2025). Predicting
  • [37] Alhussan, A. A., El -Kenawy, E. S. M., Khafaga, D. S., Alharbi, A. H., & Eid, M. M. (2025). Groundwater resource prediction and
  • [38] Mozhdehi, A. T., Khodadadi, N., Aboutalebi, M., El -Kenawy, E. S. M., Hussien, A. G., Zhao, W., ... & Mirjalili, S. (2025). Divine
  • [39] Lopes, J. M., Pinho, C. S., Martins, R. C., & Bandeira, T. (2026). Sustainability Gets Smarter: Competitive Pressure and AI L eading the
  • [40] Radwan, M., Ibrahim, A., Abdelsalam, M. M., Alhussan, A. A., Mattar, E. A., & El -Kenawy, E. S. M. (2026). Optimizing solar and wind
  • [41] Alhussan, A. A., El -Kenawy, E. S. M., Eid, M. M., & Khodadadi, N. (2026). Hybrid Al -Biruni and Puma Optimization (BERPO) for
  • [42] El-Kenawy, E. S. M., Ibrahim, A., Alhussan, A. A., Khafaga, D. S., Ahmed, A. E., & Eid, M. M. (2026). Smart city electricity loa d
  • [43] El-Kenawy, E. S. M., Khodadadi, N., Mirjalili, S., Zaki, A. M., Ibrahim, A., Alhussan, A. A., ... & Eid, M. M. (2026). Glider sn ake
  • [44] Mekaret, F., Rabehi, A., Zebentout, B., Tizi, S., Douara, A., Bellucci, S., ... & Alhussan, A. A. (2024). A comparative study of Schottky
  • [45] KOUADRI, Ali, RABEHI, Abdelhalim, BENZIANE, Ali, et al. A Robust Multi -Transform Watermarking Scheme for Medical Images
  • [46] Tibermacine, I. E., Russo, S., Scarano, G., Tedesco, G., Rabehi, A., Alhussan, A. A., ... & Napoli, C. (2025). Conditional VA E for
  • [47] Kouadri, A., Benziane, A., Rabehi, A., Rabehi, A., Alhussan, A. A., Khafaga, D. S., & El -Kenawy, E. S. M. (2025). A novel hybrid
  • [48] Russo, S., Tibermacine, I. E., Randieri, C., Rabehi, A., Alharbi, A. H., El -Kenawy, E. S. M., & Napoli, C. (2025). Exploiting facial
  • [49] Bentegri, H., Rabehi, M., Kherfane, S., Nahool, T. A., Rabehi, A., Guermoui, M., ... & El -Kenawy, E. S. M. (2025). Assessment of
  • [50] Tibermacine, I. E., Russo, S., Citeroni, F., Mancini, G., Rabehi, A., Alharbi, A. H., ... & Napoli, C. (2025). Adversarial de noising of EEG
  • [51] Ouahabi, M. S., Benyounes, A., Barkat, S., Ihammouchen, S., Rekioua, T., Rabehi, A., ... & Alharbi, A. H. (2025). Real -time sensor fault
  • [52] Belaid, A., Guermoui, M., Khelifi, R., Arrif, T., Chekifi, T., Rabehi, A., ... & Alhussan, A. A. (2024). Assessing suitable a reas for PV
  • [53] Mehallou, A., M’hamdi, B., Amari, A., Teguar, M., Rabehi, A., Guermoui, M., ... & Khafaga, D. S. (2025). Optimal multiobjecti ve
  • [54] Rabehi, A., El -Hadi, M., Benmahmoud, S., Rabehi, A., Alharbi, A. H., & El -Kenawy, E. S. M. (2025). SOCA -CFAR Processor in A
  • [55] Bakria, D., Beladel, A., Korich, B., Teta, A., Mohammedi, R. D., Laouid, A. A., ... & El -kenawy, E. S. (2025). A novel enhanced Grey