Shibal Ibrahim

End-to-end Feature Selection Approach for Learning Skinny Trees

2023 · arXiv (Cornell University)

We propose a new optimization-based approach for feature selection in tree ensembles, an important problem in statistics and machine learning. Popular tree ensemble toolkits e.g., Gradient Boosted Trees and Random Forests support feature selection post-training …