arXiv stat.ML· Nektarios Kalampalikis, Pablo M. Olmos, Isabel Valera·· 22 天前AI 评分34
一步到位校准：联合学习准确且校准的神经网络
Calibration in One Go: Jointly Learning Accurate and Calibrated Neural Networks
AI 导读
研究人员提出一种基于双层 Stackelberg 公式的端到端校准方法，可将标准校准器与神经网络骨干网络联合训练，无需独立校准集或专用校准损失。该方案仅使用交叉熵同时优化准确率与校准性能，仅引入校准器学习率这一个额外超参数。直接将最终分类层作为校准器即可实现无需额外参数的准确校准模型，性能超越 SOTA 校准器。
来源：arXiv stat.ML · arxiv.org