BrainWave: A Brain Signal Foundation Model for Clinical Applications

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Yuan, Zhizhang, Shen, Fanqi, Li, Meng, Yu, Yuguo, Tan, Chenhao, Yang, Yang
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866908510735826944
author Yuan, Zhizhang
Shen, Fanqi
Li, Meng
Yu, Yuguo
Tan, Chenhao
Yang, Yang
author_facet Yuan, Zhizhang
Shen, Fanqi
Li, Meng
Yu, Yuguo
Tan, Chenhao
Yang, Yang
contents Neural electrical activity is fundamental to brain function, underlying a range of cognitive and behavioral processes, including movement, perception, decision-making, and consciousness. Abnormal patterns of neural signaling often indicate the presence of underlying brain diseases. The variability among individuals, the diverse array of clinical symptoms from various brain disorders, and the limited availability of diagnostic classifications, have posed significant barriers to formulating reliable model of neural signals for diverse application contexts. Here, we present BrainWave, the first foundation model for both invasive and non-invasive neural recordings, pretrained on more than 40,000 hours of electrical brain recordings (13.79 TB of data) from approximately 16,000 individuals. Our analysis show that BrainWave outperforms all other competing models and consistently achieves state-of-the-art performance in the diagnosis and identification of neurological disorders. We also demonstrate robust capabilities of BrainWave in enabling zero-shot transfer learning across varying recording conditions and brain diseases, as well as few-shot classification without fine-tuning, suggesting that BrainWave learns highly generalizable representations of neural signals. We hence believe that open-sourcing BrainWave will facilitate a wide range of clinical applications in medicine, paving the way for AI-driven approaches to investigate brain disorders and advance neuroscience research.
format Preprint
id arxiv_https___arxiv_org_abs_2402_10251
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle BrainWave: A Brain Signal Foundation Model for Clinical Applications
Yuan, Zhizhang
Shen, Fanqi
Li, Meng
Yu, Yuguo
Tan, Chenhao
Yang, Yang
Neurons and Cognition
Artificial Intelligence
Machine Learning
Signal Processing
Neural electrical activity is fundamental to brain function, underlying a range of cognitive and behavioral processes, including movement, perception, decision-making, and consciousness. Abnormal patterns of neural signaling often indicate the presence of underlying brain diseases. The variability among individuals, the diverse array of clinical symptoms from various brain disorders, and the limited availability of diagnostic classifications, have posed significant barriers to formulating reliable model of neural signals for diverse application contexts. Here, we present BrainWave, the first foundation model for both invasive and non-invasive neural recordings, pretrained on more than 40,000 hours of electrical brain recordings (13.79 TB of data) from approximately 16,000 individuals. Our analysis show that BrainWave outperforms all other competing models and consistently achieves state-of-the-art performance in the diagnosis and identification of neurological disorders. We also demonstrate robust capabilities of BrainWave in enabling zero-shot transfer learning across varying recording conditions and brain diseases, as well as few-shot classification without fine-tuning, suggesting that BrainWave learns highly generalizable representations of neural signals. We hence believe that open-sourcing BrainWave will facilitate a wide range of clinical applications in medicine, paving the way for AI-driven approaches to investigate brain disorders and advance neuroscience research.
title BrainWave: A Brain Signal Foundation Model for Clinical Applications
topic Neurons and Cognition
Artificial Intelligence
Machine Learning
Signal Processing
url https://arxiv.org/abs/2402.10251