cs.LGJan 31, 2026

Deep Time-series Forecasting Needs Kernelized Moment Balancing

Authors: Licheng Pan, Haocheng Yang, Shuting He, Han Zheng, Yuzhou Chen, Guangyi Liu, Kangning Zhang, Zhixuan Chu, 

Organizations: Xiaohongshu Inc. · National University of Singapore · College of Computer Science and Technology, Zhejiang University

Abstract

Deep time-series forecasting can be formulated as a distribution balancing problem aimed at aligning the distribution of the forecasts and ground truths. According to Imbens' criterion, balancing two distributions requires that their first moments be identical with respect to any balancing function. We demonstrate that existing objectives enforce moment matching only for one or two predefined balancing functions, thus failing to achieve full distribution balance. To address this limitation, we propose direct forecasting with kernelized moment balancing (KMB-DF). Unlike existing objectives, KMB-DF adaptively selects the most informative balancing functions from a reproducing kernel Hilbert space (RKHS) to enforce sufficient distribution balancing. A tractable and differentiable formulation is derived to enable estimation from empirical samples and seamless integration into gradient-based training pipelines. Extensive experiments across multiple models and datasets show that KMB-DF consistently improves forecasting accuracy and achieves state-of-the-art performance. Code is available at https://anonymous.4open.science/r/KMB-DF-403C.

Explore similar work

CardsList
  1. DistDF: Time-Series Forecasting Needs Joint-Distribution Wasserstein Alignment

    Oct 28, 2025Hao Wang, Licheng Pan, Yuan Lu +7Time Series ForecastingOptimal Transport

  2. Learning to Difference: Adaptive Reversible Differencing (AdaRDiff) for Time Series Forecasting

    Aug 28, 2026Morad Laglil, Younes Hlal, Marouane El Hadari +2Time Series ForecastingLong-Term Time Series Forecasting