4 个月前

BRITS:双向递归时间序列插补

BRITS:双向递归时间序列插补

摘要

时间序列在许多分类/回归任务中被广泛用作信号。时间序列中普遍存在大量缺失值的情况。给定多个相关的时间序列数据,如何填补这些缺失值并预测其类别标签?现有的插补方法通常对底层数据生成过程施加强假设,例如状态空间中的线性动态。本文提出了一种基于循环神经网络(Recurrent Neural Networks, RNN)的新方法——BRITS,用于时间序列数据中缺失值的插补。我们提出的方法直接在一个双向循环动态系统中学习缺失值,无需任何特定假设。插补的值被视为RNN图中的变量,并可以在反向传播过程中有效更新。BRITS具有三个优势:(a) 能够处理时间序列中的多个相关缺失值;(b) 可以推广到具有非线性动态特性的时序数据;(c) 提供了一种数据驱动的插补程序,并适用于一般性的缺失数据场景。我们在三个真实世界的数据集上评估了我们的模型,包括空气质量数据集、医疗保健数据和人类活动定位数据。实验结果表明,我们的模型在插补和分类/回归准确性方面均优于现有最先进方法。

代码仓库

基准测试

基准方法指标
multivariate-time-series-forecasting-on-ushcnBRITS
MSE: 0.53
multivariate-time-series-imputation-onBRITS
MAE (PM2.5): 11.56
multivariate-time-series-imputation-on-1BRITS
MAE (10% of data as GT): 0.281
multivariate-time-series-imputation-on-2BRITS (SingleRes)
OOB Rate (10^−3) : 3.874
Path Difference: 0.571
Path Length: 0.702
Player Distance : 0.417
Step Change (10^−3): 4.811
multivariate-time-series-imputation-on-pemsBRITS (SingleRes)
L2 Loss (10^-4): 4.51
multivariate-time-series-imputation-on-uciBRITS
MAE (10% missing): 0.219
traffic-data-imputation-on-metr-la-pointBRITS
MAE: 2.34
traffic-data-imputation-on-pems-bay-pointBRITS
MAE: 1.47

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供