面向水库群联合调度的小样本命名实体识别方法

    公开(公告)号:CN118569260A

    公开(公告)日:2024-08-30

    申请号:CN202411036769.9

    申请日:2024-07-31

    摘要: 本发明公开了一种面向水库群联合调度的小样本命名实体识别方法,通过实体跨度检测和实体类型分类两个阶段构建模型。在跨度检测阶段使用跨度边界矩阵,学习跨度边界信息并过滤潜在错误跨度,提高检测精度;在实体类型分类阶段,通过标签引导提高原型网络构建质量。在模型训练过程中,使用对比学习改进损失函数,通过反向传播更新参数,使相同类型实体的空间表示更加接近。完成实体原型构造后,利用距离函数计算样本与类别原型之间的距离,并将其转化为相似度概率对样本进行分类。本发明实现了水库群联合调度实体的自动提取,能够在样本较少的数据集中完成命名实体识别任务,具有较高的识别效果,极大节省了人工标记所需的人力物力与时间成本。