[发明专利]基于可信图神经网络的决策可靠性评估方法、装置及设备

说明书

本申请涉及一种基于可信图神经网络的决策可靠性评估方法、装置及设备。所述方法包括：根据初始图神经网络模型输出的logit向量得到初始图神经网络模型的置信度，利用初始图神经网络模型对决策样本进行决策，得到决策结果的准确率；根据置信度和决策结果的准确率进行计算，得到初始图神经网络模型的预期校准误差；构建初始图神经网络模型的图校准损失函数，利用图校准损失函数对初始图神经网络模型进行训练，得到可信图神经网络模型；根据可信图神经网络模型的校准误差和可信图神经网络模型进行决策评估。采用本方法能够提高决策可靠性。

技术领域

本申请涉及数据处理技术领域，特别是涉及一种基于可信图神经网络的决策可靠性评估方法、装置、计算机设备和存储介质。

背景技术

车辆自动驾驶技术是为目前蓬勃发展的技术之一，而在自动驾驶的日常情境中，都需要通过感测器感测周围环境的资讯，用以自动控制车辆的动作，例如，在交通路口、车道变换以及加减速的过程中，都需要通过感测器感测周围环境的资讯，以控制车辆的动作。然而，当感测器感测及输出的资讯有失真、屏蔽或者失效的情况出现，将使得车辆的自动控制产生误差。现今的自动驾驶技术包含有一决策中心，决策中心包含多个决策模型，决策模型用于对自动驾驶过程中一些突发情况进行决策，但是往往多个决策模型在相关环境下进行决策时会出现误差，需要提出一个对决策模型进行可靠性评估的方法，衡量一个决策模型是否可以用于真实决策场景。

发明内容

基于此，有必要针对上述技术问题，提供一种能够提高决策可靠性的基于可信图神经网络的决策可靠性评估方法、装置、计算机设备和存储介质。

一种基于可信图神经网络的决策可靠性评估方法，所述方法包括：

获取初始图神经网络模型和测试集；图神经网络模型为决策模型；测试集包括决策样本和决策结果；

根据初始图神经网络模型输出的logit向量得到初始图神经网络模型的置信度，利用初始图神经网络模型对决策样本进行决策，得到决策结果的准确率；

根据置信度和决策结果的准确率进行计算，得到初始图神经网络模型的预期校准误差；

根据预期校准误差大小判断初始图神经网络模型是否需要进行校准，根据判断结果构建初始图神经网络模型的图校准损失函数，利用图校准损失函数对初始图神经网络模型进行训练，得到可信图神经网络模型；

根据可信图神经网络模型的校准误差和可信图神经网络模型进行决策评估。

在其中一个实施例中，根据置信度和决策结果的准确率进行计算，得到初始图神经网络模型的预期校准误差，包括：

根据置信度和决策结果的准确率进行计算，得到初始图神经网络模型的预期校准误差为其中，表示决策结果的标签，y_i表示测试集里的真实标签，表示预测分布，为网络做出预测决策的置信度，p表示目标分布，为网络做出正确决策的真实概率。

在其中一个实施例中，根据预期校准误差大小判断初始图神经网络模型是否需要进行校准，根据判断结果构建初始图神经网络模型的图校准损失函数，包括：

根据预期校准误差大小判断初始图神经网络模型是否需要进行校准，若预期校准误差大于预先设置的误差阈值，则判断初始图神经网络模型需要被校准，构建初始图神经网络模型的图校准损失函数。

在其中一个实施例中，构建初始图神经网络模型的图校准损失函数的过程包括：

获取目标分布和预测分布；目标分布为网络做出正确决策的真实概率；预测分布为初始图神经网络模型做出预测决策的置信度；

对目标分布和预测分布进行散度计算，得到KL散度；

根据目标分布和预测分布的熵的乘积构建正则化器，将KL散度和正则化器的正则化计算，得到图校准损失函数。