Application of machine learning to improve the efficiency of electrophysiological simulations used for the prediction of drug-induced ventricular arrhythmia.

ABSTRACT

BACKGROUND AND OBJECTIVE: In silico prediction of drug-induced ventricular arrhythmia often requires computationally intensive simulations, making its application tedious and non-interactive. This inconvenience can be mitigated using matrices of precomputed simulation results, allowing instantaneous computation of biomarkers such as action potential duration at 90% of the repolarisation (APD90). However, preparing such matrices can be computationally intensive for the method developers, limiting the range of simulated conditions. In this work, we aim to optimise the generation of these matrices so that they can be obtained with less effort and for a broader range of input values. METHODS: Machine learning methods were applied, building models trained with only a small fraction of the originally simulated results. The predictive performances of the models were assessed by comparing their predicted values with the actual simulation results, using percentual mean absolute error and mean relative error, as well as the percentage of data with a relative error below 5%. RESULTS: Our method obtained highly accurate estimations of the original values, leading to a nearly one hundred-fold decrease in computation time. This method also allows precomputing more complex matrices, describing the effect of more ion channels on the APD90. The best results were obtained by applying Support Vector Machine models, which yielded errors below 1% in most cases. This approach was further validated by predicting the APD90 of a set of 12 CiPA compounds and exporting the optimal settings for predicting APD90 using a different set of ion channels, always with satisfactory results. CONCLUSIONS: The proposed method effectively reduces the computational effort required to generate matrices of precomputed electrophysiological simulation values. The same approach can be applied in other fields where computationally costly simulations are applied repeatedly using slightly different input values.