BridgeMIL Decouples EEG Instance Learning for Better Disease Diagnosis

Zhiyuan Ma, Zeyuan Li, Zhiyi Lu, Jiacheng Hao, Youlang Du, Zhen Jiang, Xinche Zhang, Yuhao Sun, Sen Song· July 31, 2026 View original

Key takeaways

  • Traditional EEG diagnosis methods suffer from unreliable instance-level labeling.
  • BridgeMIL decouples instance representation learning from subject-level supervision, improving accuracy.
  • The two-stage framework uses self-supervised pretraining and attention-based MIL.
  • This approach leads to more structured representations and better diagnostic separation between classes.

Who benefits

HealthcareMedical DevicesPharmaceuticalsResearch & AcademiaBiotechnology

Summary

Researchers propose BridgeMIL, a two-stage framework that decouples instance representation learning from subject-level supervision in EEG-based disease diagnosis. This method significantly improves diagnostic accuracy by addressing the limitations of traditional instance-level labeling and end-to-end Multiple Instance Learning (MIL).

Traditional EEG-based disease diagnosis often involves segmenting long recordings into short instances and then assigning the subject's overall diagnosis label to each instance. This approach assumes all instances provide equally reliable diagnostic evidence, which is often not true. While Multiple Instance Learning (MIL) avoids this by treating each subject as a "bag" of instances, the scarcity of subjects compared to instances in EEG datasets can limit the quality of learned representations in end-to-end MIL systems. To overcome these limitations, a new two-stage framework called BridgeMIL has been developed. In its first stage, BridgeMIL pretrains an encoder without relying on inherited instance labels. Instead, it aligns temporally nearby windows and independently sampled within-subject sub-bags, using variance and covariance regularization to prevent collapse and reduce redundancy. The second stage transfers this pretrained encoder to an attention-based MIL aggregator, applying supervision only to the subject-level predictions. Feature retention mechanisms are used to limit representation drift. Experiments across three EEG disease datasets and five backbones show BridgeMIL achieving significantly higher mean accuracy, outperforming strong baselines. This highlights the importance of aligning supervision with the subject-level objective while effectively learning from abundant EEG instances without mislabeling them.

Why it matters

For professionals in healthcare AI and medical diagnostics, improving the accuracy and reliability of EEG-based disease detection is critical. BridgeMIL offers a more robust and effective approach to leveraging complex EEG data, potentially leading to earlier and more precise diagnoses.

How to implement this in your domain

  1. 1Adopt a two-stage training approach for EEG-based diagnostic models, separating instance representation learning from subject-level supervision.
  2. 2Implement self-supervised pretraining techniques in Stage 1, focusing on temporal and within-subject sub-bag alignment for EEG data.
  3. 3Integrate an attention-based Multiple Instance Learning (MIL) aggregator in Stage 2, applying supervision solely at the subject level.
  4. 4Evaluate BridgeMIL's performance on specific EEG datasets for diseases like epilepsy or Alzheimer's to validate its diagnostic improvements.

Original post by Zhiyuan Ma, Zeyuan Li, Zhiyi Lu, Jiacheng Hao, Youlang Du, Zhen Jiang, Xinche Zhang, Yuhao Sun, Sen Song

"arXiv:2607.27274v1 Announce Type: new Abstract: EEG-based disease diagnosis requires one prediction per subject, yet common pipelines segment recordings into short instances, inherit the subject label for every instance, and train instance-level classifiers. This assumes that all…"

View on X

Originally posted by Zhiyuan Ma, Zeyuan Li, Zhiyi Lu, Jiacheng Hao, Youlang Du, Zhen Jiang, Xinche Zhang, Yuhao Sun, Sen Song on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses

More in AI Research

AI ResearchAI Engineering & DevTools

New Framework Improves Partial Multi-View Clustering Performance.

DAS-PMVC is a novel framework for partial multi-view clustering that addresses view asymmetry and irrelevant samples by leveraging dual alignment and structure enhancement. It uses anchor graph structure alignment, structure-enhanced feature learning, and a dual alignment strategy to achieve superior clustering performance on various datasets.

Shubin Ma, Liang Zhao, Chuanye He, Zhenjiao Liu, Liang Zou, Lin Yuanbo Wu, Yu ShaoJul 31, 2026
AI Engineering & DevToolsAI Research

Dual Teachers Improve Adversarial Robustness and Accuracy.

This work extends Information Bottleneck Distillation (IBD) by introducing a "clean teacher" alongside a robust teacher to improve the robustness/accuracy tradeoff against adversarial attacks. The proposed method transfers features from both teachers to a student model, achieving better clean accuracy while maintaining adversarial robustness, outperforming original IBD and competing with state-of-the-art approaches.

Vincent Ryusuke Takahashi, Yoshinari Takeishi, Jun'ichi Takeuchi, Kave SalamatianJul 31, 2026
AI Engineering & DevToolsAI Research

Dynamic Batch Sizes Improve Large Language Model Training Efficiency.

This paper proposes a new approach to deep learning dynamics, deriving joint scaling laws for loss based on both learning rate and batch size schedules. It introduces an optimal dynamic batch size schedule that consistently outperforms static batch size baselines, highlighting its importance for large language model training.

Jiaxiang Li, Zhiqi Bu, Shiyun XuJul 31, 2026