arrow
返回

SDTR: Soft Decision Tree Regressor for Tabular Data

delete2021-01-01
delete33
delete
OA
AI
H
Haoran Luo
F
Fan Cheng *
H
Heng Yu
Y
Yuqi Yi
DOI:10.1109/ACCESS.2021.3070575delete
delete原文链接
delete分享
delete收藏
查看原文
摘要

摘要

En 中文
Deep neural networks have been proved a success in multiple fields. However, researchers still favor traditional approaches to obtain more interpretable models, such as Bayesian methods and decision trees when processing heterogeneous tabular data. Such models are hard to differentiate, thus inconvenient to be integrated into end-to-end settings. On the other hand, traditional neural networks are differentiable but perform poorly on tabular data. We propose a hierarchical differentiable neural regression model, Soft Decision Tree Regressor (SDTR). SDTR imitates a binary decision tree by a differentiable neural network and is plausible for ensemble schemes like bagging and boosting. The SDTR method was evaluated on multiple tabular-based regression tasks (YearPredictionMSD, MSLR-Web10K, Yahoo LETOR, SARCOS and Wine quality). Its performance is comparable with non-differentiable models (gradient boosting decision trees) and better than uninterpretable models (regular FCNN). On top of that, it can produce fair results with a restricted number of parameters, only using a small forest or even a single tree. We also propose an average entropy metric to evaluate the level of interpretability of a trained, soft decision tree neural network. This metric also helps to select proper structure and hyperparameters for such networks.
Keyword:
Regression tree analysis
Neural networks
Vegetation
Task analysis
Data models
Boosting
Measurement
Decision trees
neural networks
machine learning algorithms
AI总结

AI总结

对已上传原文的论文进行重点信息的提取,主要内容包括:简要概述、研究摘要、背景介绍、关键亮点、图文解析、展望与总结。

期刊

IEEE Access 封面图
IEEE Access
IF:
3.6
论文数:
9.8W
被引数:
29.4W

机构

S
shanghai jiao tong university
学者数:
15.7W
论文数: 11.7W
被引数: 159
引用论文

引用论文

err分享
err收藏
Predicting stock returns by classifier ensembles通过分类器集成预测股票收益
err2011-03-01
err123
PREAI
errTsai, Chih-Fong; Lin, Yuah-Chiao; Yen, David C.; Chen, Yan-Min
err分享
err收藏
Purification of Transcripts and Metabolites from <em>Drosophila</em> Heads
err2013-03-15
err0
errOAAI
errKurt Jensen; Jonatan Sanchez-Garcia; Caroline Williams; Swati Khare; Krishanu Mathur; Rita M. Graze; Daniel A. Hahn; Lauren M. McIntyre; Diego E. Rincon-Limas; Pedro Fernandez-Funez
err分享
err收藏
A national-scale GIS-based system for modelling impacts of land use on water quality
err2016-12-01
err0
PREAI
errAlexander H. Elliott; Annette F. Semadeni-Davies; Ude Shankar; John R. Zeldis; David M. Wheeler; David R. Plew; Gerald J. Rys; Simon R. Harris
err分享
err收藏
学者 查看更多内容