UPDF AI

Self-attention-based Diffusion Model for Time-series Imputation

Mohammad Rafid Ul Islam,Prasad Tadepalli,Alan Fern

2024 · DOI: 10.1609/aaaiss.v4i1.31827
1 Citations

TLDR

A diffusion-based generative model that captures feature and time correlations through a two-stage imputation process that outperforms state-of-the-art imputation methods and is more scalable in GPU resources is proposed.

Abstract

Time-series modeling is essential for applications in agriculture, weather forecasting, food production, and more. However, missing data due to sensor malfunctions, power outages, and human errors is a common issue, complicating the training of machine learning models. We propose a diffusion-based generative model to

address this problem and fill the gaps in the data. Our approach captures feature and time correlations through a two-stage imputation process. Our model outperforms state-of-the-art imputation methods and is more scalable in GPU resources.