ITNR: Inversion Transformer-based Neural Ranking for Cancer Drug Recommendations

Abstract

Personalized drug response prediction is an approach for tailoring effective therapeutic strategies for patients based on their tumors genomic characterization. The current study introduces a new listwise Learning-to-rank (LTR) model called Inversion Transformer-based Neural Ranking (ITNR). ITNR utilizes genomic features and a transformer architecture to decipher functional relationships and construct models that can predict patient-specific drug responses. Our experiments were conducted on three major drug response data sets, showing that ITNR reliably and consistently outperforms state-of-the-art LTR models. HighlightsO_LIThe proposed framework is a transformer-based model to predict drug responses using RNAseq gene expression profile, drug descriptors and drug fingerprints. C_LIO_LIITNR utilizes a Context-Aware-Transformer architecture as its scoring function that ensures the modeling of inter-item dependencies. C_LIO_LIWe introduced a novel loss function using the concept of Inversion and Approximate Permutation matrices. C_LIO_LIOur computational results indicated that our method leads to substantially improved performance when compared to the baseline methods across all performance metrics, which can lead to selecting highly effective personalized treatment. C_LI