A hybrid approach of transfer learning and physics-informed modelling: Improving dissolved oxygen concentration prediction in an industrial wastewater treatment plant

Abstract

Constructing first principles models is a challenging task for nonlinear and complex systems such as a wastewater treatment unit. In recent years, data-driven models are widely used to overcome the complexity. However, they often suffer from issues such as missing, low quality or noisy data. Transfer learning is a solution for this issue where knowledge from another task is transferred to target one to increase the prediction performance. In this work, the objective is increasing the prediction performance of an industrial wastewater treatment plant by transferring the knowledge of (i) an open-source simulation model, capturing process physics, albeit with dissimilarities to the target plant, (ii) another industrial plant characterized by noisy and limited data but located in the same refinery, and (iii) constructing a physics informed transfer learning model by combining (i) and (ii). The results demonstrated that test and validation performance are improved up to 27% and 59%, respectively.