论文标题
生存:一种解释机器学习生存模型的方法
SurvLIME: A method for explaining machine learning survival models
论文作者
论文摘要
提出了一种称为Survlime的新方法,用于解释机器学习生存模型。它可以看作是众所周知的方法石灰的扩展或修改。提出的方法背后的主要思想是应用COX比例危害模型,以近似测试示例的局部地区的生存模型。之所以使用COX模型,是因为它考虑了示例协变量的线性组合,因此可以将协变量的系数视为对预测的定量影响。另一个想法是通过在关注点附近的局部区域中使用一组扰动点来近似解释模型和COX模型的累积危害函数。该方法减少为解决无约束的凸优化问题。许多数值实验证明了生存效率。
A new method called SurvLIME for explaining machine learning survival models is proposed. It can be viewed as an extension or modification of the well-known method LIME. The main idea behind the proposed method is to apply the Cox proportional hazards model to approximate the survival model at the local area around a test example. The Cox model is used because it considers a linear combination of the example covariates such that coefficients of the covariates can be regarded as quantitative impacts on the prediction. Another idea is to approximate cumulative hazard functions of the explained model and the Cox model by using a set of perturbed points in a local area around the point of interest. The method is reduced to solving an unconstrained convex optimization problem. A lot of numerical experiments demonstrate the SurvLIME efficiency.