Table of Contents
Fetching ...

Prediction-Oriented Transfer Learning for Survival Analysis

Yu Gu, Donglin Zeng, D. Y. Lin

Abstract

Transfer learning is beneficial for survival analysis, especially when the target study has a limited number of events. However, existing transfer learning methods rely on the restrictive assumption that the target and source studies share similar parameters under Cox models, and most require access to individual-level source data. In this article, we propose a novel transfer learning framework that enhances model-based survival prediction by transferring predictive rather than distributional knowledge from source studies. Our approach employs flexible semiparametric transformation models for the target data while eliminating the need to model or share the source data. The ingeniously designed penalty enables simple and stable computation via an EM algorithm. We rigorously establish the asymptotic properties of the proposed estimator and show that it achieves a faster convergence rate than the target-only estimator when source knowledge is sufficiently accurate. We demonstrate the advantages of our methods through extensive simulation studies and an application to two major breast cancer studies.

Prediction-Oriented Transfer Learning for Survival Analysis

Abstract

Transfer learning is beneficial for survival analysis, especially when the target study has a limited number of events. However, existing transfer learning methods rely on the restrictive assumption that the target and source studies share similar parameters under Cox models, and most require access to individual-level source data. In this article, we propose a novel transfer learning framework that enhances model-based survival prediction by transferring predictive rather than distributional knowledge from source studies. Our approach employs flexible semiparametric transformation models for the target data while eliminating the need to model or share the source data. The ingeniously designed penalty enables simple and stable computation via an EM algorithm. We rigorously establish the asymptotic properties of the proposed estimator and show that it achieves a faster convergence rate than the target-only estimator when source knowledge is sufficiently accurate. We demonstrate the advantages of our methods through extensive simulation studies and an application to two major breast cancer studies.
Paper Structure (9 sections, 21 equations, 2 figures, 2 tables)

This paper contains 9 sections, 21 equations, 2 figures, 2 tables.

Figures (2)

  • Figure 1: Kaplan--Meier estimators for survival time distributions of the TCGA--BRCA and METABRIC studies.
  • Figure 2: Predicted survival probabilities for future patients in the TCGA--BRCA study with different tumor stages. All other covariates are fixed at the median values among the current TCGA--BRCA patients.

Theorems & Definitions (1)

  • Remark 1