基于决策树桩的最优稀疏恢复
机器学习
2023-03-09 v1 数据结构与算法
机器学习
摘要
决策树因其低计算成本、良好的预测性能以及评估特征重要性的能力而被广泛使用。尽管这些方法常用于特征选择,但其理论保证尚不为人所充分理解。本文针对使用单深度决策树的线性回归中的特征选择问题,给出了一个紧致的有限样本界。我们研究了这些“决策树桩”从 个总特征中恢复 个活跃特征(其中 )的统计性质。我们的分析提供了对高维稀疏系统的紧致样本性能保证,其与 Lasso 所得的 有限样本界一致,改进了先前针对中位数和最优分裂准则的界。我们的结果推广到非线性情形以及任意次高斯分布,表明基于树的方法在多种设置下均具有强大的特征选择性质,并进一步阐释了这些方法在实践中的成功。作为我们分析的一个副产品,我们表明即使活跃特征数 未知,也能在理论上保证恢复。我们进一步通过计算实验验证了我们的理论结果和证明方法。
引用
@article{arxiv.2303.04301,
title = {Optimal Sparse Recovery with Decision Stumps},
author = {Kiarash Banihashem and MohammadTaghi Hajiaghayi and Max Springer},
journal= {arXiv preprint arXiv:2303.04301},
year = {2023}
}
备注
Accepted to AAAI 2023