arXiv · 1810.00974
Neural Regression Trees
Abstract
Regression-via-Classification (RvC) is the process of converting a regression problem to a classification one. Current approaches for RvC use ad-hoc discretization strategies and are suboptimal. We propose a neural regression tree model for RvC. In this model, we employ a joint optimization framework where we learn optimal discretization thresholds while simultaneously optimizing the features for each node in the tree. We empirically show the validity of our model by testing it on two challenging regression tasks where we establish the state of the art.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Shahan Ali Memon, Wenbo Zhao, Bhiksha Raj, Rita Singh. 2019-04-03. Neural Regression Trees. https://doi.org/10.1109/ijcnn.2019.8852133
Cite the original work for its findings. Save a collection to share your selection of sources.