Identifying diseases that cause psychological trauma and social avoidance by GCN-Xgboost

With the rapid development of medical treatment, many patients not only consider the survival time, but also care about the quality of life. Changes in physical, psychological and social functions after and during treatment have caused a lot of troubles to patients and their families. Based on the bio-psycho-social medical model theory, mental health plays an important role in treatment. Therefore, it is necessary for medical staff to know the diseases which have high potential to cause psychological trauma and social avoidance (PTSA).

Firstly, we obtained diseases which can cause PTSA from literatures. Then, we calculated the similarities of related-diseases to build a disease network. The similarities between diseases were based on their known related genes. Then, we obtained these diseases-related proteins from UniProt. These proteins were extracted as the features of diseases. Therefore, in the disease network, each node denotes a disease and contains the information of its related proteins, and the edges of the network are the similarities of diseases. Then, graph convolutional network (GCN) was used to encode the disease network. In this way, each disease's own feature and its relationship with other diseases were extracted. Finally, Xgboost was used to identify PTSA diseases.

We developed a novel method 'GCN-Xgboost' and compared it with some traditional methods. Using leave-one-out cross-validation, the AUC and AUPR were higher than some existing methods. In addition, case studies have been done to verify our results. We also discussed the trajectory of social avoidance and distress during acute survival of breast cancer patients.