Survival analysis in R

Transcription

1 Survival analysis in R Niels Richard Hansen This note describes a few elementary aspects of practical analysis of survival data in R. For further information we refer to the book Introductory Statistics with R by Peter Dalgaard and Modeling Survival Data by Terry M. Therneau (the author of the survival-package for R) and Patricia M. Grambsch. For more advanced topics see the recent book Dynamic Regression Models for Survival Data by Torben Martinussen and Thomas Scheike. A more classical and general reference is Statistical Models Based on Counting Processes by Andersen, Borgan, Gill and Keiding. The analyzes that we will do can all be done using the functions in R-library survival. That library also contains some example data sets. > library(survival) > data(pbc) > pbc[1:10, c("time", "status", "age", "edema", "alb", "bili", + "protime")] time status age edema alb bili protime The data set pbc (primary biliary cirrhosis) contains the main variable time, which is the observed, potentially censored survival time and the status, which indicates if the observation has been censored. In addition there are 17 observed covariates, where we show above five that in previous analyzes have been found to be important. We can produce a plot of the survival times with : > nr <- 50 > plot(c(0, pbc$time[1]), c(1, 1), type = "l", ylim = c(0, nr + 1

2 + 1), xlim = c(0, max(pbc$time[1:nr]) + 10), ylab = "nr", xlab = "Survival tim > for (i in 2:nr) lines(c(0, pbc$time[i]), c(i, i)) > for (i in 1:nr) { + if (pbc$status[i] == 0) + points(pbc$time[i], i, col = "red", pch = 20) + if (pbc$status[i] == 1) + points(pbc$time[i], i) + } nr Survival time Estimation of the survival function using the Kaplan-Meier estimator can be done using the survfit function. > pbc.surv <- survfit(surv(time, status), data = pbc) > pbc.surv Call: survfit(formula = Surv(time, status), data = pbc) n events median 0.95LCL 0.95UCL

3 > plot(pbc.surv, mark.time = FALSE) The result from survfit is an R object of type survfit. Taking summary of a survfit object provides the non-parametric estimate of the survival function including additional information as a long list. The plot with the added, pointwise confidence bands is perhaps more informative. Setting mark.time=false prevents all the censored observations to be marked on the plot by a +. The function Surv applied to the time and status variables for the PBC data is a function that create a survival object. A major point in a data set like the PBC data above is that we record covariate information on the individuals. It is no problem to split the estimation of the survival function according to a factor such as the binary edema variable in the PBC data. > pbc.surv <- survfit(surv(time, status) ~ edema, data = pbc) > plot(pbc.surv, conf.int = TRUE, col = c("red", "blue")) 3

4 From the plot it seems clear that individuals without edema have a larger chance of a large survival time than those with edema. This can also be tested formally with the so-called log-rank test. > survdiff(surv(time, status) ~ edema, data = pbc) Call: survdiff(formula = Surv(time, status) ~ edema, data = pbc) N Observed Expected (O-E)^2/E (O-E)^2/V edema= edema= Chisq= 96 on 1 degrees of freedom, p= 0 But what about including the other covariates also especially the continuous ones? This can be handled in the framework of the Cox proportional hazards model. > pbc.surv <- coxph(surv(time, status) ~ edema, data = pbc) > summary(pbc.surv) 4

5 Call: coxph(formula = Surv(time, status) ~ edema, data = pbc) n= 418 coef exp(coef) se(coef) z p edema exp(coef) exp(-coef) lower.95 upper.95 edema Rsquare= (max possible= ) Likelihood ratio test= 57.7 on 1 df, p=3e-14 Wald test = 77.7 on 1 df, p=0 Score (logrank) test = 96 on 1 df, p=0 We can tell from the summary that the β corresponding to the covariate edema is significantly different from 0. We also see that the hazard rate for those with edema=1 is estimated as roughly 5 times the base-line hazard rate α 0 (t). We can include more covariates. > pbc.surv <- coxph(surv(time, status) ~ edema + age + alb + bili + + protime, data = pbc) > summary(pbc.surv) Call: coxph(formula = Surv(time, status) ~ edema + age + alb + bili + protime, data = pbc) n= 418 coef exp(coef) se(coef) z p edema e-04 age e-06 alb e-06 bili e+00 protime e-04 exp(coef) exp(-coef) lower.95 upper.95 edema age alb bili protime

6 Rsquare= 0.36 (max possible= ) Likelihood ratio test= 186 on 5 df, p=0 Wald test = 218 on 5 df, p=0 Score (logrank) test = 309 on 5 df, p=0 It is of course also possible to transform the covariates before using them in the regression. For example; > pbc.surv <- coxph(surv(time, status) ~ edema + age + log(bili) + + log(alb) + log(protime), data = pbc) > summary(pbc.surv) Call: coxph(formula = Surv(time, status) ~ edema + age + log(bili) + log(alb) + log(protime), data = pbc) n= 418 coef exp(coef) se(coef) z p edema e-03 age e-07 log(bili) e+00 log(alb) e-04 log(protime) e-03 exp(coef) exp(-coef) lower.95 upper.95 edema age log(bili) log(alb) log(protime) Rsquare= (max possible= ) Likelihood ratio test= 234 on 5 df, p=0 Wald test = 233 on 5 df, p=0 Score (logrank) test = 297 on 5 df, p=0 One will need tools for model diagnostic. One simple idea is as follows. The survival functions, S i (t), for different individuals, i, must fulfill that log( log(s i (t)) = log t 0 α(s)ds + X T i β are parallel. Now we don t know each individual survival function but if we group individuals with similar covariates we might be able to estimate, nonparametrically, the survival functions for the groups and then visually inspect if the survival functions look parallel. 6

7 Considering only one variable, bili, we divide into three groups. > group <- 1 * (pbc$bili > 1.1) + 1 * (pbc$bili > 3.3) > cfit <- survfit(surv(time, status) ~ group, data = pbc) > plot(cfit, mark.time = FALSE, fun = "cloglog", xlim = c(40, 5000)) Including in addition edema, which is already discrete we get a total of 6 groups. Note the additive specification in the formula below, which in reality stratifies the estimation into all 6 possible combinations. > cfit <- survfit(surv(time, status) ~ group + edema, data = pbc) > plot(cfit, mark.time = FALSE, fun = "cloglog", xlim = c(40, 5000)) 7

8 It is left to the reader to judge the plots... A good question is what happens if more stratification according to other variables is applied. A Theory We consider n individuals, T1,..., Tn (survival times). independent, positive random variables Estimation of the distribution by the empirical distribution: ˆF (t) = 1 n n i=1 and the survival function S(t) = 1 F (t) Ŝ(t) = 1 ˆF (t) = 1 n 1(T i t) n i=1 1(T i > t). 8

9 Censoring by positive, random variables C 1,..., C n : We observe only and Thus we observe the pairs T i = min{t i, C i } i = 1(T i C i ). (T 1, 1 ),..., (T n, n ). Define the counting process of deaths (non censored events) N(t) = i=1 1(T i t, i = 1). Define also the process of individuals at risk n Y (t) = 1(t T i ). i=1 The jumps for the counting process are given by the process N(t) = N(t) N(t ) = N(t) lim ɛ 0+ N(t ɛ) {0, 1}, which is simply an indicator process for each jump (death). The Kaplan-Meier estimator of the survival function, based on censored survival times, is Ŝ(t) = ( 1 N(s) ). s t Y (s) The factors are equal to 1 for all s where N(s) = 0. If we denote by τ i the time for the i th jump, we can also write the estimator out as ( Ŝ(t) = 1 1 ) ( 1 1 ) ( ) Y (τ 1 ) Y (τ 2 ) Y (τ N(t) ) Each factor can be interpreted as an estimator of the conditional probability of dying in the time interval (τ i, τ i+1 ] given survival to time τ i. If F is continuous differentiable with derivative f (the density for the survival distribution then), we can introduce the hazard rate: α(t) = f(t) S(t). We see that α(t) = 1 F (t + ɛ) F (t) lim ɛ 0+ ɛ S(t) = 1 lim ɛ 0+ ɛ P (T i (t, t + ɛ] Ti > t). 9

10 Thus α(t) is the instantaneous rate of death at time t. The exponential distribution provides an example where α(t) = λ is constant. In general, we can always get back the the survival function from the hazard rate by ( t ) S(t) = exp α(s)ds. 0 This means that we can specify a distribution via its hazard rate as well as its survival function. The Cox proportional hazards rate model is given as follows. We assume that we for each individual has a vector X i = (X 1i,..., X mi ) T of real valued covariates (they may be indicator variables as well). Then we specify the hazard rate for the i th individual as m α i (t, X i ) = α 0 (t) exp β j X ij = α 0 (t) exp ( X T β ) j=1 for an m-dimensional vector β of parameters. The model is a semi-parametric model. The estimation of α 0 (t) has secondary interest whereas the estimation of the β s is of primary interest. We see that α i (t, X i + (0,..., 0, δ, 0,..., 0) T ) α i (t, X i ) = e δβ j if the δ above is at the j th position in the vector. This gives an interpretation of the parameters as the log-relative risks for a given covariate. The parameters are estimated by solving the estimation equation n (X i E(β, T i )) i = 0 i=1 where E(β, T i ) = i 1(t T i )X i exp(x T i β) i 1(t T i ) exp(x T i β). 10