Multi-task Based Few-Shot Learning for Disease Similarity Measurement

To identify and explore the similarities between diseases is of great significance for u nderstanding t he pathogenic mechanisms of emerging complex diseases. Some methods try to measure the similarity of diseases through deep learning models. However, the insufficient number of labelled similar disease pairs cannot support the optimal training of the models. In this paper, we propose a Multi-Task Graph Neural Network (MTGNN) framework to retrieve similar diseases by few-shot learning. To deal with the problem of insufficient n umber o f labelled similar disease pairs, we design double tasks to optimize the graph neural network for disease similarity task (lack of labelled training data) by introducing link prediction task (sufficient labelled training data). The similarity between diseases can then be obtained by measuring the distance between disease embeddings in high-dimensional space learning from the double tasks. The experiment results illustrate the overall effectiveness by comparing with prior methods on few labeled training dataset.