Gamma distribution based predicting model for breast cancer drug response based on multi-layer feature selection

• Published in: Frontiers in genetics Vol. 14; p. 1095976
• Main Authors: Cui, Tongtong, Wang, Zeyuan, Gu, Hong, Qin, Pan, Wang, Jia
• Format: Journal Article
• Language: English
• Published: Switzerland Frontiers Media S.A 02.02.2023
• Subjects: artificial neural network; breast cancer; drug response; feature selection; generalized linear model; Genetics; machine learning; drug response; generalized linear model; artificial neural network; breast cancer; machine learning; feature selection
• DOI: 10.3389/fgene.2023.1095976
• ISSN: 1664-8021

In the pursuit of precision medicine for cancer, a promising step is to predict drug response based on data mining, which can provide clinical decision support for cancer patients. Although some machine learning methods for predicting drug response from genomic data already exist, most of them focus on point prediction, which cannot reveal the distribution of predicted results. In this paper, we propose a three-layer feature selection combined with a gamma distribution based GLM and a two-layer feature selection combined with an ANN. The two regression methods are applied to the Encyclopedia of Cancer Cell Lines (CCLE) and the Cancer Drug Sensitivity Genomics (GDSC) datasets. Using ten-fold cross-validation, our methods achieve higher accuracy on anticancer drug response prediction compared to existing methods, with an R 2 and RMSE of 0.87 and 0.53, respectively. Through data validation, the significance of assessing the reliability of predictions by predicting confidence intervals and its role in personalized medicine are illustrated. The correlation analysis of the genes selected from the three layers of features also shows the effectiveness of our proposed methods.