Variational Autoencoding with Conditional Iterative Sampling for Missing Data Imputation

Variational autoencoders (VAEs) are popular for their robust nonlinear representation capabilities and have recently achieved notable advancements in the problem of missing data imputation. However, existing imputation methods often exhibit instability due to the inherent randomness in the sampling...

Full description

Saved in:
Bibliographic Details
Main Authors: Shenfen Kuang, Jie Song, Shangjiu Wang, Huafeng Zhu
Format: Article
Language:English
Published: MDPI AG 2024-10-01
Series:Mathematics
Subjects:
Online Access:https://www.mdpi.com/2227-7390/12/20/3288
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary:Variational autoencoders (VAEs) are popular for their robust nonlinear representation capabilities and have recently achieved notable advancements in the problem of missing data imputation. However, existing imputation methods often exhibit instability due to the inherent randomness in the sampling process, leading to either underestimation or overfitting, particularly when handling complex missing data types such as images. To address this challenge, we introduce a conditional iterative sampling imputation method. Initially, we employ an importance-weighted beta variational autoencoder to learn the conditional distribution from the observed data. Subsequently, leveraging the importance-weighted resampling strategy, samples are drawn iteratively from the conditional distribution to compute the conditional expectation of the missing data. The proposed method has been experimentally evaluated using classical generative datasets and compared with various well-known imputation methods to validate its effectiveness.
ISSN:2227-7390