用图神经网络解决蛋白推断难题,提升效率与泛化能力
GraphPI: Efficient Protein Inference with Graph Neural Networks

- 将蛋白推断建模为节点分类问题,构建蛋白-肽段-质谱图
- 在无标签数据上通过自训练生成伪标签,显著减少标注依赖
- 无需微调即可跨数据集通用,计算速度远超传统算法
深度学习在生物医学研究中已带来变革性进展,但在蛋白推断中的应用受限于高质量标注数据的匮乏,且蛋白注释成本高、流程复杂。本文提出GraphPI,将蛋白推断视为节点分类任务,构建包含蛋白、肽段与质谱匹配(PSM)的图结构,利用图神经网络挖掘其内在关联。为缓解标签稀缺问题,模型在多个公开无标签蛋白数据集上训练,采用已有蛋白推断算法生成伪标签,并通过自训练机制依据置信度迭代优化标签。实验表明,由于Percolator特征具有良好的归一化特性,GraphPI具备通用性,无需针对特定数据集微调,既降低过拟合风险,又提升计算效率。在多组测试数据上表现优异,相比主流蛋白推断算法显著缩短计算时间。
原文摘要 · Abstract (English)
The integration of deep learning approaches in biomedical research has been transformative, enabling breakthroughs in various applications. Despite these strides, its application in protein inference is impeded by the scarcity of extensively labeled datasets, a challenge compounded by the high costs and complexities of accurate protein annotation. In this study, we introduce GraphPI, a novel framework that treats protein inference as a node classification problem. We treat proteins as interconnected nodes within a protein-peptide-PSM graph, utilizing a Graph Neural Network-based architecture to elucidate their interrelations. To address label scarcity, we train the model on a set of unlabeled public protein datasets with pseudo-labels derived from an existing protein inference algorithm, enhanced by self-training to iteratively refine labels based on confidence scores. Contrary to prevalent methodologies necessitating dataset-specific training, our research illustrates that GraphPI, due to the well normalized nature of Percolator features, exhibits universal applicability without dataset-specific fine-tuning, a feature that not only mitigates the risk of overfitting but also enhances computational efficiency. Our empirical experiments reveal notable performance on various test datasets and deliver significantly reduced computation times compared to common protein inference algorithms.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。