T-Cell Receptor-Peptide Interaction Prediction with Physical Model Augmented Pseudo-Labeling

Publication Date: 8/14/2022

Event: KDD 2022

Reference: pp. 3090-3097, 2022

Authors: Yiren Jian, Dartmouth College; Erik Kruus, NEC Laboratories America, Inc.; Martin Renqiang Min, NEC Laboratories America, Inc.

Abstract: Predicting the interactions between T-cell receptors (TCRs) and peptides is crucial for the development of personalized medicine and targeted vaccine in immunotherapy. Current datasets for training deep learning models of this purpose remain constrained without diverse TCRs and peptides. To combat the data scarcity issue presented in the current datasets, we propose to extend the training dataset by physical modeling of TCR-peptide pairs. Specifically, we compute the docking energies between auxiliary unknown TCR-peptide pairs as surrogate training labels. Then, we use these extended example-label pairs to train our model in a supervised fashion. Finally, we find that the AUC score for the prediction of the model can be further improved by pseudo-labeling of such unknown TCR-peptide pairs (by a trained teacher model), and re-training the model with those pseudo-labeled TCR-peptide pairs. Our proposed method that trains the deep neural network with physical modeling and data-augmented pseudo-labeling improves over baselines in the available two datasets. We also introduce a new dataset that contains over 80,000 unknown TCR-peptide pairs with docking energy scores.

Publication Link: https://dl.acm.org/doi/10.1145/3534678.3539075