EEG-MACS: Manifold Attention and Confidence Stratification for EEG-based Cross-Center Brain Disease Diagnosis under Unreliable Annotations
arxiv(2024)
摘要
Cross-center data heterogeneity and annotation unreliability significantly
challenge the intelligent diagnosis of diseases using brain signals. A notable
example is the EEG-based diagnosis of neurodegenerative diseases, which
features subtler abnormal neural dynamics typically observed in small-group
settings. To advance this area, in this work, we introduce a transferable
framework employing Manifold Attention and Confidence Stratification (MACS) to
diagnose neurodegenerative disorders based on EEG signals sourced from four
centers with unreliable annotations. The MACS framework's effectiveness stems
from these features: 1) The Augmentor generates various EEG-represented brain
variants to enrich the data space; 2) The Switcher enhances the feature space
for trusted samples and reduces overfitting on incorrectly labeled samples; 3)
The Encoder uses the Riemannian manifold and Euclidean metrics to capture
spatiotemporal variations and dynamic synchronization in EEG; 4) The Projector,
equipped with dual heads, monitors consistency across multiple brain variants
and ensures diagnostic accuracy; 5) The Stratifier adaptively stratifies
learned samples by confidence levels throughout the training process; 6)
Forward and backpropagation in MACS are constrained by confidence
stratification to stabilize the learning system amid unreliable annotations.
Our subject-independent experiments, conducted on both neurocognitive and
movement disorders using cross-center corpora, have demonstrated superior
performance compared to existing related algorithms. This work not only
improves EEG-based diagnostics for cross-center and small-setting brain
diseases but also offers insights into extending MACS techniques to other data
analyses, tackling data heterogeneity and annotation unreliability in
multimedia and multimodal content understanding.
更多查看译文
AI 理解论文
溯源树
样例
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要