A Study on Contrastive Graph Neural Network Pretraining for Predicting Transcriptome Profiles
摘要
We study graph neural network learning for transcriptomics with limited amount of labeled data. Our study reveals that simple GNN architectures perform well and do not suffer from over-fitting as the more sophisticated ones. Our study shows that although contrastive learning as a pretraining strategy has been successful in predicting properties such as formation and binding energy, it is not effective for transcriptomics. We propose attention pooling as an effective measure for improving the prediction performance. It beats all other models in our experiments. Track:CSCI-RTAI