Murdoch University Research Repository

Welcome to the Murdoch University Research Repository

The Murdoch University Research Repository is an open access digital collection of research
created by Murdoch University staff, researchers and postgraduate students.

Learn more

A reinforcement learning-based approach for imputing missing data

Awan, S.E., Bennamoun, M., Sohel, F., Sanfilippo, F. and Dwivedi, G. (2022) A reinforcement learning-based approach for imputing missing data. Neural Computing and Applications .

[img]
Preview
Free to read: https://doi.org/10.1007/s00521-022-06958-3
*No subscription required

Abstract

Missing data is a major problem in real-world datasets, which hinders the performance of data analytics. Conventional data imputation schemes such as univariate single imputation replace missing values in each column with the same approximated value. These univariate single imputation techniques underestimate the variance of the imputed values. On the other hand, multivariate imputation explores the relationships between different columns of data, to impute the missing values. Reinforcement Learning (RL) is a machine learning paradigm where the agent learns by taking actions and receiving rewards in response, to achieve its goal. In this work, we propose an RL-based approach to impute missing data by learning a policy to impute data through an action-reward-based experience. Our approach imputes missing values in a column by working only on the same column (similar to univariate single imputation) but imputes the missing values in the column with different values thus keeping the variance in the imputed values. We report superior performance of our approach, compared with other imputation techniques, on a number of datasets.

Item Type: Journal Article
Murdoch Affiliation(s): IT, Media and Communications
Publisher: Springer London
Copyright: © 2022 The Authors.
URI: http://researchrepository.murdoch.edu.au/id/eprint/63996
Item Control Page Item Control Page

Downloads

Downloads per month over past year