Skip to yearly menu bar Skip to main content

Spotlight Poster

VaRT: Variational Regression Trees

Sebastian Salazar

Great Hall & Hall B1+B2 (level 1) #2009


Decision trees are a well-established tool in machine learning for classification and regression tasks. In this paper, we introduce a novel non-parametric Bayesian model that uses variational inference to approximate a posterior distribution over the space of stochastic decision trees. We evaluate the model's performance on 18 datasets and demonstrate its competitiveness with other state-of-the-art methods in regression tasks. We also explore its application to causal inference problems. We provide a fully vectorized implementation of our algorithm in PyTorch.

Chat is not available.