结合蛋白质的小分子生成方法、装置、电子设备和存储介质

本发明涉及人工智能技术领域，提供一种结合蛋白质的小分子生成方法、装置、电子设备和存储介质，方法包括：获取包含蛋白质的复合物的全原子结构；基于已训练的全原子模型中的编码模块，提取全原子结构的编码特征；基于隐空间上已训练的去噪扩散概率模型，应用全原子结构的编码特征，生成小分子特征；基于全原子模型中的解码模块，对小分子特征进行特征解码，得到对应的小分子三维结构，并基于小分子三维结构生成对应的小分子序列。本发明提供的结合蛋白质的小分子生成方法、装置、电子设备和存储介质，实现对小分子的设计，提高了生成的小分子的多样性，同时提高了小分子生成效率和准确性。