用结构+序列信息提升大模型对蛋白质的理解能力
EvoLlama: Enhancing LLMs' Understanding of Proteins via Multimodal Structure and Sequence Representations
- 融合蛋白结构与序列编码器,通过投影对齐多模态表示
- 零样本下比其他微调模型平均高1%-8%,监督微调超基线6%
- 适合需要理解蛋白质功能与结构的生物医学研究者
当前用于理解蛋白质的大语言模型主要将氨基酸序列视为文本模态。而如ESM-2等蛋白质语言模型从自然蛋白质序列中学习了海量进化知识。此外,基于结构的编码器如ProteinMPNN通过图神经网络学习蛋白质结构信息。然而,尚未探索将蛋白质编码器融入大模型以增强其蛋白质理解能力。为填补这一空白,我们提出EvoLlama,一个连接结构编码器、序列编码器与大语言模型的多模态框架。EvoLlama包含ProteinMPNN结构编码器、ESM-2序列编码器、多模态投影层及Llama-3文本解码器。训练时,我们在蛋白导向指令和蛋白属性预测数据集上进行微调,采用自然语言指令模板进行语义化标注。实验表明,EvoLlama在零样本设置下平均优于其他微调模型1%-8%,在监督微调下平均超越最先进基线6%。在蛋白属性预测任务上,结果与特定任务最优基线相当。代码将在后续版本发布。
原文摘要 · Abstract (English)
Current Large Language Models (LLMs) for understanding proteins primarily treats amino acid sequences as a text modality. Meanwhile, Protein Language Models (PLMs), such as ESM-2, have learned massive sequential evolutionary knowledge from the universe of natural protein sequences. Furthermore, structure-based encoders like ProteinMPNN learn the structural information of proteins through Graph Neural Networks. However, whether the incorporation of protein encoders can enhance the protein understanding of LLMs has not been explored. To bridge this gap, we propose EvoLlama, a multimodal framework that connects a structure-based encoder, a sequence-based protein encoder and an LLM for protein understanding. EvoLlama consists of a ProteinMPNN structure encoder, an ESM-2 protein sequence encoder, a multimodal projector to align protein and text representations and a Llama-3 text decoder. To train EvoLlama, we fine-tune it on protein-oriented instructions and protein property prediction datasets verbalized via natural language instruction templates. Our experiments show that EvoLlama's protein understanding capabilities have been significantly enhanced, outperforming other fine-tuned protein-oriented LLMs in zero-shot settings by an average of 1%-8% and surpassing the state-of-the-art baseline with supervised fine-tuning by an average of 6%. On protein property prediction datasets, our approach achieves promising results that are competitive with state-of-the-art task-specific baselines. We will release our code in a future version.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。