Variational autoencoder–based neural electrocardiogram synthesis trained by FEM-based heart simulator

Abstract

BackgroundFor comprehensive electrocardiogram (ECG) synthesis, a recent promising approach has been based on a heart model with physical and chemical cardiac parameters. However, the problem is that such approach requires a high-cost and limited environment using supercomputers due to the massive computation. ObjectiveThe purpose of this study is to develop an efficient method for synthesizing 12-lead ECG signals from cardiac parameters. MethodsThe proposed method is based on a variational autoencoder (VAE). The encoder and decoder of the VAE are conditioned by the cardiac parameters so that it can model the relationship between the ECG signals and the cardiac parameters. The training data are produced by a comprehensive, Finite Element Method (FEM)-based heart simulator. New ECG signals can then be synthesized by inputting the cardiac parameters into the trained VAE decoder without relying on enormous computational resources. We used two metrics to evaluate the quality of ECG signals synthesized by the proposed model. ResultsExperimental results showed that the proposed model synthesized adequate ECG signals while preserving empirically important feature points and the overall signal shapes. We also explored the optimal model by varying the number of layers and the size of latent variables in the proposed model that balances the model complexity and the simulation accuracy. ConclusionThe proposed method has the potential to become an alternative to computationally-expensive FEM-based heart simulators. It is able to synthesize ECGs from various cardiac parameters within seconds on a personal laptop computer.