This article has Open Peer Review reports available.

# A comparison of the conditional inference survival forest model to random survival forests based on a simulation study as well as on two applications with time-to-event data

- Justine B. Nasejje
^{1}Email author, - Henry Mwambi
^{1}, - Keertan Dheda
^{2}and - Maia Lesosky
^{3}

**17**:115

https://doi.org/10.1186/s12874-017-0383-8

© The Author(s) 2017

**Received: **17 February 2017

**Accepted: **30 June 2017

**Published: **28 July 2017

## Abstract

### Background

Random survival forest (RSF) models have been identified as alternative methods to the Cox proportional hazards model in analysing time-to-event data. These methods, however, have been criticised for the bias that results from favouring covariates with many split-points and hence conditional inference forests for time-to-event data have been suggested. Conditional inference forests (CIF) are known to correct the bias in RSF models by separating the procedure for the best covariate to split on from that of the best split point search for the selected covariate.

### Methods

In this study, we compare the random survival forest model to the conditional inference model (CIF) using twenty-two simulated time-to-event datasets. We also analysed two real time-to-event datasets. The first dataset is based on the survival of children under-five years of age in Uganda and it consists of categorical covariates with most of them having more than two levels (many split-points). The second dataset is based on the survival of patients with extremely drug resistant tuberculosis (XDR TB) which consists of mainly categorical covariates with two levels (few split-points).

### Results

The study findings indicate that the conditional inference forest model is superior to random survival forest models in analysing time-to-event data that consists of covariates with many split-points based on the values of the bootstrap cross-validated estimates for integrated Brier scores. However, conditional inference forests perform comparably similar to random survival forests models in analysing time-to-event data consisting of covariates with fewer split-points.

### Conclusion

Although survival forests are promising methods in analysing time-to-event data, it is important to identify the best forest model for analysis based on the nature of covariates of the dataset in question.

## Keywords

## Background

The Cox-proportional hazards model [1] is a popular choice for analysis of right censored time-to-event data. The model is convenient for its flexibility and simplicity, however, it has been criticised for its restrictive proportional hazards (PH) assumption [2–4] which is often violated. A number of extensions to the Cox proportional hazards model to handle time-to-event data where the PH assumption is not met have been suggested and implemented [5–7]. These extensions often remain dependent on restrictive functions such as the heaviside functions that may be difficult to construct and implement or fail to fit the dataset in question. Other analysis approaches to handle non-proportional hazards include methods such as stratification, but these limit the ability to estimate the effect(s) of the stratification variable(s).

Survival trees and random survival forests (RSF) are an attractive alternative approach to the Cox proportional hazards models when the PH assumption is violated [8]. These methods are extensions of classification and regression trees and random forests (RF) [9, 10] for time-to-event data. Survival tree methods are fully non-parametric, flexible, and can easily handle high dimensional covariate data [11–13]. Drawbacks of random survival forests include the common drawbacks of random forests including a bias towards inclusion of variables with many split points [14–17]. This effect leads to a bias in resulting summary estimates such as variable importance [15, 17]. Conditional inference forests (CIF) are known to reduce this selection bias by separating the algorithm for selecting the best covariate to split on from that of the best split point search [15, 17, 18].

Despite the fact that the CIF survival model has been identified to reduce bias in covariate selection for splitting in survival forest models, no study has been done to compare the predictive performance of the CIF model and random survival forest models on time-to-event data in the presence of covariates that have many and fewer split-points. This study for the first time, examines and compares the predictive performance of the CIF and the two random survival forest models through a simulation study. Bootstrap cross-validated estimates of the integrated Brier scores were used as measures of predictive performance [19]. In total, twenty-two time-to-event datasets were simulated. Eighteen of the datasets were simulated in such a way that they either have binary covariates (few split-points), polytomous covariates (many split-points) or both. Four of the datasets were simulated in such a way that they have covariate interactions. Other properties of these datasets are further described in “Methods” section. The two real datasets used in this study are Dataset 1, which investigates the survival of 6692 children under the age of five in Uganda and contains categorical covariates with many levels (polytomous covariates). Dataset 2 evaluates the survival of 107 patients with extremely drug resistant tuberculosis (XDR TB) in South Africa. It is a small dataset and contains only categorical binary covariates.

This article is structured as follows: The “Methods” section describes the methods used. We discuss the methods used to evaluate the methods in the “Model evaluation” section. In the “Simulation study” section, we present the simulation study together with the simulation results. The “Real data application” section introduces the two real datasets that we used in this study and also gives the corresponding real data analyses results and lastly the “Discussion and conclusions” section presents the discussion and conclusions drawn from this study.

## Methods

A random survival forest (RSF) is an assemble of trees method for analysis of right censored time-to-event data and an extension of Brieman’s random forest method [14, 20]. Survival trees and forests are popular non-parametric alternatives to (semi) parametric models for time-to-event analysis. They offer great flexibility and can automatically detect certain types of interactions without the need to specify them beforehand [13]. A survival tree is built with the idea of partitioning the covariate space recursively to form groups of subjects who are similar according to the time-to-event outcome. Homogeneity at a node is achieved by minimizing a given impurity measure. The basic approach for building a survival tree is by using a binary split on a single predictor. For a categorical covariant *X*, a split is defined as *X*≤*c* where *c* is some constant. For a categorical covariate *X* with many split-points, the potential split is *X*∈{*c*
_{1},…,*c*
_{
k
}} where *c*
_{1},…,*c*
_{
k
} are potential split values of a predictor variable *X*. The goal in survival tree building is to identify prognostic factors that are predictive of the time-to-event outcome. In tree building, a binary split is such that the two daughter nodes obtained from the parent node are dissimilar and several split-rules (different impurity measure) for time-to-event data have been suggested over the years [13, 21].

