一种基于对抗样本生成的药物-靶标相互作用预测方法

本发明涉及一种基于对抗样本生成的药物‑靶标相互作用预测方法，包括：获取原始数据集，基于所述原始数据集利用LLM模型生成对抗样本，构建增强数据集；构建DTI预测模型，所述DTI预测模型包括药物编码器、靶标编码器和分类器；所述药物编码器用于获取药物分子的嵌入向量表示；所述靶标编码器用于获取靶标氨基酸序列的嵌入向量表示；所述分类器用于根据药物分子的嵌入向量表示和靶标氨基酸序列的嵌入向量表示预测药物分子与靶标相互作用的概率；基于原始数据集和增强数据集对DTI预测模型进行训练；将待预测的药物分子的药物分子信息和标靶的氨基酸序列信息输入DTI预测模型预测待预测药物分子是否和标靶的氨基酸序列发生相互作用。本发明能够精确解决冷启动问题，提高了模型的鲁棒性，同时模型在面对全新未知靶标时能进行有效推断。