Sharing pattern submodels for prediction with missing values
Paper i proceeding, 2023

Missing values are unavoidable in many applications of machine learning and present challenges both during training and at test time. When variables are missing in recurring patterns, fitting separate pattern submodels have been proposed as a solution. However, fitting models independently does not make efficient use of all available data. Conversely, fitting a single shared model to the full data set relies on imputation which often leads to biased results when missingness depends on unobserved factors. We propose an alternative approach, called sharing pattern submodels, which i) makes predictions that are robust to missing values at test time, ii) maintains or improves the predictive power of pattern submodels, and iii) has a short description, enabling improved interpretability. Parameter sharing is enforced through sparsity-inducing regularization which we prove leads to consistent estimation. Finally, we give conditions for when a sharing model is optimal, even when both missingness and the target outcome depend on unobserved variables. Classification and regression experiments on synthetic and real-world data sets demonstrate that our models achieve a favorable tradeoff between pattern specialization and information sharing.


Lena Stempfle

Chalmers, Data- och informationsteknik, Data Science och AI

Ashkan Panahi

Chalmers, Data- och informationsteknik, Data Science och AI

Fredrik Johansson

Chalmers, Data- och informationsteknik, Data Science och AI

Proceedings of the 37th AAAI Conference on Artificial Intelligence

Vol. 37 8 9882-9890
9781577358800 (ISBN)

AAAI Conference on Artificial Intelligence
Washington, DC,, USA,

WASP AI/MLX Forskarassistent

Wallenberg AI, Autonomous Systems and Software Program, 2019-08-01 -- 2023-08-01.


Data- och informationsvetenskap

Sannolikhetsteori och statistik





Mer information

Senast uppdaterat