The impurity measure or the split-rule of the algorithm is very important in survival tree building. In this article, we used the log-rank and the log-rank score split-rules [22–24].

### The log-rank split-rule

Suppose a node *h* can be split into two daughter nodes *α* and *β*. The best split at a node *h*, on a covariate *x* at a split point *s*
^{∗} is the one that gives the largest log-rank statistic between the two daughter nodes [22]. The algorithm for building a survival tree using the split-rule based on the log-rank statistic [13, 22, 25, 26] is given in Algorithm 1 below.

### The log-rank score split-rule

*r*=(

*r*

_{1},

*r*

_{2},…,

*r*

_{ N }), the rank vector of survival times with their indicator variable (

*T*,

*δ*)=((

*T*

_{1},

*δ*

_{1}),(

*T*

_{2},

*δ*

_{2}),…,(

*T*

_{ N },

*δ*

_{ N })) and that

*a*=

*a*(

*T*,

*δ*)=(

*a*

_{1}(

*r*),

*a*

_{2}(

*r*),…,

*a*

_{ N }(

*r*)) denotes the score vector depending on ranks in vector

*r*. Assume that the ranks order the predictor variables in such away that

*x*

_{1}<

*x*

_{2}<…<

*x*

_{ N }. The log-rank scores for an observation at

*T*

_{ l }is given by:

*T*

_{ k }.

