基于信息瓶颈训练的小分子虚拟筛选方法及装置

本发明提供一种基于信息瓶颈训练的小分子虚拟筛选方法及装置，涉及人工智能技术领域。其中，基于信息瓶颈训练的小分子虚拟筛选方法包括：利用药物小分子预训练模型和初始未扰动编码器，根据药物小分子样本的标识以及未扰动转录组数据样本，得到药物小分子样本和未扰动转录组数据样本的融合向量表示；将融合向量表示与其对应的转录组变化量样本的预测向量表示构成正样本对，将融合向量表示与不对应的转录组变化量样本的预测向量表示构成负样本对，基于对比损失函数和信息瓶颈损失函数，对药物小分子预训练模型、初始转录组变化编码器、初始未扰动编码器进行对比损失训练。本发明可以精准且鲁棒地进行药物小分子的虚拟筛选。