发明公开
- 专利标题: 一种分类模型的输出校准方法、装置及可读存储介质
-
申请号: CN202310957777.6申请日: 2023-07-31
-
公开(公告)号: CN116975690A公开(公告)日: 2023-10-31
- 发明人: 谢林柏 , 张思源 , 张善新 , 彭力
- 申请人: 江南大学
- 申请人地址: 江苏省无锡市经开区金融八街1号无锡商会大厦2201
- 专利权人: 江南大学
- 当前专利权人: 江南大学
- 当前专利权人地址: 江苏省无锡市经开区金融八街1号无锡商会大厦2201
- 代理机构: 苏州市中南伟业知识产权代理事务所
- 代理商 汤婧怡
- 主分类号: G06F18/24
- IPC分类号: G06F18/24 ; G06F18/214 ; G06F18/10
摘要:
本发明涉及一种分类模型的输出校准方法、装置及可读存储介质,属于计算机技术领域。包括:基于输出到概率映射后分类决策不变性原则构建校准模型;获取训练集并将训练集划分为多个训练子集,将训练子集输入至分类模型,基于分类结果计算分类模型的平均准确度;将分类结果输入至校准模型,计算分类模型的平均置信度;以分类模型的平均准确度和平均置信度的均方期望误差最小为目标构建优化函数,使用随机梯度下降法调整校准模型的参数,利用训练集中的训练子集不断进行迭代直到优化函数最小,得到训练好的校准模型。本发明提供的方法能够有效调整分类模型的输出概率,提高分类模型的置信度,使校准后的分类模型能够更好地应用于高风险决策领域。
公开/授权文献
- CN116975690B 一种分类模型的输出校准方法、装置及可读存储介质 公开/授权日:2024-06-07