GraphGONet: a self-explaining neural network encapsulating the Gene Ontology graph for phenotype prediction on gene expression
Abstract
Motivation: Medical care is becoming more and more specific to patients’ needs due to the increased availability of omics data. The application to these data of sophisticated machine learning models, in particular deep learning, can improve the field of precision medicine. However, their use in clinics is limited as their predictions are not accompanied by an explanation. The production of accurate and intelligible predictions can benefit from the inclusion of domain knowledge. Therefore, knowledge-based deep learning models appear to be a promising solution.
Results: In this paper, we propose GraphGONet, where the Gene Ontology is encapsulated in the hidden layers of a new self-explaining neural network. Each neuron in the layers represents a biological concept, combining the gene expression profile of a patient, and the information from its neighboring neurons. The experiments described in the paper confirm that our model not only performs as accurately as the state-of-the-art (non-explainable ones) but also automatically produces stable and intelligible explanations composed of the biological concepts with the highest contribution. This feature allows experts to use our tool in a medical setting.