Population-based evolutionary search for joint hyperparameter and architecture optimization in brain-computer interface

초록

In recent years, deep learning (DL)-based models have become the de facto standard for motor imagery brain-computer interface (MI-BCI) systems due to their notable performance. However, these models often require extensive hyperparameter optimization process to achieve optimal results. To tackle this challenge, recent studies have proposed various methods to automate this process. Despite promising results, these methods overlook the architecture elements, which are crucial factors for MI-BCI system performance and are highly intertwined with hyperparameter settings. To overcome this limitation, we propose a joint optimization framework that uses a population-based evolutionary search to optimize both hyperparameters and architectures. Our framework adopts a two-stage optimization approach that alternates between hyperparameter and architecture optimization to effectively manage the complexity of the joint search process. Furthermore, we introduce a novel ensemble method that leverages diverse promising configurations to enhance generalization and robustness. Evaluations on two public MI-BCI datasets show that our framework consistently outperforms competing methods across a range of backbone models, demonstrating its effectiveness and versatility.

키워드