arXiv:2411.07506v3 宣告类型: replace-cross 摘要：基于扩散的模型在时间序列生成方面取得了显著的成果，但面临计算效率低下的问题：通过迭代数值求解器解决高维ODEs/SDEs涉及每个样本需要数百到数千次漂移函数评估，导致成本极其高昂。为了解决这一问题，我们提出了一种基于ODE的模型FlowTS，它利用概率空间中的校正流和直线传输。通过学习分布之间的测地路径，FlowTS 通过精确的线性轨迹模拟实现计算效率，从而加快训练和生成速度，并提高性能。我们进一步引入了一种受探索与利用权衡启发的自适应采样策略，平衡了噪声适应性和精度。值得注意的是，FlowTS 使得从不确定生成到条件生成的无缝转换无需重新训练，确保了高效的实际部署。此外，为了增强生成的真实性，FlowTS 集成了趋势与季节性分解、注意注册（用于全局上下文聚合）以及旋转位置嵌入（RoPE，用于位置信息）。在不确定情况下，广泛的实验表明，FlowTS 在 Stock 和 ETTh 数据集上达到了最先进的性能，Context 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 获取。