Zyphra выпустила ZUNA1.1 — открытую ИИ-модель для расшифровки и восстановления ЭЭГ любой длины
Zyphra представила ZUNA1.1 — улучшенную версию модели для обработки электроэнцефалограмм (ЭЭГ). Главное новшество: модель теперь работает с записями произвольной длины от 0,5 до 30 секунд, обучена на более разнообразных данных и лучше восстанавливает сигналы с повреждённых или отсутствующих электродов. При этом точность не упала, а в некоторых задачах даже выросла.
Это первая открытая foundation-модель для ЭЭГ, которая реально работает с грязными клиническими данными любой конфигурации. Исследователи и стартапы получили инструмент, который раньше был доступен только крупным лабораториям — и могут запустить его на своём железе.
Zyphra выпустила ZUNA1.1 — открытую ИИ-модель для расшифровки и восстановления ЭЭГ любой длины
Zyphra представила ZUNA1.1 — обновлённую версию своей открытой модели для обработки электроэнцефалограмм (ЭЭГ) под лицензией Apache 2.0. Это не революция в точности, а серьёзный рост гибкости: модель теперь работает с записями длиной от 0,5 до 30 секунд, тогда как предыдущая версия принимала только строго пятисекундные фрагменты.
Что это такое
ZUNA1.1 — трансформер с 380 млн параметров, обученный восстанавливать, шумоподавлять и дополнять ЭЭГ-сигналы. Модель умеет:
- Реконструировать отсутствующие каналы (электроды)
- Предсказывать сигналы для новых точек на голове по их координатам
- Работать с любыми конфигурациями электродов — от простых повязок с четырьмя датчиками до исследовательских шапок на 256 каналов
Важная фишка — позиционное кодирование 4D (x, y, z, время). Модель запоминает не номер канала, а его физическое положение на черепе. Поэтому она не привязана к конкретной схеме расположения электродов и может генерировать сигналы для точек, которые вообще не записывались.
Что изменилось
Архитектура почти та же, изменения — в обучении:
-
Переменная длина входа: модель теперь принимает отрезки от 0,5 до 30 секунд (нарезается на токены по 0,125 с). При обучении использовалась упаковка нескольких примеров в один батч с маской внимания.
-
Четыре типа пропусков данных вместо одного: целые каналы, временные промежутки, локализованные пропуски в пространстве-времени, случайные точечные дыры. Это учит модель справляться с реальными артефактами.
-
Расширенный датасет: ~3,5 млн канало-часов против ~2 млн. Качество теперь оценивается не по всей записи, а посекундно для каждого канала — меньше выбрасывается полезных данных.
Результаты
На тестовых наборах ZUNA1.1 показала равную или лучшую точность по сравнению с ZUNA1 (метрика NMSE). Обе версии значительно превосходят классический метод сферической сплайн-интерполяции из библиотеки MNE.
В более реалистичном сценарии — восстановление всех электродов одной области мозга по сигналам остальных семи областей — ZUNA1.1 обогнала и ZUNA1, и сплайны.
Как пользоваться
Модель весит 380 млн параметров, работает на потребительской видеокарте, для многих задач достаточно CPU. Установка:
pip install zuna
Веса доступны на Hugging Face, код — на GitHub. Zyphra также запустила бесплатный веб-демо EEG Playground. Всё это — для исследовательского использования.
Итог: ZUNA1.1 — это не прорыв в алгоритмах, а грамотная инженерная работа, которая сделала модель по-настоящему применимой к реальным, грязным ЭЭГ-данным любой длины и конфигурации.
Ключевые выводы
- ZUNA1.1 принимает записи произвольной длины (0,5–30 с), что критично для реальных клинических данных — раньше приходилось резать всё на пятисекундные куски
- Позиционное кодирование 4D позволяет модели работать с любой схемой электродов и предсказывать сигналы для точек, которые не записывались
- Модель обучена на четырёх типах пропусков данных, имитирующих реальные артефакты: битые каналы, временные разрывы, локализованные пропуски, случайные шумы
- Датасет вырос с 2 до 3,5 млн канало-часов благодаря посекундной фильтрации качества вместо отбраковки целых записей
- 380 млн параметров — это достаточно компактно для запуска на потребительском железе, но достаточно для превосходства над классическими методами
Автор: Ксения Лаврова · Источник: marktechpost.com
**Почему это круто:** ЭЭГ — один из самых доступных способов заглянуть в работу мозга, но данные всегда грязные: электроды отваливаются, пациент моргает, запись обрывается. ZUNA1.1 — первая открытая модель, которая спроектирована под этот хаос, а не под идеальные лабораторные условия. Переменная длина входа, работа с любыми схемами электродов, обучение на реальных артефактах — это то, что нужно практикам.
**Чему верить с осторожностью:** Zyphra честно пишет «research use only», и это не формальность. Модель пока не валидирована клинически, и непонятно, как она поведёт себя с редкими патологиями или экзотическими монтажами. Плюс, обучение на «разнообразных фильтрах» заявлено как цель, но не как доказанный результат. Тем не менее, сам факт появления такого инструмента в открытом доступе — это сдвиг: foundation-модели для медицинских сигналов перестали быть закрытой игрушкой корпораций.
Комментарии