$$\rightleftharpoonup{xx}$$
$$\longleftharp{xx}$$,
$$\longrightharp{xx}$$,
La méthode proposée segmente stratégiquement les signaux EEG dans le domaine fréquentiel afin d’améliorer la résolution spectrale. Par la suite, les signaux EEG capturés sont répartis en 10 segments afin d’améliorer la résolution temporelle et la cohérence dans le calcul des caractéristiques. La méthode proposée implémente un classificateur basé sur RF, XGBoost et SVM sur des caractéristiques EEG dérivées, contribuant ainsi au développement de méthodes robustes et évolutives pour l’analyse EEG. Le modèle RF avec certains changements d’hyperparamètres s’est avéré être le plus performant parmi les autres modèles standards. Par conséquent, un modèle hybride, fusionné et empilé de métaappant est proposé qui combine RF et XG-Boost afin d’améliorer encore la précision des prédictions.
L’ensemble de données utilisé dans cette méthode a été collecté conformément aux normes éthiques institutionnelles. Les fournisseurs de jeux de données ont obtenu le consentement des participants avant la collecte des données.
1. Spécification du jeu de données
Pour prouver la robustesse du système, deux ensembles de données ont été utilisés. Le premier ensemble de données est collecté à partir du dépôt de donnéesEEG 20, comprenant 88 participants dans la condition de repos des yeux fermés, où 36 ont été diagnostiqués comme une maladie d’Arme, 23 d’entre eux FTD, et 29 étaient en bonne santé. Les enregistrements EEG ont été réalisés au repos, les yeux fermés, à l’aide d’un système à 19 canaux, après le montage international 10-20. Dans le jeu de données montré à la Figure 2, tous les signaux ont été échantillonnés à 500 Hz et fournis au format BIDS avec un prétraitement préliminaire par les contributeurs du jeu de données.
Le second ensemble de données est collecté à partir d’un dépôt publicexterne 21 comprenant 35 participants en état de repos. Parmi eux, 13 participants ont reçu un diagnostic de maladie d’Alzheimer (MA), 7 sont des patients atteints de troubles cognitifs légers (DCL) et 15 sont des personnes âgées en bonne santé. Seuls les segments de base de l’état de repos du second jeu de données ont été utilisés pour maintenir la cohérence avec le premier ensemblede données 20. Le second ensemble de données21 nécessitait un pipeline complet de prétraitement, comme mentionné à la Figure 3.
2. Prétraitement
Les études existantes reposent fortement sur des pipelines complexes de prétraitement, des étapes manuelles de correction des artefacts ou ICA-ASR pour éliminer les mouvements musculaires, qui limitent la reproductibilité dans les flux de travail cliniques routiniers. Pour remédier à ces limitations, le pipeline proposé se concentre sur une approche simplifiée basée uniquement sur l’EEG, qui élimine le besoin de procédures de suppression d’artefacts à forte intensité en calcul et met plutôt l’accent sur un filtrage contrôlé, la segmentation d’époque et le calcul de caractéristiques spécifiques à la fréquence.
- Tri des fichiers EEG
Avant le prétraitement, tous les fichiers EEG étaient automatiquement triés pour maintenir une cohérence dans l’ordre par participant et par session. Cela garantissait que l’extraction des caractéristiques, l’époque et l’alignement des étiquettes étaient uniformes sur tous les sujets et évitait les erreurs de désappariement des fichiers. Cette étape est affichée à la Figure 3 sous le nom « Trier fichier EEG ». Tout le prétraitement est effectué à l’aide d’une boîte à outils standard open source pour le traitement EEG.
- Filtrage passe-bande et reréférencement
Un filtre passe-bande Butterworth a été appliqué dans la plage de 0,5 à 40 Hz pour conserver les composantes EEG pertinentes tout en atténuant la dérive et le bruit de haute fréquence. Les signaux ont été réorientés vers la moyenne de tous les canaux EEG, minimisant ainsi le biais spatial et améliorant le rapport signal/bruit.
- Décalage de fréquence de Nyquist
Pour améliorer la résolution des composants basse fréquence, une amélioration du décalage de Nyquist a été appliquée en réduisant de moitié la fréquence d’échantillonnage de 500 Hz à 250 Hz. En suivant l’étape ci-dessus, la modification de Nyquist la réduit à 125 Hz. Comme les fréquences Delta, Theta, Alpha, Beta et Gamma sont des ondes lentes allant de 0,5 à 40 Hz, cela modifie la limite effective de Nyquist et améliore la clarté des oscillations basses fréquences, pertinentes pour la démence. Cette modification a permis de capturer plus efficacement les ondes cérébrales lentes. L’équation (1) montre clairement le résultat du médicament de Nyquist.
Échantillonnage F = 500 Hz
N échantillonnage = Féchantillonnage/2
Nnyquist = Néchantillonnage/2 (1)
Il est important de noter que Nnyquist n’est pas la fréquence d’échantillonnage opérationnelle mais simplement la fréquence maximale représentable, et n’affecte donc pas la segmentation d’époque. Ce n’est qu’une limite théorique utilisée pour garantir que le contenu fréquentiel d’intérêt (0,5-40 Hz) soit entièrement préservé.
- Fréquence d’échantillonnage d’époque jusqu’à 10 segments
Les enregistrements EEG sont divisés en 10 segments de longueur égale. Cette étape est essentielle pour capter l’activité non stationnaire du signal EEG au fil du temps. Les segments courts et uniformes permettent un calcul stable des caractéristiques spécifiques à chaque bande et enrichissent les échantillons d’entraînement pour la classification par apprentissage automatique. Ainsi, les 250 échantillons par seconde deviennent 250 × 10 = 2500 échantillons. Enfin, 10 époques de longueur égale par sujet sont prises pour des stratégies d’analyse par époque. Cette stratégie d’analyse a généré 880 échantillons pour 88 participants réels.
Le besoin d’une analyse par époque consiste à surveiller minutieusement les changements du signal cérébral, fournissant ainsi au modèle une quantité suffisante de données temporelles pour apprendre et éviter le sur-ajustement. L’importance de cette approche est d’assurer des données d’époque EEG plus fines, facilitant des analyses nécessitant une résolution temporelle plus élevée sans compromettre les valeurs totales des caractéristiques. En générant des époques de 10 secondes, la méthodologie favorise une modélisation améliorée de la dynamique EEG et s’aligne avec les durées d’époque standard dans les protocoles d’analyse EEG.
3. Extraction de caractéristiques
Les signaux EEG prétraités sont transmis aux méthodes d’extraction des caractéristiques. Les signaux EEG sont filtrés en passe-bande dans 5 bandes de fréquences standard : Delta (0,5-4 Hz), Theta (4-8 Hz), Alpha (8-13 Hz), Beta (13-25 Hz) et Gamma (25-40 Hz).
Pour chaque signal filtré, la valeur du carré moyen (RMS) était calculée sur tous les canaux EEG. La formulation mathématique de RMS est donnée dans l’équation (1) :
RMS =
(1)
Ici, xi est l’amplitude du signal EEG à l’échantillon i-ème temps. N est le nombre total d’échantillons dans le segment du signal.
Le RMS a été choisi comme caractéristique principale en raison de sa capacité à quantifier l’énergie de l’activité oscillatoire dans chaque bande de fréquences. La maladie d’Alzheimer est souvent associée à une augmentation de l’activité delta et thêta ainsi qu’à une diminution de l’activité alpha et bêta. La démence fronto-temporale (DFT) peut présenter des schémas distincts sur cesbandes 20,21. Comme la PSD est mathématiquement redondante par rapport à la RMS, elle n’est pas considérée comme une fonctionnalité. Pour maintenir un ensemble de données compact, les autres caractéristiques, telles que Hjorth et Entropy, sont exclues.
4. Fusion hybride
Les caractéristiques RMS extraites dans le domaine temporel ont été combinées avec les caractéristiques du domaine fréquentiel reçues par filtrage passe-bande en en faisant un espace hybride pour l’entrée du modèle. Le détail de la compilation est montré à la Figure 3. Le jeu de données final normalisé est organisé sous forme tabulaire avec des lignes représentant les participants et des colonnes représentant les caractéristiques de la bande temps-fréquence ainsi que des caractéristiques démographiques telles que l’âge, le sexe et le groupe. Le pipeline montré à la Figure 2 extrait efficacement les caractéristiques EEG pertinentes pour la recherche sur Alzheimer. Les caractéristiques extraites fournissent des informations sur l’activité neuronale dans les bandes de fréquences clés et peuvent être utilisées pour l’apprentissage automatique et l’analyse statistique.
5. Sélection des fonctionnalités
Pour améliorer les performances du modèle et réduire la dimensionnalité des caractéristiques, une approche basée sur l’analyse bidirectionnelle de la variance (ANOVA) a été appliquée. La sélection basée sur l’ANOVA a été appliquée avec le groupe et l’âge comme facteurs indépendants, et les bandes de fréquences RMS comme caractéristiques dépendantes. Cette analyse a évalué les effets de l’interaction groupe, âge et groupe d’âge sur chaque caractéristique. Les caractéristiques avec des valeurs p < 0,05 pour au moins un facteur ont été sélectionnées pour une classification ultérieure. Les résultats détaillés basés sur l’ANOVA sont présentés dans la section résultats. Ces caractéristiques sont les variables les plus informatives pour la classification, ce qui a contribué à améliorer la précision et la généralisation du modèle prédictif final. À la suite du test ANOVA, seules les caractéristiques les plus informatives ont été sélectionnées pour la création du modèle. Et les attributs non numériques supplémentaires, Genre et participant_id, ont été exclus de l’analyse. Le groupe de variables cibles était codé par étiquette pour la classification.
6. Description du modèle
Pour classer la maladie en 3 classes, à savoir Alzheimer, Contrôle et Démence frontale, trois modèles d’apprentissage automatique supervisés, tels que XG-Boost, Random Forest, Support Vector Machine SVM et un modèle empilé, ont été construits. Tous les modèles mentionnés utilisaient 70 % du jeu de données d’entraînement et 30 % du jeu de données de test. Chaque modèle a été sélectionné pour ses performances éprouvées en analyse de données de santé et sa capacité à gérer des espaces de caractéristiques non linéaires et à haute dimension. L’évaluation du modèle a été réalisée sur le jeu de données de validation afin d’évaluer la généralisabilité du modèle.
La performance du modèle a été évaluée à l’aide d’une matrice de confusion, comprenant les vrais positifs (TP), les faux positifs (FP), les vrais négatifs (TN) et les faux négatifs (FN). Les indicateurs de performance comprenaient la précision et un rapport de classification détaillant la précision, le rappel et les scores F1 pour chaque catégorie. L’équation suivante illustre les indicateurs de performance.
(2)
(3)
(4)
(5)
1. Classificateur XG-Boost
Cette méthode a implémenté un modèle de classification multi-classes utilisant l’algorithme XGBoost pour prédire les états cognitifs dérivés de l’EEG à partir de caractéristiques extraites, comme montré dans la sortie 1 de la Figure 3. La sortie 1 contient les caractéristiques RMS et le score MMSE. Lors de la construction du modèle XGBoost, le score MMSE est inclus pour tester le comportement non linéaire des caractéristiques RMS. Il sert de référence et de référence pour le développement du système. Le MMSE a été exclu à des étapes ultérieures afin de répondre à l’objectif d’un système entièrement automatisé et indépendant du clinicien.
XG-Boost est un cadre de gradient boosting, optimisé pour l’efficacité et la précision, ce qui le rend parfaitement adapté à la gestion de données structurées dans les tâches de classification. Les quelques hyperparamètres sont réglés sur le jeu de données EEG à l’aide de techniques telles que la validation croisée et la recherche en grille. Les valeurs clés des paramètres définies ci-dessous sont choisies pour contrôler le surajustement via régularisation et pour capturer la complexité des caractéristiques EEG sans que cela devienne trop complexe.
Les paramètres clés du modèle sont : (1) Profondeur maximale de l’arbre = 8, qui limite la profondeur des arbres de décision afin de réduire le surajustement et d’améliorer la généralisation. Parce qu’un arbre peu profond manquerait des informations inutiles, car un arbre plus profond pourrait trop s’adapter. (2) La régularisation L2 (λ = 10) ajoute une pénalité pour les coefficients élevés afin de minimiser le surajustement en contrôlant la complexité du modèle. (2) La régularisation L1 (α = 5) est introduite pour ajouter une parcimonie supplémentaire au modèle afin d’améliorer l’interprétabilité et la robustesse. Le nombre d’estimateurs (n_estimators=8) limitait le nombre d’itérations de boost pour maintenir l’efficacité computationnelle tout en équilibrant la performance.
2. Classificateur de forêt aléatoire
Random Forest est un classificateur d’ensemble qui construit plusieurs arbres de décision sur des sous-ensembles aléatoires de données et de caractéristiques, agrégeant leurs résultats pour la prédiction finale. Il offre une robustesse au bruit, gère bien les données non linéaires et réduit le surajustement en moyenne des prédictions d’arbres diverses. Dans ce travail, le modèle de la forêt aléatoire a été configuré avec des paramètres clés en utilisant la méthode d’essais et erreurs. Ces paramètres sont expliqués ci-dessous :
1) n_estimators=100
Il est choisi pour limiter le nombre d’arbres de décision dans la forêt. Plus il y a d’arbres, meilleure est la généralisation et plus les prédictions sont stables.
2) max_depth=10
Il indique la profondeur maximale de chaque arbre. Si l’arbre a une faible profondeur, il y a un risque de sous-adaptation. Alors qu’une grande profondeur peut capturer les détails et mémoriser les données d’entraînement, elle peut sur-ajuster. Par conséquent, choisir 10 est un point idéal, surtout pour les signaux bruyants ou complexes comme l’EEG.
3) random_state=40
Il corrige la graine de génération de nombres aléatoires utilisée pour l’échantillonnage Bootstrap et la construction d’arbres. Cela garantit la reproductibilité
4) n_jobs=-1
En fixant la valeur de n_jobs à -1, le modèle utilise efficacement tous les cœurs CPU.
7. Machine à vecteurs de support (SVM)
Support Vector Machine est un classificateur basé sur la marge connu pour ses performances linéaires en classification binaire etmulti-classe 22. Ce modèle utilisait un noyau de fonction de base radiale en raison de sa capacité à capturer des signaux EEG non linéaires. Cependant, le modèle a montré des performances sous-optimales, probablement en raison de l’absence de mise à l’échelle des fonctionnalités et de la taille limitée du jeu de données. Bien que la SVM soit théoriquement puissante, sa sensibilité à l’ajustement des hyperparamètres et à la distribution des données peut expliquer la précision plus faible observée par rapport aux modèles basés sur les arbres.
8. Modèle proposé de fusion HY-
Bien que le modèle Random Forest ait atteint une grande précision, pour garantir la généralisation et l’évolutivité, un modèle empilé a été construit avec Random Forest et XG-Boost. La sortie est alimentée en entrée dans un modèle de régression logistique. Et la régression logistique agit comme un méta-apprenant. Le diagramme architectural du modèle est présenté à la Figure 4.
Avec le1er ensembled’entrées 20, le modèle empilé prenait également l’entrée du second ensemble dedonnées 23. Le jeu de données de Mendely21 est prétraité sous une forme compréhensible par le modèle en extrayant le RMS et en époching jusqu’à 10 s. Les résultats obtenus par chaque modèle sont présentés dans la section résultats.