StableSleep: Source-Free Test-Time Adaptation for Sleep Staging with Lightweight Safety Rails

#### Введение Sleep staging — процесс классификации стадий сна на основе электроэнцефалограммы (EEG) — является ключевым компонентом диагностики сна. Однако модели, обученные на одной популяции, часто неэффективны при использовании на пациентах с разными физиологическими характеристиками или записью. Это приводит к расхождениям в классификации, что может снизить точность диагностики. #### Контекст Проблема кросс-пациентской адаптации в sleep staging становится важной в свете увеличения доступности телемедицины и ночных мониторингов. Использование моделей, обученных на одной популяции, часто приводит к неточным выводам при применении к незнакомым пациентам. Это ограничивает развертывание моделей в реальном мире. Авторы предлагают рецепт для test-time adaptation (TTA), который позволяет модели изменяться во время работы, не требуя доступа к исходным данным. #### Метод Метод StableSleep основывается на минимизации информационной индикаторной функции (entropy minimization) в сочетании с Batch-Norm statistic refresh. Два дополнительных механизма, называемых **entropy gate** и **EMA-based reset**, обеспечивают устойчивость. - **Entropy gate** приостанавливает адаптацию на ненадежных фрагментах (например, во время периодов открытых глаз). - **EMA-based reset** возвращает модель к начальному состоянию при заметном переключении. Это позволяет минимизировать снижение качества классификации во время интерактивной работы. #### Резюме Использованные данные: Sleep-EDF Expanded — набор данных с сотнями пациентов. Результаты: Модель StableSleep показала значимый выигрыш в точности классификации по сравнению с традиционными моделями. Она обеспечила высокую точность и минимальное замедление во время работы. Особо отмечено, что StableSleep требует минимум ресурсов и может работать на устройствах с низким вычислительным мощностью. #### Значимость Приложение StableSleep может быть применено в медицинских устройствах, таких как ночные мониторы для сна, чтобы повысить точность диагностики. Он может быть использован для мониторинга сна в реальном времени, даже у пациентов с нестандартными физиологическими характеристиками. #### Выводы StableSleep — первый подход, оптимизированный для стабильной и эффективной адаптации моделей sleep staging во время работы. Он обеспечивает минимальную задержку и низкое потребление ресурсов. Будущие исследования будут сфокусированы на улучшении механизмов адаптации и расширении применения на другие данные.