AI Models Predict Undrained Shear Strength with Incomplete Data

Haibin Xiong, Shaoheng Dai, Peng Lan, Xuzhen He, Chenxi Tong, Sheng Zhang, Daichao Sheng· August 17, 2026 View original

Key takeaways

  • New probabilistic AI models accurately predict undrained shear strength despite missing data.
  • Advanced imputation methods like Miss Forest and MICE are crucial for handling data gaps.
  • A Multi-Head Attention-based Probabilistic Neural Network (MHA-PNN) enhances prediction.
  • The MN-enhanced MHA-PNN model significantly improves accuracy and uncertainty quantification.

Who benefits

Civil EngineeringConstructionGeotechnical EngineeringMiningInfrastructure Development

Summary

This study develops probabilistic indirect models to predict undrained shear strength (su) using Atterberg limits and CPTU measurements, specifically addressing high missing data rates and variability. It evaluates advanced imputation methods and integrates a multi-head attention mechanism into an ANN for enhanced prediction accuracy and uncertainty quantification.

Researchers have developed advanced probabilistic indirect models to accurately predict undrained shear strength (su), a critical parameter in geotechnical design. This is particularly challenging due to the high rates of missing data and inherent variability in traditional empirical methods. The study leverages a global database (CLAY/10/7490) and focuses on predicting su from Atterberg limits and piezocone cone penetration (CPTU) measurements. The methodology involves a two-step process. First, three imputation methods—multivariate normal (MN), multiple imputation by chained equations (MICE), and miss forest (MF)—were tested to handle missing values, with their effectiveness validated using a Probabilistic Extreme Gradient Boosting (PXGB) model. Second, the indirect prediction model, named MHA-PNN, integrates a multi-head attention (MHA) mechanism into an artificial neural network (ANN) to improve information extraction from limited data. The MN-enhanced MHA-PNN model significantly outperformed other models in both prediction accuracy and uncertainty quantification, demonstrating its potential for robust geotechnical applications with sparse datasets.

Why it matters

Geotechnical engineers and construction professionals can leverage these advanced AI models to make more accurate and reliable predictions of soil properties, even with incomplete data. This can lead to safer, more efficient, and cost-effective infrastructure projects by reducing uncertainties in design.

How to implement this in your domain

  1. 1Assess existing geotechnical datasets for missing values and variability that could benefit from advanced imputation.
  2. 2Explore the integration of multi-head attention mechanisms into neural networks for similar prediction tasks.
  3. 3Pilot the MHA-PNN model for predicting undrained shear strength in a specific project.
  4. 4Train and validate the model using available data, comparing its performance against traditional methods.

Original post by Haibin Xiong, Shaoheng Dai, Peng Lan, Xuzhen He, Chenxi Tong, Sheng Zhang, Daichao Sheng

"arXiv:2608.13934v1 Announce Type: new Abstract: Accurate prediction of undrained shear strength (su) is crucial for geotechnical design, but is often hampered by substantial uncertainty in traditional empirical methods. This study uses the CLAY/10/7490 global database to develop…"

View on X

Originally posted by Haibin Xiong, Shaoheng Dai, Peng Lan, Xuzhen He, Chenxi Tong, Sheng Zhang, Daichao Sheng on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses