May, 2024

层次化的选择性分类

TL;DR部署深度神经网络用于风险敏感任务需要一个不确定性估计机制。本文介绍了分层选择分类,将选择分类扩展到分层环境中。我们的方法利用类别关系的固有结构,使模型在面临不确定性时减少其预测的特异性。本文首先形式化了分层风险和覆盖,并介绍了分层风险覆盖曲线。接下来,我们开发了针对分层选择分类的算法(也称为“推理规则”),并提出了一种能够以高概率保证目标准确性约束的高效算法。最后,我们对一千多个ImageNet分类器进行了大量的实证研究,揭示了诸如CLIP、ImageNet21k预训练和知识蒸馏等训练制度提升了分层选择性能。