• 论文
主办单位:煤炭科学研究总院有限公司、中国煤炭学会学术期刊工作委员会
跨粒度子图对比学习与注意力融合的药物-基因关系预测
  • 65
  • 作者

    胡冬冬彭杨谭暑秋朱小飞

  • 单位

    重庆理工大学计算机科学与工程学院

  • 摘要
    阐明药物和基因之间的相互联系是药物开发中的一个重要课题。目前,基于随机游走算法的图神经网络方法在解决药物与基因交互关系识别上已经取得了不错的效果,但是当前的方法,单一子图的方法往往容易忽略掉全局图的信息,不能够很好的聚合节点的信息,同时,药物和基因的节点表示采用简单的融合方式,不能够有效的利用节点表示的信息,用于交互关系的分类。针对上述问题,本文提出了跨粒度对比学习与注意力融合的药物-基因交互关系预测方法,一方面采用跨粒度的对比学习方法,得到远距离和近距离的节点信息,同时采用对比学习的结构增加对药物和基因节点的区分。另一方面利用注意力融合机制,充分挖掘节点中隐含的信息,将远近距离信息进行注意力融合。在2个真实数据集上的实验结果表明该模型比基线模型具有更好的分类效果。
  • 关键词

    对比学习图表示学习关系图神经网络注意力机制基因-药物关系预测

  • 文章目录
    1 相关工作
    1.1 图卷积神经网络
    1.2 多粒度子图的数据增强
    1.3 图对比学习
    2 问题定义
    3 方法与模型
    3.1 基于重启随机游走的子图抽取
    3.2 节点编码与聚合
    3.2.1 节点编码
    3.2.2 节点信息的聚合
    3.2.3 跨粒度对比学习
    3.3 注意力融合机制
    3.4 损失函数
    4 实验
    4.1 数据集
    4.2 基准方法
    4.3 实验设置
    4.4 总体实验
    4.5 消融实验
    4.6 参数敏感性分析
    4.7 每一跳最大节点数量分析
    4.8 通用性实验
    5 结束语
  • 相关文章
相关问题

主办单位:煤炭科学研究总院有限公司 中国煤炭学会学术期刊工作委员会

©版权所有2015 煤炭科学研究总院有限公司 地址:北京市朝阳区和平里青年沟东路煤炭大厦 邮编:100013
京ICP备05086979号-16  技术支持:云智互联