|
EEG Feature Learning Model Based on Intrinsic Time-Scale Decomposition and Adaptive Huber Loss
YANG Li-jun, JIANG Shu-yue, WEI Xiao-ge , XIAO Yun-hai
Chinese Quarterly Journal of Mathematics
2022, 37 (3):
281-300.
DOI: 10.13371/j.cnki.chin.q.j.m.2022.03.006
According to the World Health Organization, about 50 million people world-
wide suffer from epilepsy. The detection and treatment of epilepsy face great challenges.
Electroencephalogram (EEG) is a significant research object widely used in diagnosis and
treatment of epilepsy. In this paper, an adaptive feature learning model for EEG signals
is proposed, which combines Huber loss function with adaptive weight penalty term.
Firstly, each EEG signal is decomposed by intrinsic time-scale decomposition. Secondly,
the statistical index values are calculated from the instantaneous amplitude and frequency
of every component and fed into the proposed model. Finally, the discriminative features
learned by the proposed model are used to detect seizures. Our main innovation is to
consider a highly flexible penalization based on Huber loss function, which can set different
weights according to the influence of different features on epilepsy detection. Besides, the
new model can be solved by proximal alternating direction multiplier method, which can
effectively ensure the convergence of the algorithm. The performance of the proposed
method is evaluated on three public EEG datasets provided by the Bonn University,
Childrens Hospital Boston-Massachusetts Institute of Technology, and Neurological and
Sleep Center at Hauz Khas, New Delhi(New Delhi Epilepsy data). The recognition
accuracy on these two datasets is 98% and 99.05%, respectively, indicating the application
value of the new model.
Related Articles |
Metrics
|
|