TITAN:基于双峰注意力网络的T细胞受体特异性预测
TITAN: T Cell Receptor Specificity Prediction with Bimodal Attention Networks
AI总结:
本研究针对TCR与表位结合特异性预测任务,提出双峰注意力网络TITAN,结合SMILES原子编码、迁移学习与数据增强,在未见过的TCR预测上大幅超越现有最优模型,并在未见过的表位泛化上取得突破。
摘要:
研究背景与问题:适应性免疫系统的活性由T细胞及其特异性T细胞受体(TCR)调控,TCR可选择性识别外源抗原。实验技术的最新进展已实现对TCR及其抗原靶点(表位)的测序,使得研究TCR序列与表位结合特异性之间的缺失关联成为可能。数据稀缺与序列空间庞大使得该任务极具挑战性,截至目前,仅局限于少量表位集合的模型取得了良好性能。本研究首先将k近邻(K-NN)分类器确立为强基线,随后提出TITAN(Tcr epITope bimodal Attention Networks,TCR表位双峰注意力网络)——这是一种可显式编码TCR序列与表位的双峰神经网络,能够独立研究模型对未见过的TCR和/或表位的泛化能力。\n实验设置与结果:通过使用SMILES序列在原子水平编码表位,我们利用迁移学习和数据增强来丰富输入数据空间并提升性能。TITAN在预测未见过的TCR的特异性方面取得了高性能(10折交叉验证下ROC-AUC为0.87),且大幅超越了当前最优模型(ImRex)的结果。值得注意的是,我们基于莱文斯坦距离的K-NN分类器在未见过的TCR上也表现出具有竞争力的性能。尽管向未见过的表位的泛化仍然具有挑战性,我们报告了两项重大突破。第一,通过剖析注意力热图,我们证明了可用表位数据的稀疏性促使模型将表位隐式视为类别,这可能是一个普遍问题,会限制足够复杂的模型在未见过的表位上的性能。第二,我们表明尽管如此,TITAN在未见过的表位上仍表现出显著提升的性能,且能够将注意力集中在具有化学意义的分子结构上。
英文摘要:
Motivation: The activity of the adaptive immune system is governed by T-cells and their specific T-cell receptors (TCR), which selectively recognize foreign antigens. Recent advances in experimental techniques have enabled sequencing of TCRs and their antigenic targets (epitopes), allowing to research the missing link between TCR sequence and epitope binding specificity. Scarcity of data and a large sequence space make this task challenging, and to date only models limited to a small set of epitopes have achieved good performance. Here, we establish a k-nearest-neighbor (K-NN) classifier as a strong baseline and then propose TITAN (Tcr epITope bimodal Attention Networks), a bimodal neural network that explicitly encodes both TCR sequences and epitopes to enable the independent study of generalization capabilities to unseen TCRs and/or epitopes. Results: By encoding epitopes at the atomic level with SMILES sequences, we leverage transfer learning and data augmentation to enrich the input data space and boost performance. TITAN achieves high performance in the prediction of specificity of unseen TCRs (ROC-AUC 0.87 in 10-fold CV) and surpasses the results of the current state-of-the-art (ImRex) by a large margin. Notably, our Levenshtein-distance-based K-NN classifier also exhibits competitive performance on unseen TCRs. While the generalization to unseen epitopes remains challenging, we report two major breakthroughs. First, by dissecting the attention heatmaps, we demonstrate that the sparsity of available epitope data favors an implicit treatment of epitopes as classes. This may be a general problem that limits unseen epitope performance for sufficiently complex models. Second, we show that TITAN nevertheless exhibits significantly improved performance on unseen epitopes and is capable of focusing attention on chemically meaningful molecular structures.
机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。