Protokół ten opisuje krok po kroku implementację modelu Hybrid Deep Reinforcement Learning (HDRL) dla multimodalnej fuzji sygnałów biomedycznych. Ramy integrują techniki uczenia głębokiego i wzmacnianego, aby zoptymalizować łączenie danych EKG, EEG oraz obrazowania medycznego w czasie rzeczywistym. Sekcja opisuje również procedury ekstrakcjonowania cech, treningu i przygotowania danych stosowane w proponowanym modelu.
Materiały
Wszystkie eksperymenty prowadzono na stacji roboczej z wielordzeniowym procesorem, ≥32 GB RAM i dedykowaną kartą graficzną (≥8 GB VRAM). Środowisko implementacyjne składało się z Pythona (wersja 3.9 lub nowsza), TensorFlow (wersja 2.12) oraz Keras (wersja 2.12). Biblioteki uczenia ze wzmocnieniem zostały zainstalowane przy użyciu pipe-install stable-baselines3. Pakiet Stable-Baselines318 zapewnia efektywne implementacje algorytmów uczenia ze wzmocnieniem, takich jak Deep Q-Network (DQN) i Proximal Policy Optimisation (PPO).
Zbiory danych
Proponowany framework Hybrid Deep Reinforcement Learning (HDRL) został oceniony przy użyciu trzech publicznie dostępnych zbiorów danych biomedycznych reprezentujących typy sygnałów multimodalnych. (1) PhysioNet Challenge 2016: Zbiór 12-odprowadzeniowych nagrań EKG pacjentów z chorobami serca, w tym migotaniem przedsionków, arytmią komorową i chorobą niedokrwienną serca. Zbiór danych zawiera anotowane etykiety dla schorzeń serca19. (2) Baza danych EEG na skalpie CHB-MIT: Do wykrywania napadów padaczkowych wykorzystywany jest zbiór danych nagrań EEG na głowie, z adnotacjami oznaczającymi zdarzenia napadów i nienapadów20. (3) Baza danych MRI OASIS-3: Zbiór obrazów mózgu MRI oraz ocen klinicznych służy do wykrywania i monitorowania choroby Alzheimera21. Reprezentatywne próbki z każdej modalności wykazują wrodzoną zmienność i charakterystykę szumu obecną w multimodalnych danych biomedycznych (Rysunek 2).

