Hugging Face Daily PapersGiacomo Fidone, Alessio Cascione, Riccardo Guidotti1 min readpaperadvanced
Learning Sparse Decision Trees via Transformer Variational Auto-Encoders
Summary
The paper presents TREVIS, a method that encodes decision trees into a continuous latent space using a Tree Transformer VAE, allowing gradient-based optimization of both accuracy and structural sparsity. Experiments claim TREVIS matches the predictive performance of near-optimal algorithms while producing sparser trees.
- TREVIS maps discrete decision trees to latent vectors via a Tree Transformer VAE, converting the search space to continuous.
- A differentiable surrogate model enables joint gradient optimization of predictive performance and sparsity.
- Empirical results show TREVIS achieves comparable accuracy to state‑of‑the‑art methods with reduced tree size.
Practitioners building interpretable models will care because it offers a way to enforce sparsity without sacrificing accuracy.
5/10

