Расшифровка МРТ новорождённых с помощью AI
Ручная расшифровка МРТ мозга новорождённого — это кропотливая и крайне сложная работа. При этом упустить маленькую деталь очень легко, если делать это часами, а один такой анализ может занимать до 72-х часов. Яндекс, совместно со студентами ШАДа и экспертами из СПбГПМУ, представил модель, которая решает эту задачу за 3 секунды на CPU. Это поможет врачам анализировать показания за минуты, что позволит увидеть риски ДЦП на ранних стадиях и вовремя начать эффективную терапию.
Главная проблема в MedTech — нехватка размеченных данных. Сначала снимки предразметили open-source моделью BIBSNet. Процесс распараллелили в Docker-контейнерах на 20 машин, а полученную разметку уже проверяли и доводили до ума врачи СПбГПМУ.
Далее эмпирическим путём была выбрана архитектура U-Net с бэкбоном ResNeXt-50 и функцией потерь DiceLoss.
В итоге из-за легковесной архитектуры инференс занимает ~3 секунды на CPU. Это в 50 раз быстрее аналогов и того же BIBSNet, что сильно упрощает жизнь клиникам без бюджета на GPU. Качество при этом остаётся на уровне 0.703 по IoU, что делает сегментацию хорошей отправной точкой для врача.
После завершения тестирования разработку планируют выложить в open source, чтобы её могли использовать в медицинских и других проектах по всему миру.
Хабр
@ai_newz