ProLLM：蛋白质「思维链」增强大模型，预测蛋白质相互作用

图1：ProLLM 框架核心原理——将蛋白质信号通路转化为 Chain-of-Thought 提示词

一、痛点切入：当蛋白质「打电话」时，AI 听得懂吗？

想象一下：你身体里的每一个细胞，就像一座繁忙的城市。城市中十万计的「居民」——蛋白质，无时无刻不在相互「打电话」「发微信」，传递着维持生命的信号。这些信号通路一旦出错，就可能引发癌症、阿尔茨海默病、糖尿病等重大疾病。

传统方法怎么研究蛋白质相互作用（Protein-Protein Interaction, PPI）？生物学家往往依赖昂贵的实验（酵母双杂交、免疫共沉淀），一次只能验证几十对蛋白质。面对人体超过 650,000 种潜在蛋白质相互作用对，这种「一个一个来」的方式效率极低。

ProLLM 的出现，像是为 AI 装上了一副「生物语」耳机。 它让大语言模型能够「听懂」蛋白质说的话——把蛋白质序列翻译成自然语言，再用 Chain-of-Thought（思维链）的方式，模拟信号通路的传递逻辑，预测任意两个蛋白质是否会「通话」。

二、项目起源：顶会认可的新范式

ProLLM 全称 Protein Chain-of-Thoughts Enhanced LLM，由 MingyuJ666 团队开发，于 2024 年被国际机器学习顶级会议 COLM 2024 正式接收，arXiv 论文地址：https://arxiv.org/abs/2405.06649。

图2：ProLLM 完整工作流程

项目的核心创新是 ProCoT（Protein Chain of Thought）方法——受神经生物学中信号通路的级联放大效应启发，将蛋白质相互作用预测重新建模为链式推理过程：上游蛋白 → 若干中间蛋白 → 下游蛋白，就像多米诺骨牌一样有序传递。

三、技术原理：用「生物语言」提示大模型

3.1 为什么需要思维链？

普通的 PPI 预测模型通常输出一个简单的「是/否」或概率值。ProLLM 的思路不同：它让 LLM 先「思考」——给定一对待预测的蛋白质，模型会生成一段自然语言推理链，解释这对蛋白可能通过哪些中间蛋白建立联系，最终给出预测结论。

这种 Chain-of-Thought 方式的好处在于：

• 可解释性：不仅给出结果，还能展示推理路径，生物学家可以验证逻辑是否合理
• 准确率提升：实验表明，CoT 提示使 LLM 的 PPI 预测准确率显著高于直接问答模式
• 泛化能力：自然语言推理链帮助模型捕捉超出训练数据分布的新蛋白质组合

3.2 模型架构

ProLLM 基于 FLAN-T5（Google 开源的指令微调 T5 模型）进行开发。核心训练流程：

蛋白质序列 → 自然语言嵌入 → Chain-of-Thought 推理 → PPI 预测

技术栈包括：

• 模型：FLAN-T5（Seq2Seq LM），通过 HuggingFace Transformers 加载
• 训练框架：HuggingFace Trainer + Accelerate（分布式训练加速）
• 数据格式：CSV，包含 input_text（蛋白质序列经 ProCoT 模板化后的自然语言描述）和 output_text（预测结果）
• 嵌入替换：使用 ESM2 等蛋白质语言模型预计算的 embedding 替换原始 token，提升表示能力

3.3 数据集与预处理

ProLLM 在四个标准 PPI 数据集上训练和评估：

• SHS27K：1,748 个蛋白质，27,236 个相互作用对
• SHS148K：3,718 个蛋白质，56,242 个相互作用对
• STRING：经典大规模 PPI 数据库
• Human：人类蛋白质组数据

预处理脚本将原始蛋白质相互作用数据转换为 ProCoT 格式的 CSV 文件，每个样本的 input_text 形如：

"Given protein A (sequence: MKFL...), protein B (sequence: MALL...), identify the signaling pathway. Starting from protein A, it passes through intermediate protein(s) X, finally reaching protein B. Is there an interaction?"

output_text 则是对应的「是/否」标签或中间蛋白推理链。

四、核心功能与使用方式

4.1 训练模型

# 安装依赖
pip install -r requirements.txt

# 下载数据集（Google Drive）
# 下载 SHS27K / SHS148K / STRING / Human 数据集

# 数据预处理：生成 ProCoT 格式
python data_preprocess/SHS27K/preprocess.py

# 执行嵌入替换（需提前用 ESM2 生成 embedding）
python embedding_replacement/embedding_replacement.py

# 微调 FLAN-T5
python train_ProLLM.py \
    --model_dir /path/to/FLAN-T5 \
    --data_file SHS27K_train_10.csv \
    --num_epochs 3 \
    --batch_size 2 \
    --learning_rate 3e-4 \
    --output_dir ./output

4.2 模型推理与评估

python eva.py \
    --model_path /path/to/trained_model \
    --test_df /path/to/test_data.csv

推理使用 T5ForConditionalGeneration，配置 num_beams=5 束搜索和 max_new_tokens=1500 的生成长度上限。

五、上手门槛与局限

适合谁： 有机器学习背景的生物信息学研究者和 NLP 工程师。需要一定的 Python 能力，熟悉 PyTorch 和 HuggingFace 生态。

主要局限：

1. 无 Web 界面：所有操作通过命令行执行，非生物信息专业人员上手成本高
2. 无容器化支持：没有提供 Dockerfile 或 docker-compose，无法一键部署
3. 嵌入依赖外部：ESM2 嵌入需要自行计算并替换，项目仓库中不包含预计算好的 embedding
4. 数据集需手动下载：依赖 Google Drive 链接，下载不便
5. 许可信息缺失：项目未声明开源许可证，使用前需联系作者确认
6. GPU 显存要求高：训练推荐 16GB+ 显存（H100/A100 级别），个人研究难度大

六、行业意义与未来方向

ProLLM 是 LLM for Science 趋势的一个典型案例——将通用语言模型的推理能力迁移到生物科学领域。它证明了大模型不仅能「写文章」，还能「做生物实验规划」。

从更宏观的角度看，ProLLM 代表了一个新兴方向：用自然语言作为分子生物学的「中间表示层」。蛋白质序列→自然语言→LLM推理→预测结果，这套 pipeline 为多模态生物 AI 提供了新思路。

作者在 COLM 2024 的论文中报告了对比基线的显著提升，相关代码和数据完全开源，期待更多研究者在此基础上探索蛋白质语言模型的潜力。

---

本报告基于 GitHub 仓库 v1.0（2024-03-14 创建，2025-11-23 最后更新）生成。