Prediction with missing data via Bayesian additive regression trees
From MaRDI portal
Abstract: We present a method for incorporating missing data in non-parametric statistical learning without the need for imputation. We focus on a tree-based method, Bayesian Additive Regression Trees (BART), enhanced with "Missingness Incorporated in Attributes," an approach recently proposed incorporating missingness into decision trees (Twala, 2008). This procedure takes advantage of the partitioning mechanisms found in tree-based models. Simulations on generated models and real data indicate that our proposed method can forecast well on complicated missing-at-random and not-missing-at-random models as well as models where missingness itself influences the response. Our procedure has higher predictive performance and is more stable than competitors in many cases. We also illustrate BART's abilities to incorporate missingness into uncertainty intervals and to detect the influence of missingness on the model fit.
Recommendations
Cites work
- An investigation of missing data methods for classification trees applied to binary response data
- BART: Bayesian additive regression trees
- Bayesian Analysis of Binary and Polychotomous Response Data
- scientific article; zbMATH DE number 1294360 (Why is no real title available?)
- Monte Carlo sampling methods using Markov chains and their applications
- Pattern-Mixture Models for Multivariate Incomplete Data
- Random forests
- Statistical modeling: The two cultures. (With comments and a rejoinder).
- Stochastic gradient boosting.
- Stochastic Relaxation, Gibbs Distributions, and the Bayesian Restoration of Images
- Variable selection for BART: an application to gene regulation
Cited in
(31)- Bayesian additive regression trees using Bayesian model averaging
- Bayesian additive regression trees with model trees
- Racing trees to query partial data
- Nowcasting in a pandemic using non-parametric mixed frequency VARs
- Inference in Bayesian additive vector autoregressive tree models
- Heterogeneous causal effects with imperfect compliance: a Bayesian machine learning approach
- Discovering interactions using covariate informed random partition models
- bartMachine
- Learning algorithms to evaluate forensic glass evidence
- Propensity score estimation using classification and regression trees in the presence of missing covariate data
- Rhapsody in fractional
- Smoothing and adaptation of shifted Pólya tree ensembles
- Tree-based algorithms for missing data imputation
- Prediction and Inference With Missing Data in Patient Alert Systems
- Boosting with missing predictors
- Variable selection for BART: an application to gene regulation
- An ensemble learning method for variable selection: application to high-dimensional data and missing values
- Clustering and Prediction With Variable Dimension Covariates
- BiMM tree: a decision tree method for modeling clustered and longitudinal binary outcomes
- BART-based inference for Poisson processes
- Understanding the effect of contextual factors and decision making on team performance in Twenty20 cricket: an interpretable machine learning approach
- Ordered probit Bayesian additive regression trees for ordinal data
- Performance of variable and function selection methods for estimating the nonlinear health effects of correlated chemical mixtures: a simulation study
- An integrated Bayesian framework for multi-omics prediction and classification
- Regression with variable dimension covariates
- Local Gaussian Process Extrapolation for BART Models with Applications to Causal Inference
- A Projection Approach to Local Regression with Variable-Dimension Covariates
- On the consistency of supervised learning with missing values
- Bayesian additive regression trees and the general BART model
- Multiple imputation under missing not at random: incorporating response indicators into sequential imputation
- Improving survey inference in two-phase designs using Bayesian machine learning
This page was built for publication: Prediction with missing data via Bayesian additive regression trees
Report a bug (only for logged in users!)Click here to report a bug for this page (MaRDI item Q5256379)