OceanCBM: первая концептуальная модель для интерпретации океанических прогнозов

Исследователи разработали OceanCBM — первую концептуальную модель с узким местом (concept bottleneck model) для пространственно-временного прогнозирования и механистического анализа океанической динамики. Модель нацелена на предсказание теплосодержания верхнего перемешанного слоя — ключевого индикатора морских волн тепла.

В отличие от традиционных подходов «черного ящика», OceanCBM пропускает информацию через промежуточный слой заданных концептов, основанных на геофизической гидродинамике, а также через один свободный концепт. Такая архитектура накладывает мягкие физические ограничения, не перегружая модель, а свободный концепт помогает улавливать остаточные физические процессы и регулировать предсказания.

Для обучения используется смешанный подход (mixed supervision), при котором модель одновременно учится прогнозировать целевую величину и воспроизводить физические концепты. Это, по словам авторов, обеспечивает согласованность механистических представлений при различных инициализациях ансамбля.

Сравнение с базовыми методами показало, что модели, обученные только на предсказание или только на фиксированные концепты, демонстрируют высокую изменчивость скрытых структур, несмотря на сопоставимую предсказательную способность. OceanCBM же достигает интерпретируемых и физически обоснованных представлений без потери точности, явно характеризуя компромисс между интерпретируемостью и производительностью.

Работа опубликована на платформе arXiv. Авторы подчеркивают, что улучшение прозрачности моделей особенно важно для понимания экстремальных океанических явлений, которые сложно не только предсказать, но и диагностировать.