一种基于深度学习的辅助筛选抗癌药物组合协同预测方法

本发明公开了一种基于深度学习的辅助筛选抗癌药物组合协同预测方法，采用TIAM模型，TIAM模型进一步考虑了三元关系，TIAM使用三个输入计算出一个三维注意力张量Attention3d来表示药物‑药物‑细胞系之间的三元关系，因此在TIAM中有两个Query，即Query1，Query2，Key，Value。TIAM要求三个输入，分别是药物a，药物b，细胞系c，分别用Xa、Xb和Xc表示，经过TIAM的处理输出后得到X′a、X′b和X′c。本发明采用三元注意力机制实现了药物‑药物‑细胞系的特征提取，相比现有的技术更加准确，有关联性。