Rysunek 2: Próbka sygnałów biomedycznych z EKG, EEG i obrazowania medycznego. Reprezentatywne przykłady przebiegu EKG, sygnału EEG oraz badania obrazowego medycznego ilustrujące różne metody stosowane w badaniu. Proszę kliknąć tutaj, aby zobaczyć większą wersję tej figurki.
Wstępne przetwarzanie danych
Każda modalność przeszła procedury wstępnego przetwarzania specyficzne dla modalności, aby zapewnić standaryzowaną reprezentację wejściową. Sygnały EEG i EKG były filtrowane za pomocą filtra pasmowego Butterwortha czwartego rzędu. Zakres częstotliwości ustawiono na 0,5–40 Hz dla EEG oraz 0,5–45 Hz dla EKG. Próbki zostały podzielone na 5-sekundowe okna z 50% nakładaniem się. Każdy sygnał był normalizowany przez segmentowanie, aby uzyskać zerową średnią i jednostkową wariancję. Obrazy MRI były przetwarzane za pomocą usuwania czaszki za pomocą narzędzia FSL Brain Extraction Tool (BET). Obrazy zostały zmniejszone do 224 × 224 pikseli. Intensywność pikseli została znormalizowana do zakresu [0,1], aby ustandaryzować rozkłady wejściowe.
Ekstrakcja cech
Ekstrakcja cech była przeprowadzana niezależnie dla każdej modalności za pomocą DNN, co zapewniało skuteczne uchwycenie unikalnych cech każdej modalności.
Ekstraktory cech EKG i EEG:
Do przetwarzania sygnałów szeregowych w czasie zastosowano architekturę 1D Convolucional Neural Network (1D-CNN). Architektura składała się z (1) warstwy wejściowej (window_length, 1), (2) warstw Conv1D z 64 filtrami i rozmiarem jądra 3, (3) aktywacji ReLU, (4) MaxPooling1D z rozmiarem puli=2, (5) warstw Flatten i Dense (128 jednostek, ReLU). Przedostatnie reprezentacje warstw gęstych zostały wyodrębnione jako wektory cech specyficzne dla modalności.
Ekstraktor cech MRI
Prewytrenowany dwuwymiarowy model CNN (np. VGG16 lub ResNet50) został dostrojony do wydobywania cech przestrzennych z obrazów MRI. CNN został wytrenowany na zbiorze danych OASIS-3, a ostatnie warstwy modelu wcześniej wytrenowanego zostały zmodyfikowane, aby dostosować się do konkretnego zadania klasyfikacji21. Ostateczne warstwy klasyfikacji zostały zastąpione warstwami specyficznymi dla zadań, a z przedostatniej warstwy 5,6 wyodrębniono cechy przestrzenne wysokiego poziomu.
Architektura modelu i trening
Proponowany framework Hybrid Deep Reinforcement Learning (HDRL) został opracowany i wytrenowany w dwóch kolejnych etapach: ekstrakcja cech specyficznych dla modalności, a następnie optymalizacja fuzji oparta na uczeniu ze wzmocnieniem. W pierwszym etapie trenowano oddzielne sieci konwolucyjne neuronowe (CNN) na znakowanych danych dla modalności EKG, EEG i MRI. Podczas szkolenia nadzorowanego stosowano optymalizator Adaptive Moment Estimation (Adam) o szybkości uczenia się 0,001 i kategorycznej utracie entropii krzyżowej. Po zbieżności wyodrębniano wysokopoziomowe reprezentacje z przedostatniej warstwy każdego CNN, dając wektory cech specyficzne dla modalności: F EKG =CNN ECG(XECG), FECG = CNNECG(XECG) oraz FMRI = CNNMRI(XMRI). Wektory te zostały połączone, tworząc jednolitą wielomodalną reprezentację Fconcat = [FEKG, FEEG,F MRI], która służy jako wejście do etapu uczenia ze wzmocnieniem.
Proces fuzji został sformułowany jako Proces Decyzyjny Markowa (MDP), gdzie stan s odpowiadał połączonemu wektorowi cech Fconcat, akcja reprezentowała wybór strategii fuzji (sumowanie ważone lub bezpośrednie konkatenowanie), a nagroda r była definiowana jako dokładność klasyfikacji uzyskana z klasyfikatora w dalszej fazie wytrenowanej na połączonych cechach. Funkcję wartości akcji-Q(s,a) przybliżono za pomocą Deep Q-Network (DQN), która oszacowała oczekiwaną skumulowaną nagrodę
. Parametry modelu były iteracyjnie aktualizowane zgodnie z regułą.

gdzie α = 0,0001 to szybkość uczenia się, γ = 0,99 to współczynnik dyskontowy, a s' oznacza następny stan.
Podczas optymalizacji Uczenia Wzmocnionego model był trenowany dla 1000 odcinków o wielkości partii 32. Zastosowano strategię chciwości epsilonów, aby zrównoważyć eksplorację i eksploatację, z
inicjalizacją na poziomie 1.0 i spadkiem o 0.995 na odcinek. Przy prawdopodobieństwie
wybrano losową strategię fuzji, natomiast przy prawdopodobieństwie 1 -
, strategię o najwyższej szacowanej wartości Q. Zbieżność nagrody była monitorowana podczas całego szkolenia, a stabilna wydajność (>0,85 skumulowanej nagrody) obserwowana była zazwyczaj przez około 500 epizodów.
Workflow HDRL obejmuje wstępne przetwarzanie sygnałów multimodalnych, wyodrębnianie specyficznych dla modalności głębokich cech, tworzenie jednolitej reprezentacji stanu, wybór strategii fuzji oraz ulepszanie polityki na podstawie informacji zwrotnej od agenta DQN. Interakcja między ekstraktorami cech opartymi na CNN a agentem uczenia ze wzmocnieniem tworzy zamknięty system adaptacyjnej fuzji (Rysunek 3), umożliwiający aktualizacje polityki na podstawie informacji zwrotnej z nagrody.

Rysunek 3: Diagram przepływu proponowanego procesu hybrydowego Głębokiego Uczenia Wzmocnionego (HDRL). Diagram przepływu pracy pokazujący ekstrakcję cech CNN specyficzną dla modalności, łączenie cech, wybór strategii fusion (DQN/PPO) oparty na uczeniu ze wzmocnieniem (DQN/PPO), klasyfikację dalszą oraz aktualizację polityki opartej na nagrodach. Proszę kliknąć tutaj, aby zobaczyć większą wersję tej figurki.