Individual Prediction of Electric Field Induced by Deep-brain Magnetic Stimulation with CNN-Transformer.

Abstract

Deep-brain Magnetic Stimulation (DMS) can improve the symptoms caused by Alzheimer's disease by inducing rhythmic electric field in the deep brain, and the induced electric field is rhythm-dependent. However, calculating the induced electric field requires building a voxel model of the brain for the stimulated object, which usually takes several hours. In order to obtain the rhythm-dependent electric field induced by DMS in real time, we adopt a CNN-Transformer model to predict it. A data set with a sample size of 7350 is established for the training and testing of the model. 10-fold cross validation is used to determine the optimal hyperparameters for training CNN-Transformer. The combination of 5-layer CNN and 6-layer Transformer is verified as the optimal combination of CNN-Transformer model. The experimental results show that the CNN-Transformer model can complete the prediction in 0.731s (CPU) or 0.042s (GPU), and the overall performance metrics of prediction can reach: MAE=0.0269, RMSE=0.0420, MAPE=4.61% and R2=0.9627. The prediction performance of the CNN-Transformer model for the hippocampal electric field is better than that of the brain grey matter electric field, and the stimulation rhythm has less influence on the model performance than the coil configuration. Taking the same dataset to train and test the separate CNN model and Transformer model, it is found that CNN-Transformer has better prediction performance than the separate CNN model and Transformer model in the task of predicting electric field induced by DMS.