eeyhsong / EEG-Conformer

EEG Transformer 2.0. i. Convolutional Transformer for EEG Decoding. ii. Novel visualization - Class Activation Topography.
GNU General Public License v3.0
371 stars 50 forks source link

Some questions about conformer #18

Open ll643989 opened 1 year ago

ll643989 commented 1 year ago

Dear author, I have used your conformer recently and its effect is really good. I would like to ask you the following questions:

  1. I feel that the fitting speed of the model is slow, and the test accuracy rate has great fluctuation. Is there any other optimization suggestion?
  2. I have been paying attention to your Transformer. Why are the CSP and spatial channel attention used in Transformer abandoned here I would appreciate it if you could answer my questions?
eeyhsong commented 1 year ago

hello @ll643989, Sorry for the late reply.

  1. I haven't performed detailed experiments for the parameters of the optimizer. To stabilize the training processing, you can try different batch sizes, learning rates, beta1, beta2, etc.
  2. CSP is an excellent way to separate non-stationary time series such as EEG. However, you know we commonly use end-to-end manner in constructing deep learning methods. So we discard the CSP module in Conformer. But I believe combining new DL methods and traditional ML strategies would be better.