AI+X ElectivesResearch projectEnergy and environment

Can well injection and production data forecast earthquakes in the Los Angeles Basin?

Working alone, Qi Hui Choy merged well, injection, production and earthquake data for the Los Angeles Basin on a spatial grid, then trained XGBoost models to forecast earthquake rate and average magnitude 3 months, 1 year and 5 years ahead. Accuracy rose with the horizon: for earthquake rate, R² went from 0.367 at 3 months to 0.842 at 5 years.

ConcentrationAI+X Electives
FormatPBL poster
When2025 · Individual project
Project LeadResearcher in Geophysics and Subsurface Modelling
SpatialGridwells, injection, quakes1 · Merge dataTimeFeaturesrolling averages, ratios2 · Engineer featuresRate R²0.842XGBoost, 3 horizons3 · Forecast
Illustration of the project pipeline, drawn from the poster.
0.842R² for the 5-year earthquake-rate forecast
0.367 → 0.842Rate R² from 3-month to 5-year horizon
0.649R² for the 5-year average-magnitude forecast
3Forecast horizons
Abstract

Wastewater injection and fluid production can induce earthquakes. This project asked whether operational data can forecast earthquake rate and average magnitude in the Los Angeles Basin. Well, injection, production and earthquake datasets were cleaned and merged, and coordinates were converted into grid cells for spatial analysis. Features included rolling 3-month, 1-year and 5-year averages, lagged values, cumulative injected and produced water, oil and gas, and ratios between them. Separate XGBoost regression models were trained for each horizon. For earthquake rate, RMSE fell from 0.792 to 0.464 to 0.315 and R² rose from 0.367 to 0.647 to 0.842 across the 3-month, 1-year and 5-year forecasts. For average magnitude, RMSE fell from 0.072 to 0.030 to 0.008, with the 5-year model reaching R² 0.649. Cumulative gas injection and production and total water injected mattered most for rate; earthquake depth, water injection and recent magnitude history mattered most for magnitude. Forecasts were mapped across the basin.

Keywords · induced seismicity · XGBoost · geospatial analysis · feature engineering · Los Angeles Basin · forecasting

The team

Who did this work

Qi Hui Choy
Individual project · Qi Hui Choy
Qi Hui Choy
Qi Hui Choy
National University of Singapore · Singapore
On this project
Scoped and ran the whole project alone while most of the cohort worked in teams.

Students named here agreed to be featured. Role notes come from the Project Lead's feedback.

Background

1Human activity and earthquakes

Studies have linked cumulative injection volumes to more frequent earthquakes. If operational data carries that signal, forecasts could help agencies spot high-risk zones.

The project built on two earlier machine-learning studies of induced seismicity and extended the approach with XGBoost and spatial maps of the Los Angeles Basin.

Method

2From raw well data to spatial forecasts

  1. Merged the data. Cleaned and merged well, injection, production and earthquake datasets; converted coordinates to grid cells and polygons.
  2. Engineered features. Rolling 3-month, 1-year and 5-year averages and lags; total injected and produced water, oil and gas; oil–water, gas–water and gas–oil ratios.
  3. Trained per horizon. Separate XGBoost regression models for 3-month, 1-year and 5-year forecasts of rate and average magnitude.
  4. Mapped the results. Plotted well density, injection and production, and predicted rates across the basin.
Results

3Longer horizons forecast better

HorizonRate RMSERate R²Magnitude RMSE
3 months0.7920.3670.072
1 year0.4640.6470.030
5 years0.3150.8420.008
Table 1. Model performance by forecast horizon. The 5-year magnitude model reached R² 0.649. From the poster.
  • Earthquake rate: cumulative gas injected and produced, seasonal patterns and total water injected were the strongest drivers.
  • Average magnitude: earthquake depth, water injection volumes and recent magnitude history were the strongest drivers.
From the poster

4The poster

The final poster.

Poster: predicting earthquake rate and average magnitude using XGBoost
Figure 1. The poster: method, maps, model performance and feature importance.
Limitations

5What this does not show

  • Feature importance shows association, not cause.
  • Longer windows average out short-term noise, so higher R² does not mean individual earthquakes can be predicted.
  • The poster does not describe how the test period was separated from training data.
Skills and fit

6What this work shows

Geospatial data processingFeature engineeringGradient boosting (XGBoost)Time-series forecastingData visualizationIndependent research
Project Lead
PL

Researcher in Geophysics and Subsurface Modelling

Researcher in earthquake hazards at MIT Civil and Environmental Engineering and Harvard Earth and Planetary Sciences, with industry experience in energy-related subsurface work.

The student did the analysis and modeling. The Project Lead guided the work and reviewed it.

Good fit for applications in

Environmental data science · Energy · Geoscience

An independent project with clear metrics and maps shows research readiness.

Project Lead review

7What the Project Lead observed

The Project Lead wrote feedback on the project at the end of the program.

Rating

Written feedback only

This cohort ran before the four-level capability rating was recorded, so this page shows the Project Lead's written feedback without a level.

Initiative
Scoped, executed and advanced a complex modeling project entirely alone, while most peers worked in teams.
Quality
Delivered high-quality results; the Project Lead placed the work in the top 10% of the PBL cohort.
Work habits
The Academic Coordinator noted her intellectual rigor, organization and clear ownership.

Source: the Project Lead's and Academic Team's written feedback, 2025.

References

References

  1. Chen, T. & Guestrin, C. (2016). XGBoost: A scalable tree boosting system. KDD.
  2. Amini et al. (2024), comparing geological and operational predictors of induced seismicity with machine learning (as cited in the poster).
  3. Qin et al. (2022), Random Forest forecasting of seismicity rates across time windows (as cited in the poster).
Choy, Q. H. (2025). Predicting earthquake rate and average magnitude using XGBoost. AI for Energy and Sustainability – Shell Project, PBL poster. blendedlearn.org/outcome-earthquake-forecasting-xgboost
Build work like this

Join the AI for Energy and Sustainability – Shell Project

Work with a geophysics researcher on energy and environmental data and leave with research you can show.