Advanced EOR screening methodology based on LightGBM and random forest: A classification problem with imbalanced data
Authors: Seyyedattar Masoud , Afshar Majid , Zendehboudi Sohrab , Butt Stephen 
Published: August 2024

In an unstable oil market with volatile prices due to various natural and geopolitical factors, it is crucial for oil‐producing companies to enhance the value of their assets by improving the recovery factors of petroleum reservoirs. Primary recovery through natural depletion or artificial lift and secondary recovery using waterflooding and immiscible gas injection typically recover no more than 10%–40% of the available reserves. A significant portion of the hydrocarbons remain unproduced if enhanced oil recovery (EOR) methods are not implemented. EOR projects are extremely costly, complex, and usually have long lead times from the decision‐making and design phases to pilot and full‐field implementations. Therefore, oil and gas operator companies need reliable insights into the best possible EOR options from the early stages of any field development planning. Since screening potential EOR choices is the first step in deciding future production scenarios, a smart EOR screening tool can add significant value by streamlining the EOR decision‐making process. In this study, we developed an EOR screening tool based on two advanced machine learning classification algorithms, random forest and light gradient boosting machine (LightGBM). These tree‐based ensemble learning classifiers were trained on an extensive dataset of 1384 worldwide EOR implementations, encompassing various reservoir conditions and reservoir rock and fluid properties as the feature space, to predict the EOR type as the class label. Considering EOR screening as a classification problem, an essential aspect of model development would be addressing the data imbalance of EOR datasets. To tackle this issue, the adaptive synthetic (ADASYN) sampling method was used to reduce classification bias by oversampling the training sets to achieve uniform class distributions. We designed an iterative model development procedure in which the classifiers were trained and tested on various training and test subsets split by stratified random sampling. For each classifier, the classification results at each iteration were used to build the confusion matrix and calculate model evaluation metrics (accuracy, precision, recall, and F1–score), which were then averaged over all independent runs to provide a fair assessment of classification performance. Moreover, binary receiver operating characteristic (ROC) curves were used to evaluate the classifier predictions and improvements obtained by oversampling. The results showed that both random forest and LightGBM classifiers made accurate class predictions, with LightGBM achieving slightly better classification performance in each modelling scenario (with or without oversampling). In both cases, the oversampling of the training dataset resulted in significant improvement of the classifiers, as evidenced by higher values of the evaluation metrics, leading to considerably more accurate EOR type predictions; specifically, oversampling boosted the prediction accuracy of the random forest model from 78.3% to 89.5% and the LightGBM model from 77.5% to 90.2%. Additionally, feature importance rankings provided valuable insights into which input variables had the greatest impact on model development.