数据构建和预处理
构建用于训练和评估的质粒序列-文本对数据集。
从PIPdb、IMG/PR、PLSDB收集质粒序列,排除大于250kb的megaplasmids,共1,521,635个质粒;使用Prodigal识别CDS并翻译蛋白,进行UniRef90聚类以减少冗余;手动注释或扩展文本属性,构建10个facets的词汇表;采用PSC划分训练、验证、测试集。
PlasRAG: comprehensive plasmid characterization and retrieval through sequence-text alignment
包含多模态模型构建、消融实验、与SOTA工具对比、案例研究及真实数据集验证,证据层级多样且包含功能验证。
质粒序列数据集(统一数据库整合) 质粒系统(PS)数据集(来自人类肠道宏基因组)
训练/验证/测试集划分采用PSC规则(ANI>76%),确保序列差异性 质粒来源:PIPdb、IMG/PR、PLSDB三个数据库,排除大于250kbp的质粒 序列编码:使用Prodigal识别CDS,ESM-2(650M)编码蛋白质,最大300个蛋白质 文本编码:MedCPT QEnc(110M),冻结参数,预计算并存储文本嵌入 训练超参数:学习率2e-4,批次大小2048,训练20轮,在单块NVIDIA A100 80GB上
提炼研究问题、关键发现与证据,快速把握文章的核心贡献。
按研究推进顺序梳理实验设计、验证步骤与关键观察。
构建用于训练和评估的质粒序列-文本对数据集。
从PIPdb、IMG/PR、PLSDB收集质粒序列,排除大于250kb的megaplasmids,共1,521,635个质粒;使用Prodigal识别CDS并翻译蛋白,进行UniRef90聚类以减少冗余;手动注释或扩展文本属性,构建10个facets的词汇表;采用PSC划分训练、验证、测试集。
训练10个facets的序列-文本对比学习模型。
使用ESM-2编码蛋白质,生成序列特征;使用MedCPT QEnc编码文本;训练Perceiver Resampler作为序列编码器,通过对比损失优化序列和文本嵌入;文本编码器参数冻结,仅训练Perceiver Resampler。
验证模型组件选择的有效性。
通过更换序列编码器(Transformer, Bi-LSTM, RNN, CNN)和文本编码器(Qwen3-8B, BioMedBERT, Llama-3, Word2Vec, text-free)进行对照实验,评估F1和P@1000;与BLASTN比对方法对比。
展示PlasRAG在查询质粒上生成全面、准确描述的能力。
输入质粒序列(如pOLA52),PlasRAG预测10个特性,检索相关文献,并通过LLM生成综合答案;进行生成结果与原始摘要的比较。
评估基于布尔查询的质粒检索性能。
使用10个facets的属性文本构建逻辑表达式,从候选质粒集合中筛选满足条件的质粒;计算召回率和精确率。
验证PlasRAG在揭示新型质粒生态学见解方面的能力。
将PlasRAG应用于PlasX PS数据集,对1,169个质粒系统(PS)进行特性预测,比较backbone和compound质粒的特性纯度、风险指数分布、宿主范围与ARGs的关系。
按研究目的归类文中使用的方法,便于定位所需技术。
| 实验环节 | 名称 | 品牌 | 货号 |
|---|---|---|---|
| 基因识别 | Prodigal | -- | -- |
| 序列特征提取 | ESM-2 | -- | ESM-2650M |
| 文本特征提取 | MedCPT QEnc | -- | BioMedBERT110M |
| 训练硬件 | NVIDIA A100 80GB GPU | NVIDIA | -- |
汇总复现实验时建议重点确认的条件及原文阅读提示。
| 环节 | 核对要点 |
|---|---|
| 数据构建 | 质粒数据库来源及版本(PIPdb、IMG/PR、PLSDB);质粒数量及排除规则(>250kb);训练/验证/测试集划分方法(PSC, ANI>76%);蛋白聚类标准(UniRef90, MMseqs2) 阅读提示:Material and methods: 'Plasmid sequence data' and 'Property text data' |
| 多模态模型训练 | 序列编码器(Perceiver Resampler)和文本编码器(MedCPT QEnc)的具体配置;训练超参数(学习率2e-4, batch size 2048, epochs 20, warm-up 0.1);损失函数(STP loss, 温度参数eα) 阅读提示:Material and methods: 'Model training' |
| 模型评估 | 评估指标(F1-score, P@1000)的具体计算公式;使用的测试集;进行多次运行以计算标准差 阅读提示:Material and methods: 'Metrics' |
| S2T表征 | LLM类型(Llama-3)和温度参数(0.2);检索阈值θ;提示模板的具体内容 阅读提示:Additional file 1 |
| T2S检索 | 查询条件的构建规则(布尔表达式);检索阈值θ;候选质粒集合的来源 阅读提示:Results: 'Case study of T2S plasmid retrieval' |