Feature Importance Measurement based on Decision Tree Sampling

Chao Huang,Diptesh Das,Koji Tsuda
2023-07-25
Abstract:Random forest is effective for prediction tasks but the randomness of tree generation hinders interpretability in feature importance analysis. To address this, we proposed DT-Sampler, a SAT-based method for measuring feature importance in tree-based model. Our method has fewer parameters than random forest and provides higher interpretability and stability for the analysis in real-world problems. An implementation of DT-Sampler is available at <a class="link-external link-https" href="https://github.com/tsudalab/DT-sampler" rel="external noopener nofollow">this https URL</a>.
Machine Learning
What problem does this paper attempt to address?