A machine learning model to estimate myocardial stiffness from EDPVR

Abstract

Abstract In-vivo estimation of mechanical properties of the myocardium is essential for patient-specific diagnosis and prognosis of cardiac disease involving myocardial remodeling, including myocardial infarction and heart failure with preserved ejection fraction. Current approaches use time-consuming finite-element (FE) inverse methods that involve reconstructing and meshing the heart geometry, imposing measured loading, and conducting computationally expensive iterative FE simulations. In this paper, we propose a machine learning (ML) model that feasibly and accurately predicts passive myocardial properties directly from select geometric, architectural, and hemodynamic measures, thus bypassing exhaustive steps commonly required in cardiac FE inverse problems. Geometric and fiber-orientation features were chosen to be readily obtainable from standard cardiac imaging protocols. The end-diastolic pressure-volume relationship (EDPVR), which can be obtained using a single-point pressure-volume measurement, was used as a hemodynamic (loading) feature. A comprehensive ML training dataset in the geometry-architecture-loading space was generated, including a wide variety of partially synthesized rodent heart geometry and myofiber helicity possibilities, and a broad range of EDPVRs obtained using forward FE simulations. Latin hypercube sampling was used to create 2500 examples for training, validation, and testing. A multi-layer feed-forward neural network (MFNN) was used as a deep learning agent to train the ML model. The model showed excellent performance in predicting stiffness parameters $$a_f$$ a f and $$b_f$$ b f associated with fiber direction ( $$R^2_{a_f}=99.471\%$$ R a f 2 = 99.471 % and $$R^2_{b_f}=92.837\%$$ R b f 2 = 92.837 % ). After conducting permutation feature importance analysis, the ML performance further improved for $$b_f$$ b f ( $$R^2_{b_f}=96.240\%$$ R b f 2 = 96.240 % ), and the left ventricular volume and endocardial area were found to be the most critical geometric features for accurate predictions. The ML model predictions were evaluated further in two cases: (i) rat-specific stiffness data measured using ex-vivo mechanical testing, and (ii) patient-specific estimation using FE inverse modeling. Excellent agreements with ML predictions were found for both cases. The trained ML model offers a feasible technology to estimate patient-specific myocardial properties, thus, bridging the gap between EDPVR, as a confounded organ-level metric for tissue stiffness, and intrinsic tissue-level properties. These properties provide incremental information relative to traditional organ-level indices for cardiac function, improving the clinical assessment and prognosis of cardiac diseases.