用核方法自适应匹配分布矩,提升时序预测精度。
Deep Time-series Forecasting Needs Kernelized Moment Balancing
- 从再生核希尔伯特空间中自动选择最优平衡函数
- 在多个数据集上显著优于现有方法,实现领先性能
- 适合需要高精度时序预测的研究者和工程师
深度时序预测可视为分布平衡问题,目标是使预测分布与真实分布对齐。根据Imbens准则,真正的分布平衡需在任意平衡函数下匹配一阶矩。现有目标仅对一个或两个预定义平衡函数进行矩匹配,无法实现完整分布平衡。为此,我们提出直接预测的核化矩平衡方法(KMB-DF)。该方法从再生核希尔伯特空间(RKHS)中自适应选择最有效的平衡函数,实现充分分布平衡。我们推导出可微且可计算的目标函数,支持从经验样本高效估计,并无缝集成至梯度训练流程。在多种模型和数据集上的大量实验表明,KMB-DF持续提升预测精度,达到当前最优水平。代码已公开于 https://anonymous.4open.science/r/KMB-DF-403C。
原文摘要 · Abstract (English)
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, true distribution balance requires matching the first moments with respect to any balancing function. We demonstrate that existing objectives fail to meet this criterion, as they 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. We derive a tractable and differentiable objective that enables efficient 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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。