arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2006.11458cs.LGstat.ML
收藏

基于神经决策树的分类模型族选择

Model family selection for classification using Neural Decision Trees

Centre Borelli(博雷利中心);ENS Paris-Saclay(巴黎萨克雷高等师范学院);Universite Paris-Saclay(巴黎萨克雷大学)

Anthea Mérida Montes de Oca, Argyris Kalogeratos, Mathilde Mougeot

摘要:

模型选择包括根据要优化的度量来比较几个候选模型。这个过程通常涉及网格搜索等,以及交叉验证,这可能很耗时,并且不会提供有关数据集本身的太多信息。在本文中,我们提出了一种方法来减少任务所需的探索范围。其思想是量化需要偏离给定家族的训练实例、具有“刚性”决策边界的参考模型(例如决策树)的程度,以便获得等效或更好的模型。在我们的方法中,这是通过逐步放宽初始决策树(RM)的决策边界来实现的,只要这在分析数据集上测量的性能方面是有益的。更具体地,通过利用神经决策树来执行该松弛,神经决策树是从DT构建的神经网络。我们的方法产生的最终模型带有非线性决策边界。测量最终模型的性能,以及它与其种子RM的一致性,可以帮助用户确定他应该关注哪一系列模型。

英文摘要:

Model selection consists in comparing several candidate models according to a metric to be optimized. The process often involves a grid search, or such, and cross-validation, which can be time consuming, as well as not providing much information about the dataset itself. In this paper we propose a method to reduce the scope of exploration needed for the task. The idea is to quantify how much it would be necessary to depart from trained instances of a given family, reference models (RMs) carrying `rigid' decision boundaries (e.g. decision trees), so as to obtain an equivalent or better model. In our approach, this is realized by progressively relaxing the decision boundaries of the initial decision trees (the RMs) as long as this is beneficial in terms of performance measured on an analyzed dataset. More specifically, this relaxation is performed by making use of a neural decision tree, which is a neural network built from DTs. The final model produced by our method carries non-linear decision boundaries. Measuring the performance of the final model, and its agreement to its seeding RM can help the user to figure out on which family of models he should focus on.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