Attention-assisted hybrid 1D CNN-BiLSTM model for predicting electric field induced by transcranial magnetic stimulation coil

Abstract

AbstractDeep learning-based models such as deep neural network (DNN) and convolutional neural network (CNN) have recently been established as state-of-the-art for enumerating electric fields from transcranial magnetic stimulation coil. One of the main challenges related to this electric field enumeration is the prediction time and accuracy. Despite the low computational cost, the performance of the existing prediction models for electric field enumeration is quite inefficient. This study proposes a 1D CNN-based bi-directional long short-term memory (BiLSTM) model with an attention mechanism to predict electric field induced by a transcranial magnetic stimulation coil. The model employs three consecutive 1D CNN layers followed by the BiLSTM layer for extracting deep features. After that, the weights of the deep features are redistributed and integrated by the attention mechanism and a fully connected layer is utilized for the prediction. For the prediction purpose, six input features including coil turns of single wing, coil thickness, coil diameter, distance between two wings, distance between head and coil position, and angle between two wings of coil are mapped with the output of the electric field. The performance evaluation is conducted based on four verification metrics (e.g. R2, MSE, MAE, and RMSE) between the simulated data and predicted data. The results indicate that the proposed model outperforms existing DNN and CNN models in predicting the induced electrical field with R2 = 0.9992, MSE = 0.0005, MAE = 0.0188, and RMSE = 0.0228 in the testing stage.