A0674
Title: Learning robust treatment rules for censored data
Authors: Yifan Cui - Zhejiang University (China) [presenting]
Abstract: There is a fast-growing literature on estimating optimal treatment rules directly by maximizing the expected outcome. In biomedical studies and operations applications, censored survival outcome is frequently observed, in which case the restricted mean survival time and survival probability are of great interest. Two robust criteria are proposed for learning optimal treatment rules with censored survival outcomes; the former one targets an optimal treatment rule maximizing the restricted mean survival time, where the restriction is specified by a given quantile such as median; the latter one targets at an optimal treatment rule maximizing buffered survival probabilities, where the predetermined threshold is adjusted to account the restricted mean survival time. Theoretical justifications are provided for the proposed optimal treatment rules, and a sampling-based difference-of-convex algorithm is developed for learning them. In simulation studies, estimators show improved performance compared to existing methods. The proposed method is also demonstrated using AIDS clinical trial data.