dlmtree
dlmtree is an R package that provides constrained distributed lag
models (DLMs) using a regression tree approach within the Bayesian
additive regression trees (BART) framework, referred to as treed DLMs.
The package includes various extensions of treed DLMs, allowing for the
incorporation of different scenarios like linear, non-linear
associations, mixture exposures, and heterogeneous exposure effects. The
package is built user-friendly with a single function with three
arguments to specify treed DLMs. Functions for summarizing the model fit
and visualization are also provided.
Treed DLM Overview
| Model | Type | Family | Mixture | Heterogeneity |
|---|---|---|---|---|
| Treed distributed lag model (TDLM)2 | Linear | Gaussian | X | X |
| Binary | X | X | ||
| Count | X | X | ||
| Treed distributed lag mixture model (TDLMM)2 | Linear | Gaussian | O | X |
| Binary | O | X | ||
| Count | O | X | ||
| Treed distributed non-linear lag model (TDLNM)1, 4 | Non-linear | Gaussian | X | X |
| Binary | X | X | ||
| Monotone | Gaussian | X | X | |
| Binary | X | X | ||
| Heterogeneous distributed lag model (HDLM)3 | Linear | Gaussian | X | O |
| Heterogeneous distributed lag mixture model (HDLMM) | Linear | Gaussian | O | O |
Model Selection Guide
Installation
Installing package from GitHub:
# install.packages("devtools")
devtools::install_github("danielmork/dlmtree")
library(dlmtree)Installing package from CRAN:
install.packages("dlmtree")
library(dlmtree)References
The following paper describes this package, including a high-level overview of methods, R syntax and examples.
- Im, S., Wilson, A. and Mork, D. (In Press). “Structured Bayesian Regression Tree Models for Estimating Distributed Lag Effects: The R Package dlmtree.” The R Journal (arXiv preprint)
The majority of methods implemented in this package are described in the following methods papers as well as some on going work.
Mork, D. and Wilson, A. (2022). “Treed distributed lag nonlinear models.” Biostatistics, 23(3), 754–771 (DOI: 10.1093/biostatistics/kxaa051, arXiv preprint)
Mork, D. and Wilson, A. (2023). “Estimating perinatal critical windows of susceptibility to environmental mixtures via structured Bayesian regression tree pairs.” Biometrics, 79(1), 449-461 (DOI: 10.1111/biom.13568, arXiv preprint)
Mork, D., Kioumourtzoglou, M. A., Weisskopf, M., Coull, B. A., and Wilson, A. (2024). “Heterogeneous Distributed Lag Models to Estimate Personalized Effects of Maternal Exposures to Air Pollution.” Journal of the American Statistical Association, 119(545), 14-26 (DOI: 10.1080/01621459.2023.2258595, arXiv preprint)
Mork, D. and Wilson, A. (In press). “Incorporating prior information into distributed lag nonlinear models with zero-inflated monotone regression trees.” Bayesian Analysis. (DOI: 10.1214/23-BA1412, arXiv preprint)