Salvato in:
Dettagli Bibliografici
Autori principali: Stadelmaier, Josua, Malone, Brandon, Eggeling, Ralf
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:https://arxiv.org/abs/2403.12117
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866915174276923392
author Stadelmaier, Josua
Malone, Brandon
Eggeling, Ralf
author_facet Stadelmaier, Josua
Malone, Brandon
Eggeling, Ralf
contents We study the prediction of T-cell response for specific given peptides, which could, among other applications, be a crucial step towards the development of personalized cancer vaccines. It is a challenging task due to limited, heterogeneous training data featuring a multi-domain structure; such data entail the danger of shortcut learning, where models learn general characteristics of peptide sources, such as the source organism, rather than specific peptide characteristics associated with T-cell response. Using a transformer model for T-cell response prediction, we show that the danger of inflated predictive performance is not merely theoretical but occurs in practice. Consequently, we propose a domain-aware evaluation scheme. We then study different transfer learning techniques to deal with the multi-domain structure and shortcut learning. We demonstrate a per-source fine tuning approach to be effective across a wide range of peptide sources and further show that our final model is competitive with existing state-of-the-art approaches for predicting T-cell responses for human peptides.
format Preprint
id arxiv_https___arxiv_org_abs_2403_12117
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Transfer Learning for T-Cell Response Prediction
Stadelmaier, Josua
Malone, Brandon
Eggeling, Ralf
Cell Behavior
Machine Learning
We study the prediction of T-cell response for specific given peptides, which could, among other applications, be a crucial step towards the development of personalized cancer vaccines. It is a challenging task due to limited, heterogeneous training data featuring a multi-domain structure; such data entail the danger of shortcut learning, where models learn general characteristics of peptide sources, such as the source organism, rather than specific peptide characteristics associated with T-cell response. Using a transformer model for T-cell response prediction, we show that the danger of inflated predictive performance is not merely theoretical but occurs in practice. Consequently, we propose a domain-aware evaluation scheme. We then study different transfer learning techniques to deal with the multi-domain structure and shortcut learning. We demonstrate a per-source fine tuning approach to be effective across a wide range of peptide sources and further show that our final model is competitive with existing state-of-the-art approaches for predicting T-cell responses for human peptides.
title Transfer Learning for T-Cell Response Prediction
topic Cell Behavior
Machine Learning
url https://arxiv.org/abs/2403.12117