[发明专利]分类模型校准

摘要

分类模型校准。描述了一种对训练分类模型进行校准的计算机实现的方法。训练分类模型被训练成根据多个类对输入样本进行分类并提供相关联的预测概率，并且包括多个隐藏层和至少一个激活层。该方法包括访问训练分类模型和访问多个验证样本，每个验证样本具有基础真值标签，基础真值标签指示基础真值类。该方法进一步包括将训练分类模型应用于多个验证样本，对于每个验证样本，从在最后激活层之前的训练分类模型的层获得输出对数几率向量，以及训练校准模块。校准模块被训练来调整预测概率，预测概率从输出对数几率向量导出。校准模块包括微调子模块和分级子模块中的至少一个。该方法包括将训练校准模块附加到训练分类模型，以获得校准的分类模型。