A modified Nadaraya-Watson procedure for variable selection and nonparametric prediction with missing data

• Published in: Journal of nonparametric statistics 2024-07, Vol.36 (3), p.825-862
• Main Authors: Cheung, Kin Yap, Lee, Stephen M. S.
• Format: Article
• Language: English
• Subjects: Analysis of covariance; Feature selection; Missing data; Nadaraya-Watson estimator; nonparametric regression; Nonparametric statistics; Predictions; Statistical prediction; variable selection
• ISSN: 1048-5252; 1029-0311
• DOI: 10.1080/10485252.2023.2270079

We propose a new method for variable selection and prediction under a nonparametric regression setting, where a covariate may be missing either because its value is hidden from the observer or because it is inapplicable to the particular subject being observed. Despite its practical relevance, the problem has received little attention in the literature and its solutions are largely non-existent. Our proposal hinges on the construction of a modified Nadaraya-Watson estimator of the conditional mean regression function, with its bandwidths regularised to select variables and its weights adapted to accommodate different types of missingness. The method allows for information sharing across different missing data patterns without affecting consistency of the estimator. Unlike other conventional methods such as those based on imputations or likelihoods, our method requires only mild assumptions on the model and the missingness mechanism. For prediction we focus on finding relevant variables for predicting mean responses, conditional on covariate vectors subject to a given type of missingness. Our theoretical and numerical results show that the new method is consistent in variable selection and yields better prediction accuracy compared to existing methods.