A Better Decision Tree: The Max-Cut Decision Tree with Modified PCA Improves Accuracy and Running Time
A Better Decision Tree: The Max-Cut Decision Tree with Modified PCA Improves Accuracy and Running Time
复制标题
更好的决策树:采用改进的 PCA 的最大割决策树提高了准确性和运行时间
DOI:
10.1007/s42979-022-01147-4
复制
发表时间:
2022
期刊:
影响因子:
--
通讯作者:
Hochbaum, Dorit S.
中科院分区:
文献类型:
--
作者:
Bodine, Jonathan;Hochbaum, Dorit S.
Decision trees are a widely used method for classification, both alone and as the building blocks of multiple different ensemble learning methods. The Max Cut decision tree introduced here involves novel modifications to a standard, baseline variant of a classification decision tree, CART Gini. One modification involves an alternative splitting metric, Max Cut, based on maximizing the distance between all pairs of observations that belong to separate classes and separate sides of the threshold value. The other modification, Node Means PCA, selects the decision feature from a linear combination of the input features constructed using an adjustment to principal component analysis (PCA) locally at each node. Our experiments show that this node-based, localized PCA with the Max Cut splitting metric can dramatically improve classification accuracy while also significantly decreasing computational time compared to the CART Gini decision tree. These improvements are most significant for higher-dimensional datasets. For the example dataset CIFAR-100, the modifications enabled a 49% improvement in accuracy, relative to CART Gini, while providing aspeed up compared to the Scikit-Learn implementation of CART Gini. These introduced modifications are expected to dramatically advance the capabilities of decision trees for difficult classification tasks.
DOI:
--
发表时间:
2018
期刊:
影响因子:
--
作者:
Aung Nway Oo
通讯作者:
Aung Nway Oo
DOI:
--
发表时间:
2020
期刊:
International Conference on Knowledge Discovery and Information Retrieval
影响因子:
--
作者:
Jonathan Bodine;D. Hochbaum
通讯作者:
D. Hochbaum
DOI:
--
发表时间:
2014
期刊:
2014 IEEE International Conference on Big Data (Big Data)
影响因子:
--
作者:
D. Hochbaum;P. Baumann
通讯作者:
P. Baumann