ImputeFormer: Low Rankness-Induced Transformers for Generalizable Spatiotemporal Imputation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Nie, Tong, Qin, Guoyang, Ma, Wei, Mei, Yuewen, Sun, Jian
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909267079987200
author Nie, Tong
Qin, Guoyang
Ma, Wei
Mei, Yuewen
Sun, Jian
author_facet Nie, Tong
Qin, Guoyang
Ma, Wei
Mei, Yuewen
Sun, Jian
contents Missing data is a pervasive issue in both scientific and engineering tasks, especially for the modeling of spatiotemporal data. This problem attracts many studies to contribute to data-driven solutions. Existing imputation solutions mainly include low-rank models and deep learning models. The former assumes general structural priors but has limited model capacity. The latter possesses salient features of expressivity but lacks prior knowledge of the underlying spatiotemporal structures. Leveraging the strengths of both two paradigms, we demonstrate a low rankness-induced Transformer to achieve a balance between strong inductive bias and high model expressivity. The exploitation of the inherent structures of spatiotemporal data enables our model to learn balanced signal-noise representations, making it generalizable for a variety of imputation problems. We demonstrate its superiority in terms of accuracy, efficiency, and versatility in heterogeneous datasets, including traffic flow, solar energy, smart meters, and air quality. Promising empirical results provide strong conviction that incorporating time series primitives, such as low-rankness, can substantially facilitate the development of a generalizable model to approach a wide range of spatiotemporal imputation problems.
format Preprint
id arxiv_https___arxiv_org_abs_2312_01728
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle ImputeFormer: Low Rankness-Induced Transformers for Generalizable Spatiotemporal Imputation
Nie, Tong
Qin, Guoyang
Ma, Wei
Mei, Yuewen
Sun, Jian
Machine Learning
Missing data is a pervasive issue in both scientific and engineering tasks, especially for the modeling of spatiotemporal data. This problem attracts many studies to contribute to data-driven solutions. Existing imputation solutions mainly include low-rank models and deep learning models. The former assumes general structural priors but has limited model capacity. The latter possesses salient features of expressivity but lacks prior knowledge of the underlying spatiotemporal structures. Leveraging the strengths of both two paradigms, we demonstrate a low rankness-induced Transformer to achieve a balance between strong inductive bias and high model expressivity. The exploitation of the inherent structures of spatiotemporal data enables our model to learn balanced signal-noise representations, making it generalizable for a variety of imputation problems. We demonstrate its superiority in terms of accuracy, efficiency, and versatility in heterogeneous datasets, including traffic flow, solar energy, smart meters, and air quality. Promising empirical results provide strong conviction that incorporating time series primitives, such as low-rankness, can substantially facilitate the development of a generalizable model to approach a wide range of spatiotemporal imputation problems.
title ImputeFormer: Low Rankness-Induced Transformers for Generalizable Spatiotemporal Imputation
topic Machine Learning
url https://arxiv.org/abs/2312.01728