← Back to Machine Learning
cs.LG

A transformer learns subsurface heat from sparse borehole data

Daniel O'Malley, Christopher W. Johnson, Javier E. Santos, Pablo Lara, Sandro Malusà, Bharat Srikishan, John Kath, Arnab Mazumder, Mohamed Mehana, David Coblentz, Nathan DeBardeleben, Earl Lawrence, Hari Viswanathan

May 15, 2026

Continental geothermal mapping relies on sparse, expensive borehole data, but conventional interpolation and physics-based models struggle with sharp thermal anomalies. In-Context Earth, a transformer, uses sparse local observations as context to predict continuous temperature-at-depth fields across the contiguous United States with 4.7°C mean absolute error—beating the physics-informed Stanford Thermal Model, AlphaEarth embeddings, and kriging. Without retraining, the model generalizes to Alberta, Australia, and the UK using only 20 observations at test time, achieving 2.2–6.2°C error. Uncertainty estimates are well calibrated, and interpretability analysis reveals the model learns unobserved subsurface properties (seismic velocity, geochemistry, crustal structure) and applies them consistently. Code and model weights are not mentioned in the abstract.
Published as In-context learning enables continental-scale subsurface temperature prediction from sparse local observations arXiv:2605.16665
Read the original paper →