Accurate structure prediction of immune proteins using parameter-efficient transfer learning
Published in BioRxiv, 2024
Author: Tian Zhu*, Milong Ren*, Zaikai He, Siyuan Tao, Ming Li, Dongbo Bu, Haicang Zhang
Abstract: Accurate prediction of immune protein structures is crucial for understanding the immune system and advancing immunotherapy development. While deep learning methods have significantly advanced protein structure prediction by extracting evolutionary constraints from homologous sequences of a target protein, they struggle with immune proteins due to the limited number of known structures and the lack of homologous sequences in hypervariable regions. To address this challenge, we propose ImmuneFold, a transfer learning approach that fine-tunes ESMFold specifically for immune proteins. We leverage low-rank adaption (LoRA), a parameter-efficient fine-tuning technique that requires considerably less memory and substantially fewer parameters. Evaluations on various immune proteins, including T-cell receptors, antibodies, and nanobodies, demonstrate that ImmuneFold outperforms existing methods in prediction accuracy. Furthermore, we apply ImmuneFold to develop a zero-shot protocol for TCR-epitope binding prediction. Unlike previous supervised methods suffering from severe overfitting due to limited experimental binding data, our approach first predicts TCR-epitope structure using ImmuneFold and then directly estimates the binding affinity by calculating Rosseta energy. Evaluations on experimental binding datasets suggest that our method is robust and accurate in predicting TCR-epitope binding. In summary, ImmuneFold demonstrates accurate predictions of immune protein structures and TCR-epitope binding, highlighting its potential to advance the development of immunotherapies.