Using decision trees to understand structure in missing data

Nicholas J. Tierney, Fiona A. Harden, Maurice J. Harden, Kerrie L Mengersen

Research output: Contribution to journalArticleResearchpeer-review

8 Citations (Scopus)

Abstract

Objectives: Demonstrate the application of decision trees - classification and regression trees (CARTs), and their cousins, boosted regression trees (BRTs) - to understand structure in missing data. Setting: Data taken from employees at 3 different industrial sites in Australia. Participants: 7915 observations were included. Materials and methods: The approach was evaluated using an occupational health data set comprising results of questionnaires, medical tests and environmental monitoring. Statistical methods included standard statistical tests and the 'rpart' and 'gbm' packages for CART and BRT analyses, respectively, from the statistical software 'R'. A simulation study was conducted to explore the capability of decision tree models in describing data with missingness artificially introduced. Results: CART and BRT models were effective in highlighting a missingness structure in the data, related to the type of data (medical or environmental), the site in which it was collected, the number of visits, and the presence of extreme values. The simulation study revealed that CART models were able to identify variables and values responsible for inducing missingness. There was greater variation in variable importance for unstructured as compared to structured missingness. Discussion: Both CART and BRT models were effective in describing structural missingness in data. CART models may be preferred over BRT models for exploratory analysis of missing data, and selecting variables important for predicting missingness. BRT models can show how values of other variables influence missingness, which may prove useful for researchers. Conclusions: Researchers are encouraged to use CART and BRT models to explore and understand missing data.

Original languageEnglish
Article numbere007450
Number of pages11
JournalBMJ Open
Volume5
Issue number6
DOIs
Publication statusPublished - 2015
Externally publishedYes

Cite this