神经回归树
机器学习
2020-02-13 v2 机器学习
摘要
回归转分类（RvC）是将回归问题转换为分类问题的过程。当前RvC方法使用临时的离散化策略，并非最优。我们提出了一种用于RvC的神经回归树模型。在该模型中，我们采用联合优化框架，在学习最优离散化阈值的同时，优化树中每个节点的特征。我们通过在两个具有挑战性的回归任务上测试该模型，实证展示了其有效性，并确立了最先进水平（SOTA）。
引用
@article{arxiv.1810.00974,
title = {Neural Regression Trees},
author = {Shahan Ali Memon and Wenbo Zhao and Bhiksha Raj and Rita Singh},
journal= {arXiv preprint arXiv:1810.00974},
year = {2020}
}
备注
Accepted by The 2019 International Joint Conference on Neural Networks (IJCNN). To be published on IEEE. 8 pages, 4 figures