Integrating domain knowledge in supervised machine learning to assess the risk of breast cancer

摘要

We used various supervised machine learning and data mining techniques to generate a model for predicting risk of breast cancer in post menopausal women using genomic data, family history, and age. In this paper, we propose an approach to select nine best SNPs using various feature selection algorithms and evaluate binary classifiers performance. We have also designed an algorithm to incorporate domain knowledge into our machine learning model. Our observations revealed that the machine learning model generated using both the domain knowledge and the feature selection technique performed better compared to the naive approach of classification. It is also interesting to note that, in addition to selecting nine best SNPs, feature selection resulted in removing age from the set of features to be used for cancer risk assessment.