QuantFlow: модель Mamba с федеративным обучением для прогнозирования временных рядов

QuantFlow, вероятностная система прогнозирования временных рядов, которая решает проблему централизации данных, характерную для современных foundation models. Фреймворк объединяет инвертированное встраивание последовательностей, двунаправленные декодеры Mamba (модели пространства состояний), квантильную регрессию и федеративное обучение. Каждая переменная встраивается на полной временной окне, обрабатывается в прямом и обратном направлениях, затем проецируется на пять условных квантилей. TSMixup расширяет временное разнообразие через взвешенную интерполяцию Дирихле. В экспериментах использовались данные криптовалют, трафика, электроэнергии, температуры, гриппа и погоды. QuantFlow достиг MSE 0.2834 на ETTm1 и 0.2218 на Weather. В федеративном развёртывании со 100 клиентами на неоднородных данных система сохранила полезную точность после трёх раундов коммуникации, не централизуя исходные данные. Исследование выявило, что моделирование пространства состояний перспективно как основа для масштабируемого, учитывающего неопределённость прогнозирования с сохранением приватности, но выявило ограничения на нерегулярных эпидемиологических сигналах и долгосрочных прогнозах.

Ключевые факты

  • Mamba (модель пространства состояний) позволяет обрабатывать длинные и многомерные сигналы без ограничений трансформеров
  • Квантильная регрессия даёт полное распределение вероятностей прогноза, а не точечный прогноз
  • Федеративное обучение (20+ клиентов, три коммуникационных раунда) сохраняет приватность и работает с неоднородными данными
  • Тестирование на криптовалютах, электросетях, погоде и эпидемиологических данных показало конкурентные метрики
  • Известные ограничения: слабая работа на нерегулярных эпидемиологических сигналах и долгосрочные прогнозы

Почему это важно

Прогнозирование временных рядов критично для финансов, энергетики, транспорта, здравоохранения и промышленного мониторинга. Современные foundation models обещают лучшую передачу знаний между задачами, но зависят от централизации данных и трансформеров, что ограничивает их применимость к длинным, многомерным и приватным сигналам. QuantFlow решает эту проблему: объединяет эффективное моделирование пространства состояний с федеративным обучением, позволяя работать с распределёнными, неоднородными данными без отправки сырых записей на центральный сервер.

Кому это важно

Финансовым учреждениям (прогноз криптовалют, курсов). Энергетическим компаниям и операторам сетей. Организациям здравоохранения (прогноз заболеваемости). Транспортным системам. Промышленным объектам с дистанционным мониторингом. Любым организациям, владеющим приватными временными рядами и не желающим централизовать данные, регулируемые отрасли, финансовые консорциумы, медицинские сети.

Как это применить

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

Можно ли доверять

Бумага опубликована на arXiv (обычно предшествует формальной рецензии, но сигнал серьёзности). Экспериментальная база широка: криптовалюта, трафик, электроэнергия, температура, грипп, погода, установленные бенчмарки. Приведены конкретные метрики (MSE, коммуникационные раунды) и сравнение производительности. Авторы явно указывают известные ограничения (нерегулярные эпидемиологические сигналы, долгосрочная генерализация), что свидетельствует об ответственности исследования.

Риски и подводные камни

Система плохо работает на нерегулярных эпидемиологических временных рядах (например, вспышки гриппа). Долгосрочные прогнозы требуют дополнительных исследований, текущие результаты ограничены. Федеративное обучение требует нескольких раундов коммуникации (задержка в развёртывании). Архитектура специфична для time-series, не универсальная foundation model, её не применить к табличным или графическим данным без переделки. Квантильная регрессия добавляет вычислительную нагрузку.

«Результаты показывают, что моделирование пространства состояний открывает перспективные возможности для масштабируемого, учитывающего неопределённость прогнозирования временных рядов, сохраняющего приватность»

— QuantFlow: A Federated Mamba-Based Post-Transformer Foundation Model for Time-Series Forecasting