arXiv:2411.07506v2 宣告类型: 替换-交叉 摘要：基于扩散的模型在时间序列生成方面取得了显著成果，但效率低下：通过迭代数值求解器解决高维ODEs/SDEs需要每次样本数百到数千次漂移函数评估，导致成本高昂。为了解决这一问题，我们提出了FlowTS，这是一种基于ODE的模型，利用概率空间中的整流流和直线运输。通过学习分布之间的测地线路径，FlowTS 通过精确的线性轨迹模拟实现计算效率，从而加速训练和生成并改进性能。我们还引入了一种受探索-利用权衡启发的自适应采样策略，平衡噪声适应和精度。值得注意的是，FlowTS 能够无缝地从无条件生成切换到有条件生成而无需重新训练，确保高效的现实世界部署。此外，为了增强生成的真实感，FlowTS 集成了趋势和季节性分解、注意力注册（用于全局上下文聚合）以及旋转位置嵌入（RoPE，用于位置信息）。在无条件设置中，广泛实验表明，FlowTS 达到了最先进的性能，在Stock和ETTh数据集中，上下文FID得分为0.019和0.011（之前最佳为0.067和0.061）。在有条件设置中，我们在太阳 forecasting（MSE 213，之前最佳为375）和MuJoCo插补任务（MSE 7e-5，之前最佳为2.7e-4）中实现了优越的性能。代码可在 https://github.com/UNITES-Lab/FlowTS 获取。