Imputation with Inter-Series Information from Prototypes for Irregular Sampled Time Series
Zhihao Yu, Xu Chu, Liantao Ma, Yasha Wang, Wenwu Zhu
TL;DR
This work tackles imputing irregularly sampled multivariate time series by leveraging inter-series information through learned prototypes. It introduces PRIME, composed of a Prototype Memory, a bidirectional Prototype Gated Recurrent Unit (P-GRU), and a Prototype Refinement module, with dedicated prototype losses to maintain diversity and alignment with series representations. Empirical results on three real-world datasets show PRIME achieving up to $>26\%$ relative improvement in MSE over state-of-the-art methods, and robustness to observation rate and prototype count, while mitigating memorization effects. The approach offers a principled way to exploit cross-series patterns for improved imputation, with potential downstream benefits in healthcare, meteorology, and related domains.
Abstract
Irregularly sampled time series are ubiquitous, presenting significant challenges for analysis due to missing values. Despite existing methods address imputation, they predominantly focus on leveraging intra-series information, neglecting the potential benefits that inter-series information could provide, such as reducing uncertainty and memorization effect. To bridge this gap, we propose PRIME, a Prototype Recurrent Imputation ModEl, which integrates both intra-series and inter-series information for imputing missing values in irregularly sampled time series. Our framework comprises a prototype memory module for learning inter-series information, a bidirectional gated recurrent unit utilizing prototype information for imputation, and an attentive prototypical refinement module for adjusting imputations. We conducted extensive experiments on three datasets, and the results underscore PRIME's superiority over the state-of-the-art models by up to 26% relative improvement on mean square error.
