Badanie przeprowadzono zgodnie z Deklaracją Helsińską, a projekt został zatwierdzony przez Instytucjonalną Komisję Etyki SR University w Warangal, Telangana, Indie (numer zatwierdzenia 007/2026). Przed rozpoczęciem gromadzenia danych od wszystkich uczestników uzyskano pisemną świadomą zgodę.
Wybór uczestników
Dane EEG zebrały od 11 zdrowych uczestników (zakres wieku: 25–45 lat; średnia wieku: 28 lat) bez historii zaburzeń neurologicznych lub psychiatrycznych. W badaniu uwzględniono łącznie 1175 segmentów EEG, z czego 527 należało do klasy relaksacji, a 648 do klasy uwagi. Wykluczono uczestników, których zapisy EEG wykazywały nadmierne artefakty ruchowe lub niską jakość sygnału.
Konfiguracja akwizycji EEG
Ogólny schemat sugerowanego systemu interfejsu mózg-komputer opartego na EEG przedstawiono na Rysunku 1, który obrazuje cały proces akwizycji sygnału, klasyfikacji i zastosowania w czasie rzeczywistym. Dane EEG rejestrowano przy użyciu niewielkiego modułu akwizycji biopotencjałów połączonego z systemem opartym na mikrokontrolerze. Elektrody żelowe rozmieszczono zgodnie z międzynarodowym systemem 10–20, mianowicie w przedczołowych punktach Fp1 i Fp2, z elektrodami referencyjną i uziemiającą umieszczonymi na małżowinach usznych, jak wskazano na Rysunku 2. Sygnały były digitalizowane z częstotliwością 128 Hz i przesyłane do komputera za pomocą autorskiego skryptu akwizycyjnego napisanego w języku Python. W celu zmniejszenia impedancji elektrod i poprawy jakości sygnału zastosowano odpowiednie przygotowanie skóry oraz żel przewodzący.
Procedura eksperymentalna
Rejestracja EEG została przeprowadzona w dwóch warunkach poznawczych: relaksacji oraz uwagi, zgodnie ze zstandaryzowanym protokołem. W warunku relaksacji uczestnik siedział w komfortowym, cichym otoczeniu i otrzymał instrukcję, aby pozostać w bezruchu i ograniczyć ruchy do minimum. Przez słuchawki odtwarzano uspokajającą muzykę instrumentalną, a akwizycję EEG rozpoczęto za pomocą polecenia python collect.py. Sygnały EEG rejestrowano przez 5 min na sesję, powtarzając proces w czterech sesjach. W warunku uwagi uczestnik otrzymał polecenie wykonywania zadań wymagających koncentracji, takich jak gry interaktywne, przy zachowaniu tych samych ustawień akwizycji co w warunku relaksacji. Sygnały EEG rejestrowano przez 5 min na sesję, powtarzając proces w czterech sesjach. Każdy uczestnik brał udział w badaniu łącznie przez 40 min, na co składało się 20 min relaksacji oraz 20 min zadań wymagających uwagi, co umożliwiło wiarygodne porównanie stanów poznawczych.
Przetworzony zestaw danych został zorganizowany w następujące oznakowane segmenty: Relaksacja (0); Uwaga (1). Każdy segment trwał 4 s z 50% nakładaniem się i został przedstawiony jako spektrogram z wykorzystaniem krótkoczasowej transformaty Fouriera.
Przetwarzanie sygnałów, reprezentacja cech i implementacja modelu
Uzyskane sygnały EEG zostały poddane wstępnej obróbce w celu wyeliminowania szumów i artefaktów. Aby usunąć zakłócenia sieci elektrycznej, zastosowano filtr notch 50 Hz, natomiast do wyboru odpowiednich pasm częstotliwości EEG użyto pasmowo-przepustowego filtra Butterwortha 4. rzędu w zakresie od 0,5 do 30 Hz. Do identyfikacji segmentów zanieczyszczonych artefaktami wykorzystano progowanie amplitudy oraz inspekcję wizualną, w wyniku czego usunięto około 8–12% danych. W celu zwiększenia odporności modelu zastosowano metody augmentacji danych, takie jak dodawanie szumu gaussowskiego, maskowanie czasowe oraz maskowanie częstotliwościowe. Sygnały EEG zostały przefiltrowane, a następnie podzielone na 4-sekundowe okna z 50-procentowym nakładaniem.
Poszczególne segmenty EEG przekształcono w reprezentację czasowo-częstotliwościową (Krótkoczasowa Transformata Fouriera), zgodnie z Równaniem (1), aby umożliwić ekstrakcję charakterystyk czasowych i widmowych.
(1)
Otrzymane spektrogramy zostały znormalizowane i zmienione do stałych wymiarów, aby posłużyć jako dane wejściowe dla modelu klasyfikacji. Aby zapobiec wyciekowi danych spowodowanemu nakładaniem się próbek, zbiór danych został przed segmentacją podzielony na podzbiory treningowy (70%), walidacyjny (10%) i testowy (20%), co przedstawiono na Rysunku 3. Dodatkowo, w celu oceny generalizacji międzyosobniczej, przeprowadzono walidację krzyżową metodą leave-one-subject-out (LOSO), w której dane jednego badanego służyły do testowania, natomiast pozostałe dane wykorzystano do treningu.
Zaproponowana architektura SET-Net, przedstawiona na Rysunku 4, została zaprojektowana w celu wychwycenia wielowymiarowych cech EEG. Model integruje szkielet splotowej sieci neuronowej (CNN) do ekstrakcji reprezentacji przestrzenno-spektralnych, po którym następuje blok squeeze-and-excitation (SE), który adaptacyjnie rekalibruje istotność cech w poszczególnych kanałach, zgodnie z opisem w równaniu (2). Szkielet CNN składa się z dwóch warstw splotowych o rozmiarze jądra 3 × 3, po których następują normalizacja wsadowa (batch normalization) oraz funkcja aktywacji ELU (exponential linear unit)
(2)
Wewnętrzna struktura bloku SE została przedstawiona na Rysunku 5. W celu uchwycenia długodystansowych zależności czasowych zaimplementowano koder transformera, którego mechanizm uwagi został zdefiniowany w Równaniu (3) i zilustrowany na Rysunku 6.
(3)
Do zredukowania map cech do stałego rozmiaru 16 × 16 wykorzystano warstwę adaptacyjnego uśredniania (adaptive average pooling). Współczynnik redukcji bloku SE wynosi 8 i służy do rekalibracji odpowiedzi cech w ujęciu kanałowym. Koder transformera składał się z czterech warstw z czterema głowicami uwagi, rozmiarem wymiaru sieci typu feed-forward wynoszącym 256 oraz rozmiarem zanurzenia modelu (embedding size) wynoszącym 128.
Końcowa klasyfikacja została przeprowadzona przy użyciu warstw w pełni połączonych z regularyzacją dropout oraz funkcji aktywacji softmax w celu rozróżnienia stanów uwagi i relaksacji. Szczegółowa architektura warstw proponowanego modelu przedstawiona jest w Tabeli 1.
Model został wytrenowany przy użyciu optymalizatora AdamW z początkową prędkością uczenia 0,001 i współczynnikiem redukcji wag (weight decay) 1×10⁻4. W celu poprawy zbieżności zastosowano harmonogram cosine annealing, a trening przeprowadzono przez 50 epok przy rozmiarze partii (batch size) wynoszącym 32. Problem niezbalansowania klas rozwiązano poprzez zastosowanie ważonej funkcji straty entropii krzyżowej, zgodnie z opisem w równaniu (4).
(4)
Tabela 2 podsumowuje konfigurację treningu, a Tabela 3 przedstawia poszukiwanie hiperparametrów zaproponowanej sieci SET-Net, która została wykorzystana w klasyfikacji stanu skupienia i relaksacji na podstawie EEG. Wybrane parametry zapewniają stabilność treningu, zwiększoną generalizację oraz tolerancję na zmiany w sygnałach EEG. Wybrane parametry stabilizują proces uczenia, poprawiają zdolność do generalizacji i uodparniają model na zmiany sygnału EEG. Do pomiaru wydajności modelu wykorzystano dokładność (accuracy), precyzję (precision), czułość (recall), wynik F1 (F1-score) oraz ROC-AUC. Wszystkie potoki obejmujące przetwarzanie wstępne, trening modelu i ewaluację zostały napisane w języku Python przy użyciu frameworka PyTorch, aby zapewnić powtarzalność. Tabela 4 zawiera szczegółowy opis rozwiązywania problemów, który służy jako pomoc w diagnozowaniu i rozwiązywaniu kwestii związanych z akwizycją EEG, przetwarzaniem wstępnym, trenowaniem modelu oraz zastosowaniem w czasie rzeczywistym. Tabela 5 przedstawia zestawienie materiałów urządzenia EEG, aby umożliwić łatwe stworzenie struktury do przeprowadzania eksperymentów oraz konfiguracji doświadczalnej. Do implementacji zaproponowanego frameworka SET-Net wykorzystano język Python 3.8 wraz z frameworkiem głębokiego uczenia PyTorch. Do wstępnego przetwarzania sygnałów i ekstrakcji cech użyto standardowych bibliotek obliczeń naukowych: NumPy, SciPy i Matplotlib. Każda procedura eksperymentalna była wykonywana na systemie z procesorem Intel i7 i 16 GB RAM, aby przyspieszyć trening modeli.
DOSTĘPNOŚĆ DANYCH:
Zanonimizowany zestaw danych EEG wygenerowany i przeanalizowany podczas niniejszego badania jest publicznie dostępny w repozytorium Zenodo pod następującym linkiem dostępu: https://zenodo.org/records/19627795.
Wszystkie dane zostały zanonimizowane w celu zapewnienia poufności uczestników, zgodnie z instytucjonalnymi wytycznymi etycznymi zatwierdzonymi przez Komisję Etyki Instytucjonalnej Uniwersytetu SR (numer zatwierdzenia 007/2026).
Przetworzone informacje, ustawienia modelu oraz szczegóły implementacji niezbędne do odtworzenia wyników zostały przedstawione w manuskrypcie, a kod źródłowy jest publicznie dostępny pod adresem: https://github.com/Ravichanderj/SET-Net-EEG-BCI.