MDH-NAS: Accelerating EEG Signal Classification With Mixed-Level Differentiable and Hardware-Aware Neural Architecture Search
In noninvasive brain-computer interfaces (BCIs), EEG analysis plays a critical role, with neural networks serving as a cornerstone for signal decoding. Existing neural network approaches for EEG signal recognition require extensive manual design and hyperparameter tuning, leading to inefficiencies and making them impractical for embedded devices due to their large model size. To address these limitations, we propose mixed-level differentiable and hardware-aware neural architecture search (MDH-NAS), a framework that automatically generates lightweight neural networks tailored for EEG classification. Unlike traditional DARTS methods, MDH-NAS employs a hybrid optimization strategy that balances global and local search spaces, thereby accelerating and refining architecture discovery. It introduces explicit size constraints during the search process to ensure deployability on embedded devices. MDH-NAS demonstrates autonomous generation of architectures for tasks such as motor imagery (MI) and depression recognition, achieving 87.80% accuracy on the BCI-IV dataset and 90.09% on the MODMA dataset. When deployed on the EAIDK-610 board across heterogeneous tasks, it attains 85.37% accuracy on the EEG Motor Movement/Imagery dataset. This method reduces architecture discovery time by 89% and enhances prediction accuracy by 8.70% compared to baseline methods, highlighting its potential for scalable EEG analysis and real-world embedded deployment.
Authors
- Lixian Zhu (ORCID: https://orcid.org/0000-0002-9842-9993)
- Fuze Tian (ORCID: https://orcid.org/0000-0001-6734-0585)
- Kai Zheng (ORCID: https://orcid.org/0000-0001-5931-0394)
- Ran Cai (ORCID: https://orcid.org/0009-0009-1821-7498)
- Jian Zhang (ORCID: https://orcid.org/0000-0003-1327-7896)
- Xiaokun Jin (ORCID: https://orcid.org/0009-0007-0399-4329)
- Bin Hu (ORCID: https://orcid.org/0000-0003-3514-5413)
- Su Wang (ORCID: https://orcid.org/0000-0001-8030-6314)
- Shuting Sun (ORCID: https://orcid.org/0000-0001-9647-8233)
Institutions
- Beijing Institute of Technology (CN)
- Lanzhou University (CN)
Publication Details
- Journal
- IEEE Internet of Things Journal
- Published
- 2025-03-24
- DOI
- https://doi.org/10.1109/jiot.2025.3553450
- Citations
- 5
- Primary Topic
- EEG and Brain-Computer Interfaces
- Type
- article
- Field-Weighted Citation Impact
- 7.16
Funders
- National Natural Science Foundation of China
- China Postdoctoral Science Foundation
- Fundamental Research Funds for the Central Universities