基于置信度的级联延迟何时足够?
机器学习
2024-01-24 v2 机器学习
摘要
级联是一种使推断成本随样本自适应变化的经典策略,其中依次调用一系列分类器。延迟规则决定是否调用序列中的下一个分类器,或终止预测。一种简单的延迟规则采用当前分类器的置信度,例如基于最大预测softmax概率。尽管此类基于置信度的延迟对级联结构一无所知——例如未对下游模型的误差建模——但其常在实践中表现极佳。本文旨在更好地理解基于置信度的延迟可能失效的条件,以及何时替代延迟策略能表现更优。我们首先给出最优延迟规则的理论刻画,精确刻画基于置信度的延迟可能受损的设定。随后我们研究事后延迟机制,并证明在以下设定中其能显著优于基于置信度的延迟:(i)下游模型为仅在一部分输入上表现良好的专用模型,(ii)样本受标签噪声影响,以及(iii)训练集与测试集间存在分布偏移。
引用
@article{arxiv.2307.02764,
title = {When Does Confidence-Based Cascade Deferral Suffice?},
author = {Wittawat Jitkrittum and Neha Gupta and Aditya Krishna Menon and Harikrishna Narasimhan and Ankit Singh Rawat and Sanjiv Kumar},
journal= {arXiv preprint arXiv:2307.02764},
year = {2024}
}
备注
NeurIPS 2023