基于编码器与注意力机制增强的分子图像处理方法、装置

本申请涉及图像处理技术领域，公开了一种基于编码器与注意力机制增强的分子图像处理方法、装置，该方法包括：将分子图像的节点特征矩阵映射到隐藏特征空间以生成初始的节点特征；将初始的节点特征传递至单层编码器模块；通过单层编码器模块中的查询和键空间的点积计算注意力权重；基于计算得到的注意力权重加权值向量，更新节点特征；通过多头注意力机制增强的GNN模块，提取初始的节点特征中的局部邻域信息；基于更新后的节点特征与提取的局部邻域信息，生成分子图的图级表示。如此提高了分子图在复杂结构下的局部信息表达能力，为分子的整体表示提供了更加丰富和精准的数据支持。