一种基于潜在扩散Transformer的三维分子性质预测方法

本发明公开了一种基于潜在扩散Transformer的三维分子性质预测方法，首先，在三维Transformer编码器中，本方法利用原子坐标计算出扭转角以及二面角信息，并将其融入到分子的潜在表征中，以充分捕捉分子的三维结构信息；其次，在潜在空间中，本方法对分子的三维潜在表征进行聚类并生成伪标签，将其作为引导扩散过程的条件，使得模型能够逐步逼近分子三维结构与其目标性质的分布关系。最后，在任务特定的解码器中，本方法将经过引导的潜在表示解码为目标分子的性质预测分布结果。与现有技术相比较，该方法相比于基于确定性模型的方法具有更强的预测性能，能够建模三维分子性质的分布，进而提升三维分子性质预测的准确率。