Learning Binary Decision Trees by Argmin Differentiation

Valentina Zantedeschi, Matt J. Kusner, Vlad Niculae

Abstract: We address the problem of learning binary decision trees that partition data for some downstream task. We propose to learn discrete parameters (i.e., for tree traversals and node pruning) and conti...