The Czech Academy of Sciences, Institute of Scientific Instruments, Brno, Czech Republic.
Department of Radioelectronics, Brno University of Technology, Brno, Czech Republic.
J Neural Eng. 2023 Jun 16;20(3). doi: 10.1088/1741-2552/acdc54.
The current practices of designing neural networks rely heavily on subjective judgment and heuristic steps, often dictated by the level of expertise possessed by architecture designers. To alleviate these challenges and streamline the design process, we propose an automatic method, a novel approach to enhance the optimization of neural network architectures for processing intracranial electroencephalogram (iEEG) data.We present a genetic algorithm, which optimizes neural network architecture and signal pre-processing parameters for iEEG classification.Our method improved the macro1 score of the state-of-the-art model in two independent datasets, from St. Anne's University Hospital (Brno, Czech Republic) and Mayo Clinic (Rochester, MN, USA), from 0.9076 to 0.9673 and from 0.9222 to 0.9400 respectively.By incorporating principles of evolutionary optimization, our approach reduces the reliance on human intuition and empirical guesswork in architecture design, thus promoting more efficient and effective neural network models. The proposed method achieved significantly improved results when compared to the state-of-the-art benchmark model (McNemar's test,≪ 0.01). The results indicate that neural network architectures designed through machine-based optimization outperform those crafted using the subjective heuristic approach of a human expert. Furthermore, we show that well-designed data preprocessing significantly affects the models' performance.
目前设计神经网络的方法主要依赖于主观判断和启发式步骤,这些往往受到架构设计师专业水平的限制。为了缓解这些挑战并简化设计过程,我们提出了一种自动方法,即一种新颖的方法,用于优化用于处理颅内脑电图 (iEEG) 数据的神经网络架构。我们提出了一种遗传算法,用于优化神经网络架构和 iEEG 分类的信号预处理参数。我们的方法将两个独立数据集(捷克共和国布尔诺的圣安妮大学医院和美国明尼苏达州罗切斯特的梅奥诊所)中最先进模型的宏 1 分数从 0.9076 提高到 0.9673,从 0.9222 提高到 0.9400。通过结合进化优化原则,我们的方法减少了对架构设计中人类直觉和经验猜测的依赖,从而促进了更高效和有效的神经网络模型。与最先进的基准模型相比,所提出的方法取得了显著的改进结果(McNemar 检验,≪0.01)。结果表明,通过基于机器的优化设计的神经网络架构优于人类专家的主观启发式方法设计的架构。此外,我们表明精心设计的数据预处理对模型的性能有显著影响。