Uncertainty-Aware Denoising Network for Artifact Removal in EEG Signals
The electroencephalogram (EEG) is extensively employed for detecting various brain electrical activities. Nonetheless, EEG recordings are susceptible to undesirable artifacts, resulting in misleading data analysis and even significantly impacting the interpretation of results. While previous efforts...
| Main Authors: | , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
IEEE
2023-01-01
|
| Series: | IEEE Transactions on Neural Systems and Rehabilitation Engineering |
| Subjects: | |
| Online Access: | https://ieeexplore.ieee.org/document/10312758/ |
| _version_ | 1797627792988307456 |
|---|---|
| author | Xiyuan Jin Jing Wang Lei Liu Youfang Lin |
| author_facet | Xiyuan Jin Jing Wang Lei Liu Youfang Lin |
| author_sort | Xiyuan Jin |
| collection | DOAJ |
| description | The electroencephalogram (EEG) is extensively employed for detecting various brain electrical activities. Nonetheless, EEG recordings are susceptible to undesirable artifacts, resulting in misleading data analysis and even significantly impacting the interpretation of results. While previous efforts to mitigate or reduce the impact of artifacts have achieved commendable performance, several challenges in this domain still persist: 1) due to black-box skepticism, deep-learning-based automatic EEG artifact removal methods have been impeded from being applied in clinical environments. How to support reliable denoised EEG signals with high accuracy is important; and 2) effectively exploring valuable local and global information from contaminated contexts remains challenging. On the one hand, feature extraction and aggregation in prior works are often performed blindly and assumed to be accurate, which is not always the case. On the other hand, global contextual information is gradually modeled by local fixed single-scaled convolutional filters layer by layer, which is neither efficient nor effective. To address the above challenges, we propose an Uncertainty-aware Denoising Network (UDNet) with multi-scaled pooling attention for efficient context capturing. Specifically, we predict the aleatoric and epistemic uncertainty existing during the denoising process to assist in finding and reducing the uncertain feature representation. We further propose a simple yet effective architecture to capture local and global contexts at multiple scales. Our proposed method can serve as an effective metric for identifying low-confidence epochs that warrant deferral to human experts for further inspection and assessment. Experimental results on two public datasets show that the proposed model outperforms state-of-the-art baselines. |
| first_indexed | 2024-03-11T10:30:31Z |
| format | Article |
| id | doaj.art-40bd83de348141c7b8a3cd5b1e155e6d |
| institution | Directory Open Access Journal |
| issn | 1558-0210 |
| language | English |
| last_indexed | 2024-03-11T10:30:31Z |
| publishDate | 2023-01-01 |
| publisher | IEEE |
| record_format | Article |
| series | IEEE Transactions on Neural Systems and Rehabilitation Engineering |
| spelling | doaj.art-40bd83de348141c7b8a3cd5b1e155e6d2023-11-15T00:00:05ZengIEEEIEEE Transactions on Neural Systems and Rehabilitation Engineering1558-02102023-01-01314470448010.1109/TNSRE.2023.333096310312758Uncertainty-Aware Denoising Network for Artifact Removal in EEG SignalsXiyuan Jin0https://orcid.org/0000-0002-2101-188XJing Wang1https://orcid.org/0000-0002-1017-2231Lei Liu2Youfang Lin3https://orcid.org/0000-0002-1611-4323School of Computer and Information Technology, Beijing Jiaotong University, Beijing, ChinaSchool of Computer and Information Technology, Beijing Jiaotong University, Beijing, ChinaSchool of Computer and Information Technology, Beijing Jiaotong University, Beijing, ChinaSchool of Computer and Information Technology, Beijing Jiaotong University, Beijing, ChinaThe electroencephalogram (EEG) is extensively employed for detecting various brain electrical activities. Nonetheless, EEG recordings are susceptible to undesirable artifacts, resulting in misleading data analysis and even significantly impacting the interpretation of results. While previous efforts to mitigate or reduce the impact of artifacts have achieved commendable performance, several challenges in this domain still persist: 1) due to black-box skepticism, deep-learning-based automatic EEG artifact removal methods have been impeded from being applied in clinical environments. How to support reliable denoised EEG signals with high accuracy is important; and 2) effectively exploring valuable local and global information from contaminated contexts remains challenging. On the one hand, feature extraction and aggregation in prior works are often performed blindly and assumed to be accurate, which is not always the case. On the other hand, global contextual information is gradually modeled by local fixed single-scaled convolutional filters layer by layer, which is neither efficient nor effective. To address the above challenges, we propose an Uncertainty-aware Denoising Network (UDNet) with multi-scaled pooling attention for efficient context capturing. Specifically, we predict the aleatoric and epistemic uncertainty existing during the denoising process to assist in finding and reducing the uncertain feature representation. We further propose a simple yet effective architecture to capture local and global contexts at multiple scales. Our proposed method can serve as an effective metric for identifying low-confidence epochs that warrant deferral to human experts for further inspection and assessment. Experimental results on two public datasets show that the proposed model outperforms state-of-the-art baselines.https://ieeexplore.ieee.org/document/10312758/Artifact removaldeep neural networkuncertainty estimation |
| spellingShingle | Xiyuan Jin Jing Wang Lei Liu Youfang Lin Uncertainty-Aware Denoising Network for Artifact Removal in EEG Signals IEEE Transactions on Neural Systems and Rehabilitation Engineering Artifact removal deep neural network uncertainty estimation |
| title | Uncertainty-Aware Denoising Network for Artifact Removal in EEG Signals |
| title_full | Uncertainty-Aware Denoising Network for Artifact Removal in EEG Signals |
| title_fullStr | Uncertainty-Aware Denoising Network for Artifact Removal in EEG Signals |
| title_full_unstemmed | Uncertainty-Aware Denoising Network for Artifact Removal in EEG Signals |
| title_short | Uncertainty-Aware Denoising Network for Artifact Removal in EEG Signals |
| title_sort | uncertainty aware denoising network for artifact removal in eeg signals |
| topic | Artifact removal deep neural network uncertainty estimation |
| url | https://ieeexplore.ieee.org/document/10312758/ |
| work_keys_str_mv | AT xiyuanjin uncertaintyawaredenoisingnetworkforartifactremovalineegsignals AT jingwang uncertaintyawaredenoisingnetworkforartifactremovalineegsignals AT leiliu uncertaintyawaredenoisingnetworkforartifactremovalineegsignals AT youfanglin uncertaintyawaredenoisingnetworkforartifactremovalineegsignals |