mlsanityData to 14 Oct 2026

Do Masked Autoencoders Improve Downhole Prediction? An Empirical Study on Real Well Drilling Data

arXiv 2604.20909 ↗First posted 21 Apr 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Downhole drilling telemetry presents a fundamental labeling asymmetry: surface sensor data are generated continuously at 1~Hz, while labeled downhole measurements are costly, intermittent, and scarce. Current machine learning approaches for downhole metric prediction universally adopt fully supervised training from scratch, which is poorly suited to this data regime. We present the first empirical evaluation of masked autoencoder (MAE) pretraining for downhole drilling metric prediction. Using two publicly available Utah FORGE geothermal wells comprising approximately 3.5 million timesteps of multivariate drilling telemetry, we conduct a systematic full-factorial design space search across 72 MAE configurations and compare them against supervised LSTM and GRU baselines on the task of predicting Total Mud Volume. Results show that the best MAE configuration reduces test mean absolute error by 19.8% relative to the supervised GRU baseline, while trailing the supervised LSTM baseline by 6.4%. Analysis of design dimensions reveals that latent space width is the dominant architectural choice (Pearson r = -0.59 with test MAE), while masking ratio has negligible effect, an unexpected finding attributed to high temporal redundancy in 1~Hz drilling data. These results establish MAE pretraining as a viable paradigm for drilling analytics and identify the conditions under which it is most beneficial.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 6 of its 24 references that are in the map. First posted on arXiv 21 Apr 2026.

Builds on

6 of 24 references.

References · 6

  1. 2014Adam: A Method for Stochastic Optimizationcited by 37,240
  2. 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,403
  3. 2019Parameter-Efficient Transfer Learning for NLPcited by 4,236
  4. 2014Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translationcited by 4,057
  5. 2023Ti-MAE: Self-Supervised Masked Time Series Autoencoderscited by 63
  6. 2022MTSMAE: Masked Autoencoders for Multivariate Time-Series Forecastingcited by 9

Built on by

Not cited yet.