Искусственный интеллект предсказывает послеродовую депрессию по данным о родах

Послеродовая депрессия — это тяжёлое состояние, которое развивается у многих женщин после родов. Обычные статистические методы не всегда точно предсказывают, у кого оно возникнет. Исследователи решили применить машинное обучение — метод, при котором компьютер сам находит сложные закономерности в данных.

В исследовании участвовали 273 женщины после родов. Собрали информацию о 44 разных показателях: возраст, течение беременности, особенности родов, социально-экономические условия. Затем данные разделили на две части: на одной модель обучалась, на другой проверялась.

Среди девяти разных алгоритмов лучшим оказался «случайный лес» — модель, которая объединяет множество простых решений. Она смогла предсказать депрессию через шесть недель после родов с точностью выше средней. Чтобы понять, почему модель так решила, использовали метод SHAP — он показывает, какие факторы повлияли на прогноз сильнее всего.

Самыми важными оказались незапланированный способ родоразрешения и сильная боль во время родов. А более высокий социально-экономический статус, наоборот, снижал риск. Это не значит, что причина именно в них, но такие признаки помогают врачам обратить внимание на женщин из группы риска.

Авторы подчёркивают: модель нужно проверить на больших группах из разных регионов, добавив биологические и психологические показатели. Пока это только научный инструмент, а не готовое приложение для диагностики.