WRA-MTSI: A Robust Extended Source Imaging Algorithm Based on Multi-Trial EEG

Objective: Reconstructing brain activities from electroencephalography (EEG) signals is crucial for studying brain functions and their abnormalities. However, since EEG signals are nonstationary and vulnerable to noise, brain activities reconstructed from single-trial EEG data are often unstable, and significant variability may occur across different EEG trials even for the same cognitive task. Methods: In an effort to leverage the shared information across the EEG data of multiple trials, this paper proposes a multi-trial EEG source imaging method based on Wasserstein regularization, termed WRA-MTSI. In WRA-MTSI, Wasserstein regularization is employed to perform multi-trial source distribution similarity learning, and the structured sparsity constraint is enforced to enable accurate estimation of the source extents, locations and time series. The resulting optimization problem is solved by a computationally efficient algorithm based on the alternating direction method of multipliers (ADMM). Results: Both numerical simulations and real EEG data analysis demonstrate that WRA-MTSI outperforms existing single-trial ESI methods (e.g., wMNE, LORETA, SISSY, and SBL) in mitigating the influence of artifacts in EEG data. Moreover, WRA-MTSI yields superior performance compared to other state-of-the-art multi-trial ESI methods (e.g., group lasso, the dirty model, and MTW) in estimating source extents. Conclusion and significance: WRA-MTSI may serve as an effective robust EEG source imaging method in the presence of multi-trial noisy EEG data.