AT Loss: Advanced Torrential Loss Function for Precipitation Forecasting
2509.01348v1
cs.LG, cs.AI, physics.ao-ph
2025-09-05
Авторы:
Jaeho Choi, Hyeri Kim, Kwang-Ho Kim, Jaesung Lee
Резюме на русском
## Контекст
Прогнозирование мощности осадков приобретает всё большее значение в контексте изменения климата. Обычные методы, такие как численные модели осадков и климатические модели, часто сталкиваются со сложностями при оценке интенсивных осадков. В этой области машинного обучения набирает обороты как эффективный альтернативный подход. Однако, многие современные модели все ещё полагаются на стандартные функции потерь, нежели на целесообразные подходы, которые могут учитывать конкретные особенности прогноза осадков. Например, критерий критического успеха (CSI), широко используемый в этой области, становится менее эффективным во времена продолжительных засушных периодов. В статье предлагается новый подход, основанный на реинтерпретации критерия CSI, чтобы сделать прогнозирование осадков более точным и устойчивым.
## Метод
В этом исследовании предлагается новая функция потерь, названная AT Loss (Advanced Torrential Loss), которая оптимизирует прогнозирование осадков. Она определяется с помощью реинтерпретации критерия Критического Успеха (CSI) в виде квадратичной задачи неограниченной бинарной оптимизации (QUBO). Эта задача, в свою очередь, релаксируется в дифференцируемую функцию потерь AT через аппроксимацию. Эта функция потерь использует пространство бинарных переменных для моделирования сложностей при прогнозировании осадков, особенно в условиях продолжительных просушных периодов. Эта новая методология предназначена для повышения точности и устойчивости прогноза осадков.
## Результаты
Опыты проводились на большом объёме данных, включая временные ряды погодных наблюдений. Тестирование показало, что AT Loss превосходит стандартные функции потерь, такие как Mean Squared Error (MSE) и похожие критерии, в плане точности прогноза и стабильности. Особенно выдающимися результатами стало прогнозирование осадков во времена продолжительных просушных периодов, где критерий CSI часто теряет эффективность. Данные эксперименты подтверждают, что AT Loss сохраняет свою точность и стабильность даже в таких сложных условиях.
## Значимость
AT Loss может быть применена в различных областях, включая климатическое моделирование, гидрологию и управление безопасностью водных ресурсов. Одним из основных преимуществ является её устойчивость к продолжительным просушным периодам, когда стандартные методы часто теряют эффективность. Эта функция потерь может существенно повысить точность и надежность прогнозов осадков, что имеет решающее значение для принятия решений в сферах, требующих аккуратной оценки климатических рисков.
Abstract
Accurate precipitation forecasting is becoming increasingly important in the
context of climate change. In response, machine learning-based approaches have
recently gained attention as an emerging alternative to traditional methods
such as numerical weather prediction and climate models. Nonetheless, many
recent approaches still rely on off-the-shelf loss functions, and even the more
advanced ones merely involve optimization processes based on the critical
success index (CSI). The problem, however, is that CSI may become ineffective
during extended dry periods when precipitation remains below the threshold,
rendering it less than ideal as a criterion for optimization. To address this
limitation, we introduce a simple penalty expression and reinterpret it as a
quadratic unconstrained binary optimization (QUBO) formulation. Ultimately, the
resulting QUBO formulation is relaxed into a differentiable advanced torrential
(AT) loss function through an approximation process. The proposed AT loss
demonstrates its superiority through the Lipschitz constant, forecast
performance evaluations, consistency experiments, and ablation studies with the
operational model.