Диагностика расстройств шизофренического спектра на основе нейрофизиологических индикаторов процесса антиципации с помощью моделей машинного обучения
Автор - Рабинович Эрнест Ильич
Научный консультант - Антюфриева Любовь
Курс: «Нейронные сети и их применение в научных исследованиях»
Нарушение прогностического кодирования при шизофрении
Нарушение точности нисходящих прогнозов
Дизрегуляция салиенса
Диагностика расстройств шизофренического спектра на основе нейрофизиологических индикаторов процесса антиципации с помощью моделей машинного обучения.
Данные
Схема предъявления стимулов
Диагностика расстройств шизофренического спектра на основе нейрофизиологических индикаторов процесса антиципации с помощью моделей машинного обучения.
Данные
Диагностика расстройств шизофренического спектра на основе нейрофизиологических индикаторов процесса антиципации с помощью моделей машинного обучения.
Выделение признаков
Диагностика расстройств шизофренического спектра на основе нейрофизиологических индикаторов процесса антиципации с помощью моделей машинного обучения.
PeakToPeak Amplitude
ERP 80% dev - ERP 80% st
ERP 50% dev - ERP 50% st
Данные
Диагностика расстройств шизофренического спектра на основе нейрофизиологических индикаторов процесса антиципации с помощью моделей машинного обучения.
Модели
Диагностика расстройств шизофренического спектра на основе нейрофизиологических индикаторов процесса антиципации с помощью моделей машинного обучения.
Подбор гиперпараметров:
Диагностика расстройств шизофренического спектра на основе нейрофизиологических индикаторов процесса антиципации с помощью моделей машинного обучения.
Fully connected NeuralNet
Диагностика расстройств шизофренического спектра на основе нейрофизиологических индикаторов процесса антиципации с помощью моделей машинного обучения.
Layer | Type | Numbers of Neurons | Activation | Other parameters |
1 | Input | - | - | - |
2 | Dense | 64 | ReLU | - |
3 | Dense | 128 | ReLU | - |
4 | Dense | 256 | ReLU | - |
5 | Dense | 128 | ReLU | - |
6 | Dropout | - | - | p = 0.2 |
7 | Dense | 64 | ReLU | - |
8 | Dense | 16 | - | - |
Диагностика расстройств шизофренического спектра на основе нейрофизиологических индикаторов процесса антиципации с помощью моделей машинного обучения.
TripletLoss
Random forest
Диагностика расстройств шизофренического спектра на основе нейрофизиологических индикаторов процесса антиципации с помощью моделей машинного обучения.
Cross-val:
Mean = 0.88,
std = 0.11
Test:
Mean = 0.87
std = 0.09
Balanced Random forest
Диагностика расстройств шизофренического спектра на основе нейрофизиологических индикаторов процесса антиципации с помощью моделей машинного обучения.
Cross-val:
Mean = 0.85,
std = 0.14
Test:
Mean = 0.85
std = 0.11
Decision Tree
Диагностика расстройств шизофренического спектра на основе нейрофизиологических индикаторов процесса антиципации с помощью моделей машинного обучения.
Cross-val:
Mean = 0.82,
std = 0.15
Test:
Mean = 0.82
std = 0.10
SVM polynomial
Диагностика расстройств шизофренического спектра на основе нейрофизиологических индикаторов процесса антиципации с помощью моделей машинного обучения.
Cross-val:
Mean = 0.89,
std = 0.11
Test:
Mean = 0.89
std = 0.10
SVM linear
Диагностика расстройств шизофренического спектра на основе нейрофизиологических индикаторов процесса антиципации с помощью моделей машинного обучения.
Cross-val:
Mean = 0.88,
std = 0.11
Test:
Mean = 0.88
std = 0.10
SVM sigmoid
Диагностика расстройств шизофренического спектра на основе нейрофизиологических индикаторов процесса антиципации с помощью моделей машинного обучения.
Cross-val:
Mean = 0.89,
std = 0.11
Test:
Mean = 0.89
std = 0.10
SVM rbf
Диагностика расстройств шизофренического спектра на основе нейрофизиологических индикаторов процесса антиципации с помощью моделей машинного обучения.
Cross-val:
Mean = 0.89,
std = 0.11
Test:
Mean = 0.88
std = 0.10
XGBoost
Диагностика расстройств шизофренического спектра на основе нейрофизиологических индикаторов процесса антиципации с помощью моделей машинного обучения.
Cross-val:
Mean = 0.89,
std = 0.10
Test:
Mean = 0.88
std = 0.08
Диагностика расстройств шизофренического спектра на основе нейрофизиологических индикаторов процесса антиципации с помощью моделей машинного обучения.
| Precision | Recall | F1 |
Random Forest | 0.84 | 0.87 | 0.84 |
Balanced Random Forest | 0.85 | 0.87 | 0.84 |
Decision Tree | 0.77 | 0.80 | 0.76 |
SVC (polynomial) | 0.86 | 0.88 | 0.86 |
SVC (sigmoid) | 0.90 | 0.91 | 0.89 |
SVC (linear) | 0.85 | 0.87 | 0.85 |
SVC (rbf) | 0.86 | 0.88 | 0.86 |
XGBoost | 0.82 | 0.86 | 0.83 |
Средняя производительность моделей, оцененная на 100 различных разбиениях на трейн/тест
Диагностика расстройств шизофренического спектра на основе нейрофизиологических индикаторов процесса антиципации с помощью моделей машинного обучения.
| Precision | Recall | F1 |
Random Forest | 0.9 | 0.93 | 0.91 |
Balanced Random Forest | 0.88 | 0.88 | 0.86 |
Decision Tree | 0.80 | 0.85 | 0.80 |
SVC (polynomial) | 0.90 | 0.91 | 0.89 |
SVC (linear) | 0.92 | 0.92 | 0.91 |
SVC (rbf) | 0.88 | 0.89 | 0.87 |
Средняя производительность моделей, оцененная на 100 различных разбиениях на трейн/тест, после жадного отбора признаков
Диагностика расстройств шизофренического спектра на основе нейрофизиологических индикаторов процесса антиципации с помощью моделей машинного обучения.
Shap
MKL
Диагностика расстройств шизофренического спектра на основе нейрофизиологических индикаторов процесса антиципации с помощью моделей машинного обучения.
Cross-val:
Mean = 0.87,
std = 0.15
Test:
Mean = 0.87
std = 0.09
Диагностика расстройств шизофренического спектра на основе нейрофизиологических индикаторов процесса антиципации с помощью моделей машинного обучения.
Средняя производительность алгоритма EasyMKL, оцененная на 100 различных разбиениях на трейн/тест
Диагностика расстройств шизофренического спектра на основе нейрофизиологических индикаторов процесса антиципации с помощью моделей машинного обучения.
Обучение полносвязной нейронной сети с помощью Triplett Loss
Диагностика расстройств шизофренического спектра на основе нейрофизиологических индикаторов процесса антиципации с помощью моделей машинного обучения.
Получение эмбеддингов по эпохам обучения
Диагностика расстройств шизофренического спектра на основе нейрофизиологических индикаторов процесса антиципации с помощью моделей машинного обучения.
Средняя производительность классификации эмбеддингов, полученных полносвязной нейронной сетью, с помощью kNN, оцененная на 100 различных разбиениях на трейн/тест
Заключение
Диагностика расстройств шизофренического спектра на основе нейрофизиологических индикаторов процесса антиципации с помощью моделей машинного обучения.
Полученные результаты показывают релевантность используемых данных для целей диагностики шизофрении
Наиболее информативными являются лобно-центральные отведения ЭЭГ
Лучшую точность показывает SVM после жадного отбора признаков