Zyphra ZUNA: Новая модель для анализа данных ЭЭГ

Ученые из Zyphra разработали ZUNA – фундаментную модель с 380 миллионами параметров, предназначенную для анализа сигналов ЭЭГ. Модель способна к восстановлению и улучшению качества данных, независимо от конфигурации электродов. ZUNA построена на основе диффузионного автоэнкодера и использует пространственно-временное кодирование.

Особенность ZUNA в том, что она не привязана к фиксированному расположению электродов, в отличие от многих существующих моделей. Это позволяет ей эффективно работать с разными наборами данных и адаптироваться к новым условиям записи. Модель обучалась на огромном объеме данных – более 2 миллиона часов записей ЭЭГ из 208 общедоступных источников.

В тестах ZUNA показала себя лучше, чем традиционные методы интерполяции, особенно при большом количестве отсутствующих данных. Разработчики выложили веса модели под лицензией Apache-2.0, а также совместимый с MNE-Python стек для работы с ней.