Simulation-Driven Machine Learning for Predicting Stent Expansion in Calcified Coronary Artery

Abstract

In this work, we integrated finite element (FE) method and machine learning (ML) method to predict the stent expansion in a calcified coronary artery. The stenting procedure was captured in a patient-specific artery model, reconstructed based on optical coherence tomography images. Following FE simulation, eight geometrical features in each of 120 cross sections in the pre-stenting artery model, as well as the corresponding post-stenting lumen area, were extracted for training and testing the ML models. A linear regression model and a support vector regression (SVR) model with three different kernels (linear, polynomial, and radial basis function kernels) were adopted in this work. Two subgroups of the eight features, i.e., stretch features and calcification features, were further assessed for the prediction capacity. The influence of the neighboring cross sections on the prediction accuracy was also investigated by averaging each feature over eight neighboring cross sections. Results showed that the SVR models provided better predictions than the linear regression model in terms of bias. In addition, the inclusion of stretch features based on mechanistic understanding could provide a better prediction, compared with the calcification features only. However, there were no statistically significant differences between neighboring cross sections and individual ones in terms of the prediction bias and range of error. The simulation-driven machine learning framework in this work could enhance the mechanistic understanding of stenting in calcified coronary artery lesions, and also pave the way toward precise prediction of stent expansion.