Prediction of Left Ventricular Mechanics Using Machine Learning

Abstract

The goal of this paper was to provide a real-time left ventricular (LV) mechanics simulator using machine learning (ML). Finite element (FE) simulations were conducted for the LV with different material properties to obtain a training set. A hyperelastic fiber-reinforced material model was used to describe the passive behavior of the myocardium during diastole. The active behavior of the heart resulting from myofiber contractions was added to the passive tissue during systole. The active and passive properties govern the LV constitutive equation. These mechanical properties were altered using optimal Latin hypercube design of experiments to obtain training FE models with varied active properties (volume and pressure predictions) and varied passive properties (stress predictions). For prediction of LV pressures, we used eXtreme Gradient Boosting (XGboost) and Cubist, and XGBoost was used for predictions of LV pressures, volumes as well as LV stresses. The LV pressure and volume results obtained from ML were similar to FE computations. The ML results could capture the shape of LV pressure as well as LV pressure-volume loops. The results predicted by Cubist were smoother than those from XGBoost. The mean absolute errors were as follows: XGBoost volume: 1.734 ± 0.584 ml, XGBoost pressure: 1.544 ± 0.298 mmHg, Cubist volume: 1.495 ± 0.260 ml, Cubist pressure: 1.623 ± 0.191 mmHg, myofiber stress: 0.334 ± 0.228 kPa, cross myofiber stress: 0.075 ± 0.024 kPa, and shear stress: 0.050 ± 0.032 kPa. The simulation results show ML can predict LV mechanics much faster than the FE method. The ML model can be used as a tool to predict LV behavior. Training of our ML model based on a large group of subjects can improve its predictability for real world applications.