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