Heterogeneity aware random forest for drug sensitivity prediction

Raziur Rahman, Kevin Matlock, Souparno Ghosh, Ranadip Pal

Research output: Contribution to journalArticlepeer-review

40 Scopus citations


Samples collected in pharmacogenomics databases typically belong to various cancer types. For designing a drug sensitivity predictive model from such a database, a natural question arises whether a model trained on diverse inter-tumor heterogeneous samples will perform similar to a predictive model that takes into consideration the heterogeneity of the samples in model training and prediction. We explore this hypothesis and observe that ensemble model predictions obtained when cancer type is known out-perform predictions when that information is withheld even when the samples sizes for the former is considerably lower than the combined sample size. To incorporate the heterogeneity idea in the commonly used ensemble based predictive model of Random Forests, we propose Heterogeneity Aware Random Forests (HARF) that assigns weights to the trees based on the category of the sample. We treat heterogeneity as a latent class allocation problem and present a covariate free class allocation approach based on the distribution of leaf nodes of the model ensemble. Applications on CCLE and GDSC databases show that HARF outperforms traditional Random Forest when the average drug responses of cancer types are different.

Original languageEnglish
Article number11347
JournalScientific reports
Issue number1
StatePublished - Dec 1 2017


Dive into the research topics of 'Heterogeneity aware random forest for drug sensitivity prediction'. Together they form a unique fingerprint.

Cite this