$$\rightleftharpoonup{xx}$$
$$\longleftharp{xx}$$,
$$\longrightharp{xx}$$,
El método propuesto segmenta la señal EEG estratégicamente en el dominio de la frecuencia para mejorar la resolución espectral. Posteriormente, las señales EEG capturadas se epochinan en 10 segmentos para mejorar la resolución temporal y la consistencia en el cálculo de características. El método propuesto implementa un clasificador basado en RF, XGBoost y SVM sobre características derivadas del EEG, contribuyendo así al desarrollo de métodos robustos y escalables para el análisis del EEG. El modelo RF con ciertos cambios en los hiperparámetros ha demostrado ser el mejor desempeño entre otros modelos estándar. Por ello, se propone un modelo híbrido, fusionado y apilado metalearner que combina RF y XG-Boost para mejorar aún más la precisión de la predicción.
El conjunto de datos utilizado en este método se recopiló conforme a los estándares éticos institucionales. Los proveedores del conjunto de datos obtuvieron el consentimiento de los participantes antes de la adquisición de datos.
1. Especificación del conjunto de datos
Para demostrar la robustez del sistema, se han utilizado dos conjuntos de datos. El primer conjunto de datos se recoge del repositoriode EEG 20, que comprende 88 participantes en estado de reposo con ojos cerrados, donde 36 fueron diagnosticados con EA, 23 de ellos FTD y 29 estaban sanos. Las grabaciones EEG se adquirieron en estado de reposo, con los ojos cerrados, utilizando un sistema de 19 canales tras el montaje internacional 10-20. En el conjunto de datos mostrado en la Figura 2, todas las señales fueron muestreadas a 500 Hz y proporcionadas en formato BIDS con preprocesamiento preliminar por parte de los colaboradores del conjunto.
El segundo conjunto de datos se recoge de un repositorio públicoexterno 21 que comprende 35 participantes en estado de reposo. De los cuales, 13 participantes tienen diagnóstico de enfermedad de Alzheimer (EA), 7 son pacientes con deterioro cognitivo leve (DCL) y 15 son ancianos sanos. Solo se utilizaron los segmentos de referencia del estado de reposo del segundo conjunto de datos para mantener la consistencia con el primerconjunto de datos 20. El segundo conjunto dedatos 21 requería una cadena completa de preprocesamiento, como se menciona en la Figura 3.
2. Preprocesamiento
Los estudios existentes dependen en gran medida de complejos procesos de preprocesamiento, pasos manuales de corrección de artefactos o ICA-ASR para eliminar movimientos musculares, que limitan la reproducibilidad en los flujos de trabajo clínicos rutinarios. Para abordar estas limitaciones, la pipeline propuesta se centra en un enfoque simplificado solo con EEG que elimina la necesidad de procedimientos computacionalmente intensivos para eliminar artefactos y, en su lugar, enfatiza el filtrado controlado, la segmentación de épocas y el cálculo de características específicas por frecuencia.
- Ordenación de archivos EEG
Antes del preprocesamiento, todos los archivos EEG se ordenaban automáticamente para mantener una consistencia en el orden por participante y por sesión. Esto garantizaba que la extracción de características, la epocalidad y la alineación de etiquetas fueran uniformes en todos los sujetos y evitaba errores de desajuste en archivos. Este paso se muestra en la Figura 3 como "Ordenar archivo EEG." Todo el preprocesamiento se realiza utilizando un kit de herramientas estándar de código abierto para procesamiento EEG.
- Filtrado pasabanda y re-referenciación
Se aplicó un filtro pasa-banda Butterworth en el rango de 0,5-40 Hz para conservar componentes relevantes del EEG mientras atenuaba la deriva y el ruido de alta frecuencia. Las señales se volvieron a referenciar a la media de todos los canales EEG, minimizando así el sesgo espacial y mejorando la relación señal-ruido.
- Desplazamiento de frecuencia de Nyquist
Para mejorar la resolución de los componentes de baja frecuencia, se aplicó una mejora de desplazamiento de Nyquist reduciendo a la mitad la frecuencia de muestreo de 500Hz a 250Hz. Siguiendo el paso anterior, la modificación de Nyquist lo reduce a 125Hz. Como las frecuencias Delta, Theta, Alfa, Beta y Gamma son ondas lentas que oscilan entre 0,5 y 40 Hz, esto desplaza el límite efectivo de Nyquist y mejora la claridad de las oscilaciones de baja frecuencia, relevantes para la demencia. Esta modificación ayudó a capturar las ondas cerebrales lentas de forma más adecuada. La ecuación (1) muestra claramente el resultado del medicamento de Nyquist.
Muestreo F = 500 Hz
N muestreo = Fmuestreo/2
Nnyquist= Nmuestreo/2 (1)
Es importante señalar que Nnyquist no es la frecuencia de muestreo operativa, sino simplemente la frecuencia máxima representable, y por lo tanto no afecta a la segmentación de época. Es solo un límite teórico utilizado para garantizar que el contenido de frecuencia de interés (0,5-40 Hz) se preserve completamente.
- Frecuencia de muestreo de época a 10 segmentos
Las grabaciones EEG se dividen en 10 segmentos de igual longitud. Este paso es esencial para captar la actividad no estacionaria de la señal EEG a lo largo del tiempo. Los segmentos cortos y uniformes permiten un cálculo estable de características específicas de banda y enriquecen muestras de entrenamiento para la clasificación por aprendizaje automático. Por tanto, las 250 muestras por segundo se convierten en 250 × 10 = 2500 muestras. Finalmente, se toman 10 épocas de igual duración por sujeto para estrategias de análisis por época. Esta estrategia de análisis generó 880 muestras para 88 participantes reales.
La necesidad de un análisis por época es monitorizar minuciosamente los cambios en la señal cerebral, proporcionando así al modelo una cantidad considerable de datos temporales para aprender y evitar el sobreajuste. La importancia de este enfoque radica en asegurar datos de EEG más finos, facilitando análisis que requieren mayor resolución temporal sin comprometer los valores totales de características. Generando épocas de 10 segundos, la metodología permite mejorar la modelización de la dinámica del EEG y se alinea con las duraciones estándar de épocas en los protocolos de análisis EEG.
3. Extracción de características
Las señales EEG preprocesadas se pasan a métodos de extracción de características. Las señales EEG se filtran pasa-banda en 5 bandas de frecuencia estándar: Delta (0,5-4 Hz), Theta (4-8 Hz), Alfa (8-13 Hz), Beta (13-25 Hz) y Gamma (25-40 Hz).
Para cada señal filtrada, se calculaba el valor de la Raíz Cuadrática Media (RMS) en todos los canales EEG. La formulación matemática de RMS se da en la Ecuación (1):
RMS =
(1)
Aquí, xi es la amplitud de la señal EEG en la muestra de la i vez. N es el número total de muestras en el segmento de señal.
El RMS fue seleccionado como característica principal debido a su capacidad para cuantificar la energía de la actividad oscilatoria en cada banda de frecuencia. La enfermedad de Alzheimer suele asociarse con un aumento de la actividad delta y theta y una reducción de la actividad alfa y beta. La demencia frontotemporal (DFT) puede mostrar patrones distintos en estas bandas20,21. Como PSD es matemáticamente redundante respecto al RMS, no se considera una característica. Para mantener un conjunto de datos compacto, se excluyen las características restantes, como Hjorth y Entropy.
4. Fusión híbrida
Las características RMS extraídas en el dominio temporal se combinaron con características en el dominio de la frecuencia recibidas del filtrado pasa banda convirtiéndolo en un espacio híbrido para la entrada del modelo. El detalle de la compilación se muestra en la Figura 3. El conjunto de datos normalizado final está organizado en un formato tabular con filas que representan a los participantes y columnas que representan características de la banda tiempo-frecuencia y características demográficas como edad, género y grupo. La tubería mostrada en la Figura 2 extrae eficientemente características del EEG relevantes para la investigación del Alzheimer. Las características extraídas proporcionan información sobre la actividad neuronal dentro de bandas clave de frecuencia y pueden utilizarse para aprendizaje automático y análisis estadístico.
5. Selección de características
Para mejorar el rendimiento del modelo y reducir la dimensionalidad de características, se aplicó un enfoque basado en el Análisis de Varianza bidireccional (ANOVA). La selección basada en ANOVA se aplicó con el Grupo y la edad como factores independientes, y las bandas de frecuencia RMS como características dependientes. Este análisis evaluó los efectos de la interacción entre grupos, edades y grupos de edad en cada característica. Se seleccionaron características con valores p < 0,05 para al menos un factor para una clasificación posterior. Los resultados detallados basados en ANOVA se muestran en la sección de resultados. Estas características son las variables más informativas para la clasificación, lo que ayudó a mejorar la precisión y generalización del modelo predictivo final. Como resultado de la prueba ANOVA, solo se seleccionaron las características más informativas para la creación del modelo. Y los atributos adicionales no numéricos, género y participant_id, fueron excluidos del análisis. El grupo de variables objetivo se codificaba con etiqueta para su clasificación.
6. Descripción del modelo
Para clasificar la enfermedad en 3 clases, a saber: Alzheimer, Control y Demencia Frontal, se construyeron tres modelos supervisados de aprendizaje automático, como XG-Boost, Random Forest, Support Vector Machine SVM y un modelo apilado. Todos los modelos mencionados utilizaron el 70% del conjunto de datos de entrenamiento y el 30% del conjunto de datos de prueba. Cada modelo fue seleccionado por su rendimiento probado en el análisis de datos sanitarios y su capacidad para manejar espacios de características no lineales y de alta dimensión. Se realizó una evaluación del modelo en el conjunto de datos de validación para evaluar la generalizabilidad del mismo.
El rendimiento del modelo se evaluó utilizando una matriz de confusión, que comprende los verdaderos positivos (TP), falsos positivos (FP), los verdaderos negativos (TN) y los falsos negativos (FN). Las métricas de rendimiento incluían precisión y un informe de clasificación que detallaba la precisión, la recuperación y las puntuaciones F1 para cada clase. La siguiente ecuación ilustra las métricas de rendimiento.
(2)
(3)
(4)
(5)
1. Clasificador XG-Boost
Este método implementó un modelo de clasificación multiclase utilizando el algoritmo XGBoost para predecir estados cognitivos derivados de EEG a partir de características extraídas, como se muestra en la salida 1 de la Figura 3. La salida 1 contiene características RMS y la puntuación MMSE. Al construir el modelo XGBoost, se incluye la puntuación MMSE para probar el comportamiento no lineal de las características RMS. Se utiliza como referencia y referencia para el desarrollo del sistema. El MMSE fue excluido en etapas posteriores para cumplir el objetivo de un sistema totalmente automatizado e independiente del profesional.
XG-Boost es un framework de aumento de gradiente optimizado para la eficiencia y precisión, lo que lo hace muy adecuado para manejar datos estructurados en tareas de clasificación. Los pocos hiperparámetros se ajustan en el conjunto de datos EEG utilizando técnicas como la validación cruzada y la búsqueda en cuadrícula. Los valores clave de los parámetros definidos a continuación se eligen para controlar el sobreajuste mediante regularización y para capturar la complejidad de las características del EEG sin que sean demasiado complejas.
Los parámetros clave del modelo son: (1) Profundidad máxima del árbol = 8, que limita la profundidad de los árboles de decisión para reducir el sobreajuste y mejorar la generalización. Porque un árbol poco profundo perdería información innecesaria, ya que un árbol más profundo podría sobreajustarse. (2) La regularización L2 (λ = 10) añade una penalización para coeficientes grandes para minimizar el sobreajuste controlando la complejidad del modelo. (2) Se introduce la regularización L1 (α = 5) para aportar mayor esparsidad al modelo y así mejorar la interpretabilidad y robustez. El número de estimadores (n_estimators=8) limitaba el número de iteraciones de impulso para mantener la eficiencia computacional mientras se equilibraba el rendimiento.
2. Clasificador de Bosque Aleatorio
Random Forest es un clasificador de conjunto que construye múltiples árboles de decisión sobre subconjuntos aleatorios de datos y características, agregando sus resultados para la predicción final. Ofrece robustez al ruido, maneja bien los datos no lineales y reduce el sobreajuste promediando predicciones de árboles diversos. En este trabajo, el modelo de Bosque Aleatorio se configuró con parámetros clave utilizando el método de prueba y error. Estos parámetros se explican a continuación:
1) n_estimators=100
Se elige para limitar el número de árboles de decisión en el bosque. Cuantos más árboles, mejor la generalización y más estables serán las predicciones.
2) max_depth=10
Indica la profundidad máxima de cada árbol. Si el árbol tiene poca profundidad, existe la posibilidad de que se subajuste. Mientras que una alta profundidad puede capturar detalles y memorizar los datos de entrenamiento, puede sobreajustarse. Por lo tanto, elegir el 10 es un punto ideal, especialmente para señales ruidosas o complejas como el EEG.
3) random_state=40
Corrige la semilla de generación de números aleatorios usada para el muestreo Bootstrap y la construcción de árboles. Esto garantiza la reproducibilidad
4) n_jobs=-1
Al establecer el valor de n_jobs a -1, el modelo utiliza todos los núcleos de la CPU de forma eficiente.
7. Máquina de Vectores de Soporte (SVM)
Support Vector Machine es un clasificador basado en márgenes conocido por su rendimiento lineal en la clasificación binaria y multiclase22. Este modelo utilizaba un núcleo de función base radial debido a su capacidad para capturar señales EEG no lineales. Sin embargo, el modelo mostró un rendimiento subóptimo, probablemente debido a la falta de escalado de características y al tamaño limitado del conjunto de datos. Aunque la SVM es teóricamente potente, su sensibilidad al ajuste de hiperparámetros y a la distribución de datos puede explicar la menor precisión observada en comparación con los modelos basados en árboles.
8. Modelo propuesto de fusión HY-
Aunque el modelo Random Forest alcanzó alta precisión para garantizar la generalización y escalabilidad, se construyó un modelo apilado con Random Forest y XG-Boost. La salida se introduce como entrada a un modelo de regresión logística. Y la regresión logística actúa como un metaaprendizaje. El diagrama arquitectónico del modelo se muestra en la Figura 4.
Junto con elprimer conjunto de entradas20, el modelo apilado también recibía entrada del segundo conjunto dedatos 23. El conjunto de datos de Mendely21 se preprocesa en una forma comprensible para el modelo extrayendo RMS y epocando hasta 10 s. Los resultados obtenidos por cada modelo se muestran en la sección de resultados.