最优生存树
机器学习
2020-12-09 v1 机器学习
摘要
基于树的模型因能够识别参数模型范围之外的复杂关系而日益流行。生存树方法对这些模型进行调整,以允许对删失结局进行分析,这类结局常出现在医学数据中。我们提出了一种新的最优生存树(Optimal Survival Trees, OST)算法,该算法利用混合整数优化(MIO)和局部搜索技术来生成全局优化的生存树模型。我们证明,OST 算法在准确性上优于现有的生存树方法,尤其是在大型数据集中。
引用
@article{arxiv.2012.04284,
title = {Optimal Survival Trees},
author = {Dimitris Bertsimas and Jack Dunn and Emma Gibson and Agni Orfanoudaki},
journal= {arXiv preprint arXiv:2012.04284},
year = {2020}
}