where \(\bar {a}\) and \(S_{a}^{2}\) are the mean and sample variance of the scores {*a*
_{
j
}:*j*=1,2,…*n*}. The best split is the one that maximizes |*i*(*x*,*s*
^{⋆})| over all \(x_{j}^{'}\)s and possible splits *s*
^{⋆}.

Trees are generally unstable and hence researchers have recommended the growing of a collection of trees [10, 27], commonly referred to as random survival forests [20, 26].

### Random survival forests algorithm

The random survival forests algorithm implementation is shown in Algorithm 2 [20, 26].

For this study, we used the log-rank and the log-rank score split-rules in Step 2 of the algorithm. Two random survival forest algorithms were generated denoted as RSF1 and RSF2. RSF1 consists of survival trees built using the log-rank split-rule whereas RSF2 consists of survival trees built using the log-rank score split-rule.

The random survival forests algorithm, has been criticised for having a bias towards selecting variables with many split points and the conditional inference forest algorithm has been identified as a method to reduce this selection bias. Conditional inference forests are formulated in such a way that it separates the algorithm for selecting the best splitting covariate is separated from the algorithm for selecting the best split point [15–18]. To illustrate this, consider a dataset with a time-to-event outcome variable *T* and two explanatory variables *x*
_{1} and *x*
_{2} with *k*
_{1} and *k*
_{2} possible split-points, respectively. Furthermore, consider that *T* is independent of *x*
_{1} and *x*
_{2}, and that *k*
_{1}<*k*
_{2}. In the random survival forests algorithm, the search for the best covariate to split on and the best split-point by comparing the effect for both the covariates on *T*, gives *x*
_{2} the highest probability of being selected just by chance.

### Conditional inference trees and forests

Algorithm 3 outlines the general algorithm for building a conditional inference tree as presented by [28].

For time-to-event data, the optimal split-variable in step 1 is obtained by testing the association of all the covariates to the time-to-event outcome using an appropriate linear rank test [28, 29]. The covariate with the strongest association to the time-to-event outcome based on permutation tests [28], is selected for splitting. In covariates selection, a linear rank test based on the log-rank transformation (log-rank scores) is performed. Using the distribution of the resulting rank statistic, *p*-values are evaluated and the covariates with minimum *p*-value is known to have the strongest association to the outcome [17, 30, 31]. Although the standard association test is done in the first step, a standard binary split is done in the second step. A single tree is considered unstable and hence research has recommended the growing of an entire forest [9, 10, 20]. The forest of conditional inference trees results into a conditional inference (CIF) model. The CIF model algorithm for time-to-event data is implemented in the R package called party.

To compare the performance of the three models used in this study, integrated Brier scores are used [32] which are described in the section below.

## Model evaluation

*t*, the Brier score for a single subject is defined as the squared difference between observed event status (e.g., 1=alive at time

*t*and 0=dead at time t) and a model based prediction of surviving time

*t*. Using the test sample of size denoted as

*N*

_{test}, Brier scores at time

*t*are given by

Where \(\widehat {G}\left (t|x\right)\approx P\left (C > t|X=x\right)\) is the Kaplan-Meier estimate of the conditional survival function of the censoring times.

The integrated Brier scores (*IBS*) are given by

To avoid the problem of overfitting that arises from using the same dataset to train and test the model, we used the Bootstrap cross-validated estimates of the integrated Brier scores [19]. The prediction errors are evaluated in each bootstrap sample.

These have been implemented in the pec package [19]. We fit a pec object with the three rival prediction models (RSF1, RSF2 and CIF). The three models were passed on as a list to the pec object and chose *s*
*p*
*l*
*i*
*t*
*M*
*e*
*t*
*h*
*o*
*d*=*B*
*o*
*o*
*t*632*p*
*l*
*u*
*s*. We set *B*=5, to have reasonable run times and reported Bootstrap cross-validated estimates for integrated Brier scores. Prediction error rates of 50% or higher are useless because they are no better than tossing a coin [12, 33].

## Simulation study

### Simulated time-to-event datasets

To simulate time-to-event datasets for this study, two frameworks were used. The first framework is more flexible and it generates time-to-event data from known distributions for proportional hazard models by inverting the cumulative baseline hazard function. The desired censoring parameters were achieved by randomly generating them from a binomial distribution. This framework was used to generate polytomous and datasets with covariate interactions [34–36]. The second framework uses a nested numerical integration and a root-finding algorithm to choose the censoring parameter that achieves predefined censoring rates in simulated time-to-event datasets [34]. This framework was used to generate time-to-event datasets with binary covariates.

#### Time-to-event datasets with binary covariates

*X*

_{ j }={

*X*

_{1},

*X*

_{2},

*X*

_{3},

*X*

_{4},

*X*

_{5},

*X*

_{6},

*X*

_{7},

*X*

_{8},

*X*

_{9},

*X*

_{10}}. Each of the covariates is randomly generated from a Bernoulli distribution with probability

*p*

_{ j },

*X*

_{ j }∼

*B*(

*p*

_{ j }). Weibull event times were considered, these were generated using a baseline hazard function \(h_{0}(t)=\frac {\theta }{\rho ^{\theta }}t^{\theta -1}\,\). The scale parameter is given by \(\lambda =\exp \left (-\frac {\beta _{0}}{\theta }-\sum _{j=1}^{n}\frac {\beta _{j}}{\theta }X_{j}\right)\), where

*β*

_{0}is the coefficient of the intercept term log(

*ρ*

^{−θ }). The shape parameter

*θ*was set at 0.8,1.5, or 1 to represent decreasing, increasing and constant hazard, respectively. The corresponding intercept for each dataset was

*β*

_{0}=−0.98,−1.44, or 0.9. The regression coefficients for the 10 covariates were defined as {0.5,−0.045,0.6,−0.03,−2,0.5,0.25,−0.04,0.33,0.3}, {0.1, −0.8,0.5,−0.2,−3,0.7,0.2,0.4,0.3}, {0.4,−0.7,1,−2,−3,0.7,0.06,0.5−0.43,0.3}, respectively. Censoring times were generated from a Weibull distribution with a shape parameters 0.4,2.4, or 1. The censoring parameter

*θ*was computed numerically to get 50

*%*,20

*%*and 80%, respectively. In total, six time-to-event datasets were generated from this covariate design and other properties of these datasets are stated in Table 1.

Simulated time-to-event datasets

Properties of simulated time-to-event datasets | ||||
---|---|---|---|---|

Type of covariates | Datasets | Sample size | % of censoring | Nature of the |

hazard | ||||

Binary | Data 1 | 100 | 80 | Increasing |

Data 2 | 100 | 50 | Decreasing | |

Data 3 | 250 | 20 | Constant | |

Data 4 | 1000 | 80 | Increasing | |

Data 5 | 1500 | 50 | Decreasing | |

Data 6 | 2000 | 20 | Constant | |

Polytomous | Data 1 | 100 | 20 | Increasing |

Data 2 | 100 | 80 | Constant | |

Data 3 | 250 | 50 | Decreasing | |

Data 4 | 1000 | 20 | Increasing | |

Data 5 | 1500 | 80 | Constant | |

Data 6 | 2000 | 50 | Decreasing | |

Binary & polytomous | Data 1 | 1000 | 20 | Increasing |

Data 2 | 100 | 80 | Decreasing | |

Data 3 | 250 | 50 | Constant | |

Data 4 | 1000 | 20 | Increasing | |

Data 5 | 1500 | 80 | Decreasing | |

Data 6 | 2000 | 50 | Constant | |

Interactions | Data 1 | 100 | 20 | Increasing |

Data 2 | 100 | 50 | Decreasing | |

Data 3 | 1000 | 20 | Increasing | |

Data 4 | 1500 | 50 | Decreasing |

#### Time-to-event datasets with polytomous covariates

Covariates were generated by sampling with replacement from a list of desired categories. Event times were generated by inverting cumulative hazard function. *T*=−1∗ log(*U*)∗*ρ*∗*e*
*x*
*p*(−*λ*)^{(}1/*θ*). Where \(\lambda =\exp \left (-\frac {\beta _{0}}{\theta }-\sum _{j=1}^{n}\frac {\beta _{j}}{\theta }X_{j}\right)\). The shape parameter *θ* was set at 0.5,1.5 and 1 to yield time-to-event datasets with a decreasing, increasing and a constant hazard, respectively. The corresponding censoring times were also generated from a Weibull distribution with a shape and scale parameter of 0.4, 1.2 and 1.1, receptively. In this covariate design, six datasets were generated. Other properties for these datasets are given in Table 1.

#### Time-to-event datasets with binary and polytomous covariates

We used the same frame work for generating survival times as that of generating time-to-event datasets with polytomous covariates described above. Binary covariates, were added to the dataset by generating them from a Bernoulli distribution. In total, six datasets were generated.

#### Time-to-event datasets with covariate interactions

We used the same frame work for generating survival times as that for generating time-to-event datasets with polytomous covariates and four datasets were generated. The codes used to generate these datasets is provided as an additional file. In total, twenty-two datasets are generated for this simulation study. The Table 1 presents properties for each simulated dataset.

In this study, the two random survival forest methods, that is, the one consisting of trees built with the log-rank split-rule (RSF1) and the other consisting of survival trees built with the log-rank score split-rule (RSF2) are fit to the data and compared to the CIF model.

To have reasonable run times, 100 survival trees are grown for each survival forest fitted on each simulated dataset and this is repeated 100 times. The models are then evaluated using bootstrap cross-validated estimates of the integrated Brier scores. This estimate is recorded from each fitted survival forest [37–39]. All computations and analyses were carried out in R, using R version 3.3.2. We used the randomForestSRC [26], party [40], pec [19], rms [41], and doMC packages.

## Results on simulated datasets

For each repetition, bootstrapped-cross-validated integrated brier scores were recorded. The results are then reported using box-plots as shown below.

Generally, all the three survival forest models have a good predictive performance based on the bootstrap cross-validated estimates of integrated Brier score. However, there are some differences in the performance of each of the models on each of the simulated dataset as discuss above. The results in summary suggest that conditional inference forests have a good predictive performance compared to the two random survival forest models especially on time-to-event datasets with polytomous covariates. The model is comparable in predictive performance to random survival forests models in the analysis of simulated time-to-event datasets with binary covariates.

## Real data application

To further investigate the results obtained from the simulation study on the predictive performance of the three survival forest models, we analysed two real datasets whose covariate properties are similar to those used in the simulation study. Note that in identifying the most important covariates in explaining survival in the analysis of both datasets, permutation importance was used as measure of variable importance [20, 42].

### Dataset 1

The dataset can be found on the demographic health survey website [43]. In this survey, a representative sample of 10,086 households was selected during the 2011 Uganda Demographic and Health Survey (UDHS). The sample was selected in two stages. First a total of 404 enumeration areas (EAs) were selected from among a list of clusters sampled for the 2009/10 Uganda National Household Survey (2010 UNHS). In the second stage of sampling, households in each cluster were selected from a complete listing of households. Eligible women for the interview were aged between 15–49 years of age who were either usual residents or visitors present in the selected household on the night before the survey. Out of 9247 eligible women, 8674 were successively interviewed with a response rate of 94*%* (91*%* in urban and 95% in rural areas). The study population for this analysis includes infants born between exactly one and five years preceding the 2011 UDHS.

### Explanatory variables

Characteristics and the distribution of deaths for covariates in Dataset 1

Characteristics | Dead N(%) | Alive N(%) | Total | Characteristics | Dead N(%) | Alive N(%) | Total |
---|---|---|---|---|---|---|---|

Mother’s education level | Mother’s occupation | ||||||

Illiterate Mothers | 344(7.7) | 4149(92.3) | 4493 | Not-working | 93(6.9) | 1260(93.1) | 1353 |

Mother completed primary | 119(6.4) | 1749(93.6) | 1868 | Sales and Services | 110 (6.5) | 1589 (93.5) | 1699 |

Secondary and higher | 14(4.2) | 317(95.8) | 331 | Agriculture | 274(7.5) | 3366(92.5) | 3640 |

Partner’s level of education | Births in past 5 years | ||||||

Illiterate Father | 266(7.7) | 3180(92.3) | 3446 | 1-Birth | 93(4.5) | 1982(95.5) | 2075 |

Father completed primary | 170(6.9) | 2287(93.1) | 2457 | 2-Birth | 227(6.5) | 3288(93.5) | 3515 |

Secondary and higher | 41(5.2) | 748(94.8) | 789 | 3-Births | 140(13.6) | 887(86.4) | 1027 |

Birth status | 4-Births | 17(22.7) | 58(77.3) | 75 | |||

Singleton births | 431(6.7) | 6048(93.3) | 6479 | Births in past 1 year | |||

Multiple births (Twins) | 46(21.5) | 167(78.5) | 213 | No-births | 309(6.8) | 4212(93.2) | 4521 |

Sex of the child | 1-Birth | 163(7.6) | 1971(92.4) | 2134 | |||

Males | 258(7.8) | 3067(92.2) | 3325 | 2-Births | 5(13.5) | 32(86.5) | 37 |

Females | 212(6.3) | 3155(93.7) | 3367 | Children Under 5 in Household | |||

Type of place of residence | No-child | 101(34.9) | 188(65.1) | 289 | |||

Urban | 81(5.8) | 1308(94.2) | 1389 | 1-Child | 178(10.5) | 1511(89.5) | 1689 |

Rural | 396(7.5) | 4907(92.5) | 5303 | 2-Children | 146(4.9) | 2831(95.1) | 2977 |

Wealth index | 3-Children | 35(2.5) | 1349(97.5) | 1384 | |||

Poorest | 131(7.5) | 1623(92.5) | 1754 | 4-Children | 17(4.8) | 336(95.2) | 353 |

Poorer | 112(8.5) | 1205(91.5) | 1317 | Mother’s age group | |||

Middle | 86(7.2) | 1109(92.8) | 1195 | Less than 20 years | 29(8.9) | 296(91.1) | 325 |

Richer | 72(6.9) | 969(93.1) | 1041 | 20-29 years | 235(6.5) | 3376(93.5) | 3611 |

Richest | 76(5.5) | 1309(94.5) | 1385 | 30-39 years | 164(7.4) | 2054(92.6) | 2218 |

Children ever born | 40 years | 49(7.9) | 489(90.1) | 538 | |||

One child | 20(3.3) | 581(96.7) | 601 | Birth order number | |||

Two children | 81(7.1) | 1065(92.9) | 1146 | First child | 95(7.6) | 1154(92.4) | 1249 |

Three children | 67(6.6) | 953(93.4) | 1020 | Second to Third child | 117(5.6) | 1974(94.4) | 2091 |

Four and more | 309(7.9) | 3616(92.1) | 3925 | 4 | 149(7.1) | 1949(92.9) | 2098 |

Birth order number |
| 116(9.3) | 1138(90.7) | 1254 | |||

First child | 95(7.6) | 1154(92.4) | 1249 | Sex of household head | |||

Second to Third child | 117(5.6) | 1974(94.4) | 2091 | Male | 341(6.7) | 4771(93.3) | 5112 |

4 | 149(7.1) | 1949(92.9) | 2098 | Female | 136(8.6) | 1444(91.4) | 1580 |

| 116(9.2) | 1138(90.8) | 1254 | Source of drinking water | |||

Religion | Piped water | 76(5.9) | 1204(94.1) | 1280 | |||

Catholics | 217(7.4) | 2722(92.6) | 2939 | Borehole | 216(7.3) | 2731(92.7) | 2947 |

Muslims | 69(7.5) | 852(92.5) | 921 | Well | 93(6.9) | 1261(93.1) | 1354 |

Other Christians | 187(6.8) | 2571(93.2) | 2758 | Surface/Rain/Pond/Lake/tank | 70(8.5) | 756(91.5) | 826 |

Others | 4(5.4) | 70(94.6) | 74 | Other | 22(7.7) | 263(92.3) | 285 |

Type of toilet facility | Age at first birth | ||||||

Flush toilet | 5(4.1) | 116(95.9) | 121 | Less than 20 years | 347(7.5) | 4291(92.5) | 4638 |

Pitlatrine | 376(6.9) | 5031(93.1) | 5407 | 20-29 years | 127(6.3) | 1899(93.7) | 2026 |

No-facility | 96(8.2) | 1068(91.8) | 1164 | 30-39 years | 3(12.0) | 22(88.0) | 25 |

The time-to-event outcome of interest is time to death of children under the age of five. The range of values of this outcome lie between one month and 59 months of age. Children that were alive at the time of the interview were considered to be right censored. The dataset has a high censoring rate of 93%.

### Dataset 2

Between August 2002 and October 2012, a total of 107 adult patients with microbiologically confirmed XDR-TB from three provinces in South Africa, were hospitalised for treatment in three tuberculosis treatment facilities (Brooklyn Chest Hospital, Western Cape [B], Gordonia Hospital, Upington, Northern Cape, [H] and Sizwe Tropical Disease Hospital, Johannesburg, Gauteng [S]). All the three hospitals are specialist referral centres for the treatment of drug resistant TB, aimed at serving patients from across respective provinces. This dataset has been published in [47].

#### Explanatory variable

The covariates of interests were selected based on literature [47, 48] and limitations of high level of missingness.

Characteristics and the distribution of deaths for covariates in Dataset 2

Characteristics | Dead N(%) | AliveN (%) | Total | Characteristics | Dead N(%) | Alive N(%) | Total |
---|---|---|---|---|---|---|---|

Age at diagnosis | Ethionamide | ||||||

Below 30 | 35(81.3) | 8(18.6) | 43 | Not prescribed | 25(64.10) | 14(35.89) | 39 |

Above 30 | 43(68.25) | 20(31.75) | 63 | Prescribed | 54(79.41) | 14(20.59) | 68 |

Gender | Ofloxacin | ||||||

Females | 41(83.67) | 8(16.33) | 49 | Not prescribed | 48(70.59) | 20(29.41) | 68 |

Males | 38(65.52) | 20(34.48) | 58 | Prescribed | 31(79.49) | 8(20.51) | 39 |

smoking status | Ofloxacin and moxifloxacin | ||||||

No | 28(65.12) | 15(34.88) | 43 | Not prescribed | 72(72.73) | 27(27.27) | 99 |

Yes | 38(79.17) | 10(20.83) | 48 | Prescribed | 7(87.50) | 1(12.50) | 8 |

HIV plus ART status | Amikacin | ||||||

HIV -ve | 46(73.02) | 17(26.98) | 63 | Not prescribed | 76(73.79) | 27(26.21) | 103 |

HIV +ve ART | 24(68.57) | 11(31.43) | 35 | Prescribed | 3(75.00) | 1(25.00) | 4 |

HIV +ve no ART | 9 (100.00) | 0 (0.00) | 9 | Capreomycin | |||

Cohort | Not prescribed | 8(88.98) | 1(11.11) | 9 | |||

B | 54(83.08) | 11(16.92) | 65 | Prescribed | 71(72.45) | 27(27.55) | 98 |

N | 12(80.00) | 3(20.00) | 15 | Dapsone | |||

S | 13(48.15) | 14(51.85) | 27 | Not prescribed | 43(67.19) | 21(32.81) | 64 |

Race | Prescribed | 36(83.72) | 7(16.28) | 43 | |||

Blacks | 34(64.15) | 19(35.85) | 53 | Augmentin | |||

Mixed ancestry | 45(83.33) | 9(16.67) | 54 | Not prescribed | 28(66.67) | 14(33.33) | 42 |

Drugs used | Prescribed | 51(78.46) | 14(21.54) | 65 | |||

Isoniazid | Clofazamine | ||||||

Not prescribed | 57(83.82) | 11(16.18) | 68 | Not prescribed | 70(82.35) | 15(17.65) | 85 |

Prescribed | 22(56.41) | 17(43.59) | 39 | Prescribed | 9(40.91) | 13(59.09) | 22 |

Etambutol | Azithromycin | ||||||

Not prescribed | 39(66.10) | 20(33.89) | 59 | Not prescribed | 75(76.53) | 23(23.47) | 98 |

Prescribed | 40(83.33) | 8(16.67) | 48 | Prescribed | 4(44.44) | 5(55.56) | 9 |

Pyrazinainamide | Amoxicillin | ||||||

Not prescribed | 14(58.33) | 10(41.67) | 24 | Not prescribed | 49(71.01) | 20(28.99) | 69 |

Prescribed | 65(78.3) | 18(21.69) | 83 | Prescribed | 30(78.95) | 8(21.05) | 38 |

Clarithromycin | |||||||

Not prescribed | 19(70.37) | 8(29.63) | 27 | ||||

Prescribed | 60(75.00) | 20(25.00) | 80 |

The median survival time was 30.9(IQR =19.5) months. A total of 79 (74%) patients died, with a low censoring rate of 26%.

### Analysis

#### Dataset 1

Other covariates strongly associated to under-five child mortality in Uganda include; the number of births in the past five years, birth order, wealth index and the total number of children ever born. Both random survival forest models have similar results in identifying the same factors affecting the time to survival of children under the age of five years in Uganda.

#### Dataset 2

The three survival forest models give similar results in determining the factors affecting the survival of patients with XDR TB.

## Results on real datasets

## Discussion and conclusions

In this study, we compared the predictive performance of three survival forests models on twenty-two simulated time-to-event datasets and two real time-to-event datasets. First, eighteen datasets were simulated to have covariate properties of interest that is, fewer split-points vs many split-points. Four more time-to-event datasets with covariate interactions were also simulated. The first two forest models are random survival forests models with trees built based on the log-rank and the log-rank score split-rule, respectively. The third survival forest model consists of conditional inference trees.

The results from comparing the predictive performance on three survival forest models on simulated time-to-event datasets indicate that the three random survival forest models have a good predictive performance. Despite this fact, the study has shown that there are some variations in predictive performance for these three models in the presence of covariates with many vs those with fewer split-points. The study suggests that conditional inference forests are superior in predictive performance to random survival forests on time-to-event datasets with polytomous covariates. The results also indicate that the three models are comparable in predictive performance on time-to-event datasets with categorical covariates that are binary in nature. The superiority in performance of the CIF model is likely due to the way it handles the split variable and the split point selection especially in the presence of covariates with many split-points. These results are similar to those from the simulation study. This study therefore confirms the results that conditional inference forests are desirable in analysing time-to-event data consisting of covariates with many split-points. This results is therefore in agreement with the assertion made from a study by [28] that the CIF model is desirable in analysing time-to-event data in the presence of covariates with many split points.

The main finding of this study is that random survival forests perform comparably to conditional inference forests in analysing time-to-event data consisting of covariates with few split-points and that conditional inference forests are desirable in situations where the data consists of covariates with many split-points. It is therefore important for researchers to select the best survival forest model to analyse any time-to-event dataset based on the nature of its covariates.

Note that the conditional inference forest for time-to-event analysis and random survival forests have a difference in the way they calculate the predicted time-to-event probabilities and it is not yet clear whether this has an influence on their overall predictive performance [19]. The CIF model utilizes a weighted Kaplan-Meier estimate based on all subjects from the training dataset and it therefore put more weight on terminal nodes where there is a large number of subjects at risk whereas random survival forests use equal weights on all terminal nodes. Further studies need to be done to understand whether this property has an influence on the predictive performance of these models.

The limitation of this study is that we used random survival forest models that consists of survival trees based on the log-rank split-rule. Recent studies have raised concerns that since the log-rank split-rule is based on the proportional hazards assumption, it may negatively affect the predictive performance of the survival forest model. A recent study has recommended the use of the integrated absolute difference between the two daughter nodes survival functions as the splitting rule especially in circumstances when the hazard functions cross [50]. Further studies would therefore compare the predictive performance of the CIF model to robust random survival forest models resulting from using robust split-rules especially in the presence of covariates that violate the PH assumption. Another limitation of the study is that the real datasets used had missing data and we assumed that the data was missing completely at random. Only a complete case analysis was considered which negatively affect outcomes.

## Declarations

### Acknowledgements

The first author acknowledges financial support from DAAD and the University of Kwazulu-Natal for her PhD studies. We thank Prof Keertan Dheda, Director of the Lung Infection and Immunity Unit and Head of the Division of Pulmonology, Department of Medicine, at the University of Cape Town for permission to utilised the TB XDR data in this analysis, and all of the contributors and participants to the study. We also acknowledge the DHS Program for making data from Uganda available, and the contributions of all the women who participated in the survey together with the team that conducted the survey. And lastly, the first author acknowledges Mr. Andile Gumede for his technical support during the production of this research article.

### Funding

We thank the University of Kwazulu-Natal for supporting NJB in her PHD work. The first author is also financially supported by DAAD.

### Availability of data and materials

The authors confirm that all data underlying the findings are fully available without restriction. Dataset 1 is held by the Demographic and Health Survey program and freely available to the public but a request has to be sent to the Demographic and Health Survey program. The link to access it is http://www.dhsprogram.com/data/dataset_admin/download-datasets.cfm.

Dataset 2 maybe requested from Prof Keertan Dheda, Director of the Lung Infection and Immunity Unit and Head of the Division of Pulmonology, Department of Medicine, at the University of Cape Town.

### Authors’ contributions

Conceived by NJB, MH, ML and KD. Analyzed the data: NJB. Wrote the first draft of the manuscript: NJB. Contributed to the writing of the manuscript: NJB, MH, ML and KD. Agree with the manuscript’s results and conclusions: NJB, MH, ML and KD. All authors have read and reviewed the manuscript.

### Ethics approval and consent to participate

The ethical statement for Dataset 1 is available on the DHS ethical clearance certificate and it states that: The IRB-approved procedures for DHS public-use datasets do not in any way allow respondents, households, or sample communities to be identified. There are no names of individuals or household addresses in the data files. The geographic identifiers only go down to the regional level (where regions are typically very large geographical areas encompassing several states/provinces). Each enumeration area (Primary Sampling Unit) has a PSU number in the data file, but the PSU numbers do not have any labels to indicate their names or locations. In surveys that collect GIS coordinates in the field, the coordinates are only for the enumeration area (EA) as a whole, and not for individual households, and the measured coordinates are randomly displaced within a large geographic area so that specific enumeration areas cannot be identified. No ethical clearance was required from the University of Kwazulu-Natal.

The ethical statement for Dataset 2, however, is available on request from Prof Keertan Dheda, the Director of the Lung Infection and Immunity Unit and Head of the Division of Pulmonology, Department of Medicine, at the University of Cape Town.

### Consent for publication

Dataset 1 was obtained from the Demographic and Health Survey data project. The Demographic Health Survey Data is collected according to the rules and guidelines stipulated by WHO World Health Survey on consent from the participants stated below:

Participation in the survey is voluntary and the respondent can refuse to be interviewed. The interviewer is responsible for explaining what the survey is about, providing all the necessary information, and making sure the respondent understands the implications of his/her participation before giving his/her consent. The information given should be simple and clear and adapted to the respondent’s level of understanding. Consents must be documented by asking the respondents to sign an Informed Consent Forms (Household Informant Consent Form; Individual Consent Form) before doing the interview. These forms must mention who will be doing the study, the types of questions that will be asked, why the study is being done, and who will have access to the information provided. The interviewer must check that the respondent has read and understood the form before signing, and should offer to go over it with him /her emphasizing the different items mentioned. If the respondent is illiterate or unable to read for himself/herself (e.g. due to a visual impairment), the form will be read and explained to him/her. In cases where it is not appropriate for the respondent to sign the form, the interviewer alone will sign the form. In cases where the respondent is being dissuaded from, or coerced into, participating in the study by a third party such as a spouse, relative or any other member in the community, the interviewer should make it clear that it is the respondent alone who must decide whether or not s/he wishes to be interviewed.

Dataset 2 was obtained from a study conducted by a group lead by Prof Keertan Dheda, the Director of the Lung Infection and Immunity Unit and Head of the Division of Pulmonology, Department of Medicine, at the University of Cape Town. He gave permission to utilise the dataset in this analysis, and all of the contributors and participants to the study consented to the study.

### Competing interests

The authors declare that they have no competing interests.

### Publisher’s Note

Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.

**Open Access** This article is distributed under the terms of the Creative Commons Attribution 4.0 International License(http://creativecommons.org/licenses/by/4.0/), which permits unrestricted use, distribution, and reproduction in any medium, provided you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons license, and indicate if changes were made. The Creative Commons Public Domain Dedication waiver(http://creativecommons.org/publicdomain/zero/1.0/) applies to the data made available in this article, unless otherwise stated.

## Authors’ Affiliations

## References

- Cox DR. Regression models and life tables (with discussion). J R Stat Soc Ser B. 1972; 34(2):187–220.Google Scholar
- Platt RW, Joseph K, Ananth CV, Grondines J, Abrahamowicz M, Kramer MS. A proportional hazards model with time-dependent covariates and time-varying effects for analysis of fetal and infant death. Am J Epidemiol. 2004; 160(3):199–206.View ArticlePubMedGoogle Scholar
- Ng’andu NH. An empirical comparison of statistical tests for assessing the proportional hazards assumption of cox’s model. Stat Med. 1997; 16(6):611–26.View ArticlePubMedGoogle Scholar
- Fisher LD, Lin DY. Time-dependent covariates in the cox proportional-hazards regression model. Annu Rev Public Health. 1999; 20(1):145–57.View ArticlePubMedGoogle Scholar
- Therneau TM. Extending the Cox model In: Lin DY, Fleming TR, editors. Proceedings of the First Seattle Symposium in Biostatistics. New York: Springer Verlag: 1997. p. 51–84.Google Scholar
- Wei L. The accelerated failure time model: a useful alternative to the cox regression model in survival analysis. Stat Med. 1992; 11(14-15):1871–9.View ArticlePubMedGoogle Scholar
- Therneau TM, Grambsch PM. Modeling survival data: extending the Cox model.New York: Springer Verlag; 2000.View ArticleGoogle Scholar
- Ehrlinger J. ggRandomForests Exploring random forest survival. R Vignette. 2016.Google Scholar
- Breiman L, Friedman J, Stone CJ, Olshen RA. Classification and regression trees. Belmont: CRC press; 1984.Google Scholar
- Breiman L. Random forests. Mach Learn. 2001; 45(1):5–32.View ArticleGoogle Scholar
- Fernández T, Rivera N, Teh YW. Gaussian processes for survival analysis. In: Advances in Neural Information Processing Systems.New York: Curran Associates: 2016. p. 5015–023.Google Scholar
- Taylor JM. Random survival forests. J Thorac Oncol. 2011; 6(12):1974–5.View ArticlePubMedGoogle Scholar
- Bou-Hamad I, Larocque D, Ben-Ameur H, et al. A review of survival trees. Stat Surv. 2011; 5:44–71.View ArticleGoogle Scholar
- Ziegler A, König IR. Mining data with random forests: current options for real-world applications. Wiley Interdiscip Rev Data Min Knowl Disc. 2014; 4(1):55–63.View ArticleGoogle Scholar
- Strobl C, Boulesteix AL, Zeileis A, Hothorn T. Bias in random forest variable importance measures: Illustrations, sources and a solution. BMC Bioinforma. 2007; 8(1):1.View ArticleGoogle Scholar
- Loh WY. Fifty years of classification and regression trees. Int Stat Rev. 2014; 82(3):329–48.View ArticleGoogle Scholar
- Wright MN, Dankowski T, Ziegler A. Unbiased split variable selection for random survival forests using maximally selected rank statistics. Stat Med. 2017; 36(8):1272–84. doi:10.1002/sim.7212.sim.7212.
- Das A, Abdel-Aty M, Pande A. Using conditional inference forests to identify the factors affecting crash severity on arterial corridors. J Saf Res. 2009; 40(4):317–27.Google Scholar
- Mogensen UB, Ishwaran H, Gerds TA. Evaluating random forests for survival analysis using prediction error curves. J Stat Softw. 2012; 50(11):1.View ArticlePubMedPubMed CentralGoogle Scholar
- Ishwaran H, Kogalur UB, Blackstone EH, Lauer MS. Random survival forests. Ann Appl Stat. 2008;841–60.Google Scholar
- Gordon L, Olshen RA. Tree-structured survival analysis. Cancer Treat Rep. 1985; 69:1065–1069.PubMedGoogle Scholar
- Ciampi A, Chang CH, Hogg S, McKinney S. Recursive partition: A versatile method for exploratory-data analysis in biostatistics. In: Biostatistics.New York: Springer: 1987. p. 23–50.Google Scholar
- Hothorn T, Lausen B. On the exact distribution of maximally selected rank statistics. Comput Stat Data Anal. 2003; 43(2):121–37.View ArticleGoogle Scholar
- Lausen B, Schumacher M. Maximally selected rank statistics. Biometrics. 1992;73–85.Google Scholar
- Segal MR. Regression trees for censored data. Biometrics. 1988;35–47.Google Scholar
- Ishwaran H, Kogalur UB. randomForestSRC: Random Forests for Survival, Regression and Classification (RF-SRC). R package version 1.4.0. 2014. https://cran.r-project.org/.
- Dietterich T. Ensemble Learning. Tha Handbook of Brain Theory and Neural Networks. Cambridge MA: The MIT Press; 2002.Google Scholar
- Hothorn T, Hornik K, Zeileis A. Unbiased recursive partitioning: A conditional inference framework. J Comput Graph Stat. 2006; 15:651–74.View ArticleGoogle Scholar
- Strasser H, Weber C. On the asymptotic theory of permutation statistics. Math Methods Stat. 1999; 8:220–50.Google Scholar
- Harrington D. Linear rank tests in survival analysis In: Armitage P, Colton T, editors. Encyclopedia of biostatistics(2nd edn). New York: Wiley Online Library: 2005. p. 2802–2812.Google Scholar
- Hothorn T, Hornik K, Strobl C, Zeileis A. Party: a laboratory for recursive partitioning. R package version 1.0-23. 2015. https://cran.r-project.org/web/packages/party/index.html.
- Graf E, Schmoor C, Sauerbrei W, Schumacher M. Assessment and comparison of prognostic classification schemes for survival data. Stat Med. 1999; 18(17-18):2529–45.View ArticlePubMedGoogle Scholar
- Chen G, Kim S, Taylor JM, Wang Z, Lee O, Ramnath N, Reddy RM, Lin J, Chang AC, Orringer MB, et al.Development and validation of a quantitative real-time polymerase chain reaction classifier for lung cancer prognosis. J Thorac Oncol. 2011; 6(9):1481–7.View ArticlePubMedPubMed CentralGoogle Scholar
- Wan F. Simulating survival data with predefined censoring rates for proportional hazards models. Stat Med. 2017; 36.5:838.View ArticleGoogle Scholar
- Bender R, Augustin T, Blettner M. Generating survival times to simulate cox proportional hazards models. Stat Med. 2005; 24(11):1713–23.View ArticlePubMedGoogle Scholar
- Crowther MJ, Lambert PC. Simulating biologically plausible complex survival data. Stat Med. 2013; 32(23):4118–34.View ArticlePubMedGoogle Scholar
- Kohavi R, et al. A study of cross-validation and bootstrap for accuracy estimation and model selection. In: Ijcai. vol. 14, no. 2. Morgan Kaufmann: Los Altos: 1995. p. 1137–1145.Google Scholar
- Refaeilzadeh P, Tang L, Liu H. Cross-Validation In: Liu L, Özsu MT, editors. Boston: Springer: 2009. p. 532–8.Google Scholar
- Bengio Y, Grandvalet Y. No unbiased estimator of the variance of k-fold cross-validation. J Mach Learn Res. 2004; 5(Sep):1089–05.Google Scholar
- Hothorn T, Hornik K, Strobl C, Zeileis A, Hothorn MT. Package ‘party’. Packag Ref Man Party Version 0.9-998. 2015; 16:37.Google Scholar
- Harrell Jr FE, Harrell Jr MFE, Hmisc D. Package ‘rms’; 2017. https://cran.r-project.org/web/packages/rms/index.html.
- Strobl C, Boulesteix A, Kneib T, Augustin T, Zeileis A. Conditional variable importance for random forests. BMC Bioinforma. 2008; 9:307.View ArticleGoogle Scholar
- Demographic and Healthy Survey Datasets. http://dhsprogram.com/data/available-datasets.cfm. Accessed 25 Oct 2016.
- Ssewanyana S, Younger SD. Infant mortality in uganda: Determinants, trends and the millennium development goals. J Afr Econ. 2008; 17(1):34–61.View ArticleGoogle Scholar
- Ayiko R, Antai D, Kulane A. Trends and determinants of under-five mortality in uganda. East Afr J Public Health. 2009; 6(2):136–40.PubMedGoogle Scholar
- Demombynes G, Trommlerová SK. What has driven the decline of infant mortality in kenya?Washington: Policy research working paper No. WPS 60572010: World Bank; 2012.View ArticleGoogle Scholar
- Pietersen E, Ignatius E, Streicher EM, Mastrapa B, Padanilam X, Pooran A, Badri M, Lesosky M, van Helden P, Sirgel FA, et al. Long-term outcomes of patients with extensively drug-resistant tuberculosis in south africa: a cohort study. Lancet. 2014; 383(9924):1230–9.View ArticlePubMedGoogle Scholar
- Kim DH, Kim HJ, Park SK, Kong SJ, Kim YS, Kim TH, Kim EK, Lee KM, Lee SS, Park JS, et al. Treatment outcomes and long-term survival in patients with extensively drug-resistant tuberculosis. Am J Respir Crit Care Med. 2008; 178(10):1075–82.View ArticlePubMedGoogle Scholar
- Nasejje JB, Mwambi HG, Achia TN. Understanding the determinants of under-five child mortality in uganda including the estimation of unobserved household and community effects using both frequentist and bayesian survival analysis approaches. BMC Public Health. 2015; 15(1):1.View ArticleGoogle Scholar
- Moradian H, Larocque D, Bellavance F. L_1 splitting rules in survival forests. Lifetime Data Anal. 2016:1–21.Google Scholar