Imputation of mixed data with multilevel singular value decomposition

TL;DR

This paper introduces a novel multilevel singular value decomposition method for imputing missing values in large, mixed, multilevel datasets, offering computational efficiency and applicability to real-world medical data.

Contribution

It presents the first SVD-based imputation method capable of handling mixed data types in multilevel datasets, improving speed and scalability over existing solutions.

Findings

Outperforms competitors in handling various dataset sizes

Computationally faster than existing methods

Successfully applied to medical data from multiple hospitals

Abstract

Statistical analysis of large data sets offers new opportunities to better understand many processes. Yet, data accumulation often implies relaxing acquisition procedures or compounding diverse sources. As a consequence, such data sets often contain mixed data, i.e. both quantitative and qualitative and many missing values. Furthermore, aggregated data present a natural \textit{multilevel} structure, where individuals or samples are nested within different sites, such as countries or hospitals. Imputation of multilevel data has therefore drawn some attention recently, but current solutions are not designed to handle mixed data, and suffer from important drawbacks such as their computational cost. In this article, we propose a single imputation method for multilevel data, which can be used to complete either quantitative, categorical or mixed data. The method is based on multilevel singular value decomposition (SVD), which consists in decomposing the variability of the data into two components, the between and within groups variability, and performing SVD on both parts. We show on a simulation study that in comparison to competitors, the method has the great advantages of handling data sets of various size, and being computationally faster. Furthermore, it is the first so far to handle mixed data. We apply the method to impute a medical data set resulting from the aggregation of several data sets coming from different hospitals. This application falls in the framework of a larger project on Trauma patients. To overcome obstacles associated to the aggregation of medical data, we turn to distributed computation. The method is implemented in an R package.