English

Debiased Machine Learning for Partially Linear Accelerated Failure Time Models

Methodology 2026-08-08 v1

Abstract

The Cox model remains the default for survival analysis, but the proportional hazards assumption is often violated and hazard ratios can be difficult to interpret. Accelerated failure time (AFT) models provide an intuitive time-scale alternative, yet flexible covariate adjustment while preserving valid inference on a target exposure remains challenging. For the partially linear AFT model under right censoring, a rank-based debiased machine learning (DML) framework remains undeveloped: the rank-based pairwise moment is not Neyman orthogonal and standard cross-fitting does not directly apply to U-statistics. We develop the first such framework by combining an orthogonalized rank-based U-statistic, a censoring-corrected influence function, and block-pairwise cross-fitting, yielding valid inference under flexible nuisance estimation. Simulations and an application to All of Us electronic health record data demonstrate finite-sample performance and practical utility.

Keywords

Cite

@article{arxiv.2608.07841,
  title  = {Debiased Machine Learning for Partially Linear Accelerated Failure Time Models},
  author = {Tomoki Okuno and Sijie Zheng and Brendon Chau and Gang Li and Jin Zhou and Hua Zhou},
  journal= {arXiv preprint arXiv:2608.07841},
  year   = {2026}
}

Comments

57 pages, 3 figures, including supplementary material