Разработка модели машинного обучения для прогноза результата склеропластики у детей.
д.м.н. Костенев Сергей Владимирович
к.м.н. Кечин Евгений Владимирович
аспирант Шихалиева Эльвира Абдулжалиловна
Актуальность
Близорукость является наиболее распространенной аномалией рефракции и ведущей причиной снижения остроты зрения во всем мире
По оценкам, к 2050 году число близоруких увеличится до 4,76 миллиарда человек, а это в свою очередь составляет 49,8% мирового населения. (Holden B. et al. 2016)
Рис.1 Модель визуально регулируемого контроля роста глаз и состояния рефракции
Holden, B. A., Fricke, T. R., Wilson, D. A., Jong, M., Naidoo, K. S., Sankaridurg, P., Wong, T. Y., Naduvilath, T. J., & Resnikoff, S. (2016). Global Prevalence of Myopia and High Myopia and Temporal Trends from 2000 through 2050. Ophthalmology, 123(5), 1036–1042
Хирургическое лечение
«малая» (малоинвазивная) склеропластика
«большая» (бандажирующая) склеропластика
по модифицированной методике Снайдер-Томпсона
Модификации методик М. В. Зайковой, по Н.Н. Пивоварову, по Э.С. Аветисову - Е.П. Тарутте.
Федеральные клинические рекомендации "Миопия" (одобрены Минздравом России) 2024г. [Federal clinical guidelines "Myopia" (approved by the Ministry of Health of the Russian Federation) 2024г.]
Тарутта Е.П., Иомдина Е.Н., Кружкова Г.В., Маркосян Г.А. Отдаленные результаты склерореконструктивного лечения прогрессирующей миопии//Российский офтальмологический журнал. - 2011. - Т. 4, N 1. - С. 71 - 75. [Tarutta E.P., Iomdina E.N., Kruzhkova G.V., Markosyan G.A. Long-term results of scleroreconstructive treatment of progressive myopia//Russian Ophthalmological Journal. - 2011. - vol. 4, N 1. - pp. 71-75.]
Цель исследования – разработать модель машинного обучения для прогнозирования исхода склеропластики у детей через 12 месяцев после операции
Медико-демографические показатели пациентов включенных в набор данных для разработки моделей машинного обучение для бинарной классификации исхода склеропластики через 12 мес после операции представлен в табл. 1.
Таблица 1 - Медико-демографические показатели пациентов
Рисунок 1 – Матрица корреляций по Спирмену между данными до склеропластики и годовым градиентом прогрессирования
Разработка моделей машинного обучения для бинарной классификации прогноза результата склеропластики
Таблица 2 – Метрики качества моделей машинного обучения для бинарной классификации исхода (благоприятный / неблагоприятный) склеропластики у детей через 12 месяцев
Модель Model | Тестовая выборка Test dataset | Обучающая выборка Train dataset | ||||||||
AUC | Accuracy | Precision | Recall | F1 | AUC | Accuracy | Precision | Recall | F1 | |
Extra Trees Classifier | 0,79 | 0,70 | 0,70 | 0,70 | 0,70 | 0,68 | 0,63 | 0,63 | 0,63 | 0,62 |
Linear Discriminant Analysis | 0,78 | 0,73 | 0,73 | 0,73 | 0,72 | 0,64 | 0,59 | 0,59 | 0,59 | 0,59 |
Gradient Boosting Classifier | 0,77 | 0,70 | 0,70 | 0,70 | 0,70 | 0,71 | 0,67 | 0,67 | 0,67 | 0,67 |
Naive Bayes | 0,77 | 0,75 | 0,76 | 0,75 | 0,75 | 0,65 | 0,59 | 0,60 | 0,59 | 0,59 |
Logistic Regression | 0,77 | 0,70 | 0,70 | 0,70 | 0,70 | 0,64 | 0,57 | 0,57 | 0,57 | 0,56 |
CatBoost Classifier | 0,76 | 0,68 | 0,68 | 0,68 | 0,67 | 0,69 | 0,66 | 0,66 | 0,66 | 0,66 |
K Neighbors Classifier | 0,76 | 0,70 | 0,71 | 0,70 | 0,70 | 0,61 | 0,56 | 0,56 | 0,56 | 0,55 |
MLP Classifier | 0,75 | 0,65 | 0,66 | 0,65 | 0,65 | 0,63 | 0,56 | 0,57 | 0,56 | 0,54 |
Decision Tree Classifier | 0,73 | 0,60 | 0,60 | 0,60 | 0,60 | 0,93 | 0,84 | 0,84 | 0,84 | 0,84 |
Quadratic Discriminant Analysis | 0,72 | 0,68 | 0,68 | 0,68 | 0,67 | 0,69 | 0,59 | 0,59 | 0,59 | 0,59 |
Random Forest Classifier | 0,72 | 0,70 | 0,70 | 0,70 | 0,70 | 0,83 | 0,73 | 0,73 | 0,73 | 0,73 |
Ada Boost Classifier | 0,69 | 0,68 | 0,68 | 0,68 | 0,67 | 0,69 | 0,66 | 0,66 | 0,66 | 0,66 |
Light Gradient Boosting Machine | 0,68 | 0,68 | 0,68 | 0,68 | 0,67 | 0,63 | 0,63 | 0,63 | 0,63 | 0,62 |
Gaussian Process Classifier | 0,64 | 0,63 | 0,63 | 0,63 | 0,62 | 0,88 | 0,80 | 0,80 | 0,80 | 0,79 |
Extreme Gradient Boosting | 0,59 | 0,50 | 0,50 | 0,50 | 0,40 | 1,00 | 0,89 | 0,91 | 0,89 | 0,88 |
SVM - Radial Kernel | 0,53 | 0,53 | 0,53 | 0,53 | 0,52 | 0,96 | 0,84 | 0,84 | 0,84 | 0,84 |
Dummy Classifier | 0,45 | 0,45 | 0,45 | 0,45 | 0,45 | 0,43 | 0,43 | 0,43 | 0,43 | 0,43 |
Ridge Classifier | 0,00 | 0,70 | 0,70 | 0,70 | 0,70 | 0,00 | 0,56 | 0,56 | 0,56 | 0,55 |
SVM - Linear Kernel | 0,00 | 0,65 | 0,66 | 0,65 | 0,65 | 0,00 | 0,57 | 0,58 | 0,57 | 0,56 |
Рисунок 2 – ROC кривая модели Extra Trees Classifier для бинарной классификации исхода результата склеропластики у детей через 12 месяцев (0 класс – благоприятный исход, 1 класс – неблагоприятный исход)
Матрица ошибок на наблюдениях тестового набора данных модели Extra Trees Classifier для бинарной классификации исхода результата склеропластики у детей через 12 месяцев представлена на рис. 3. Из данных матрицы видно, что модель в 70% случаев предсказывает верно благоприятный исход и также в 70% случаев верно предсказывает неблагоприятный исход
Разработаны 19 моделей машинного обучения для бинарной классификации исхода склеропластики у детей через 12 месяцев после операции (благоприятный / неблагоприятный исход), среди которых наилучшее качество по метрике AUC показала модель Extra Trees Classifier (AUC 0,79), точность (accuracy) модели прогнозирования исхода составила 0,70.
Рисунок 3 – Матрица ошибок на наблюдениях тестового набора данных модели Extra Trees Classifier для бинарной классификации исхода результата склеропластики у детей через 12 месяцев (0 класс – благоприятный исход, 1 класс – неблагоприятный исход
Результаты