Efficient simulation of CO₂ migration dynamics in deep saline aquifers using a multi-task deep learning technique with consistency

Abstract

CO₂ sequestration and storage in deep saline aquifers is a promising technology for mitigating the excessive concentration of the greenhouse gas in the atmosphere. However, accurately predicting the migration of CO₂ plumes requires complex multi-physics-based numerical simulation approaches, which are prohibitively expensive due to highly nonlinear coupled governing equations and uncertainties in heterogeneous spatial parameter distributions. To address this challenge, we developed an end-to-end deep learning workflow employing encoder–decoder architectures with residual network (ResNet) to efficiently predicts the spatial–temporal evolution of the solution CO₂-brine ratio (R_s) and gas saturation (S_g) – the two essential tasks for quantifying the amount of trapped CO₂ – given heterogeneous permeability fields as input. Specifically, we introduce a general multi-task learning with consistency (MTLC) framework to simultaneously predict R_s and S_g. The MTLC model leverages related tasks with less computational expensive labeled datasets to improve generalization ability. In our study, predictions for multiple tasks from the same permeability realization are not independent but expected to be consistent, as the proposed framework utilizes data-driven cross-task consistency constraints to augment learning of related tasks. Our deep learning model is trained based on physical trapping mechanisms, which play a dominant role in the CO₂ migration process. The results demonstrate that the MTLC model with joint learning yields more accurate predictions and improved generalization for predicting CO₂ migration in several test cases. Furthermore, our workflow is 10⁵ times faster than a high-fidelity physics-based numerical simulator, making it a viable alternative for field-scale applications.