公开(公告)号：CN119474396A

公开(公告)日：2025-02-18

申请号：CN202411278245.0

申请日：2024-09-12

Applicant: 厦门大学

Inventor： 苏劲松 ,  闵子君 ,  刘冰帅 ,  宋佳 ,  张亮

IPC: G06F16/36 ,  G06F16/383 ,  G16C20/90 ,  G16C20/70 ,  G06N3/0455 ,  G06N3/0895

Abstract: 本发明提出了一种多粒度对齐的文本分子检索模型的训练、检索方法及装置，该训练方法包括获取分子异构图和对应的文本描述，分子异构图包括原子节点、子结构节点和分子节点，分子节点与全部子结构节点相连，每个子结构节点与其包括的原子节点相连；构建文本分子检索模型，并将文本描述输入到文本分子检索模型以得到词元表示和句子表示；将分子异构图输入到文本分子检索模型以得到原子表示、子结构表示和分子表示；将词元表示和子结构表示之间的对齐关系建模成最优传输问题，并聚合与每个子结构表示对齐的词元表示，以得到多词元表示；采用对比学习损失函数优化多粒度之间的对齐，以得到训练好的文本分子检索模型；从而提高检索结果的准确性。