EPIC-TRACE: predicting TCR binding to unseen epitopes using attention and contextualized embeddings

Motivation T cells play an essential role in adaptive immune system to fight pathogens and cancer but may also give rise to autoimmune diseases. The recognition of a peptide-MHC (pMHC) complex by a T cell receptor (TCR) is required to elicit an immune response. Many machine learning models have been developed to predict the binding, but generalizing predictions to pMHCs outside the training data remains challenging. Results We have developed a new machine learning model that utilizes information about the TCR from both α and β chains, epitope sequence, and MHC. Our method uses ProtBERT embeddings for the amino acid sequences of both chains and the epitope, as well as convolution and multi-head attention architectures. We show the importance of each input feature as well as the benefit of including epitopes with only a few TCRs to the training data. We evaluate our model on existing databases and show that it compares favorably against other state-of-the-art models.

DOI

10.1093/bioinformatics/btad743

Citation

Korpela, D, Jokinen, E, Dumitrescu, A, Huuhtanen, J, Mustjoki, S & Lähdesmäki, H 2023, ' EPIC-TRACE: predicting TCR binding to unseen epitopes using attention and contextualized embeddings ', Bioinformatics, vol. 39, no. 12, btad743 . https://doi.org/10.1093/bioinformatics/btad743

Permanent link to this item

https://urn.fi/URN:NBN:fi:aalto-202401171477

Collections

[article-cris] Perustieteiden korkeakoulu / SCI

Show all metadata

EPIC-TRACE: predicting TCR binding to unseen epitopes using attention and contextualized embeddings

Access rights

URL

Journal Title

Journal ISSN

Volume Title

Authors

Date

Department

Major/Subject

Mcode

Degree programme

Language

Pages

Series

Abstract

Description

Keywords

Other note

DOI

Citation

Permanent link to this item

Collections