关键短语生成模型的训练方法、关键短语生成方法及装置

    公开(公告)号:CN119514539A

    公开(公告)日:2025-02-25

    申请号:CN202411342995.X

    申请日:2024-09-25

    Applicant: 厦门大学

    Abstract: 本申请的实施例提供了一种关键短语生成模型的训练方法、关键短语生成方法及装置。该训练方法包括:获取训练文档及其对应的若干关键短语真值;基于最优传输理论,将若干关键短语真值分配给若干控制码;将训练文档输入至生成器中,以使生成器输出若干候选关键短语;将若干候选关键短语输入至选择器中,以使选择器输出决策标签,决策标签用以指示保留或丢弃对应的候选关键短语;构建生成器对应的第一损失函数,以及选择器对应的第二损失函数,并分别根据第一损失函数和第二损失函数对生成器和选择器进行调优,以得到目标关键短语生成模型。本申请实施例的技术方案可以兼顾关键短语生成的精确度和召回率,保证关键短语生成质量。

Patent Agency Ranking