基于Transformer和LSTM的药物相互作用预测

药物相互作用的不良反应已经成为消化系统疾病、心血管疾病等发病率升高的重要原因之一,并且导致药物退出市场,因此准确预测药物相互作用受到了广泛关注.针对传统Encoder-Decoder模型无法捕捉药物子结构之间依赖的问题,提出了基于Transformer和LSTM的药物相互作用预测模型TransDDI(TransformerDDI).TransDDI包括数据预处理模块、潜在特征抽取模块和映射模块3部分.数据预处理模块利用SPM算法从药物的SMILES格式输入中提取出表征药物的频繁子结构,形成药物特征向量,进而生成药物对的特征向量.潜在特征抽取模块利用Transformer充分挖掘特征向量中子结构之间蕴含的信息,突出不同子结构的不同重要作用,生成潜在特征向量.映射模块主要是将药物对的潜在特征向量和数据库中频繁子结构的向量形成字典表示,并且利用融合了LSTM的神经网络进行预测.在真实数据集BIOSNAP和DrugBank上,将所提模型与另外6种机器学习、深度学习方法进行实验比较.结果显示,TransDDI准确率更高,便于药物相互作用预测.