让模型在不确定时选择专家或拒绝回答,提升准确率与效率。
Theory and Algorithms for Learning with Multi-Class Abstention and Multi-Expert Deferral
- 设计新型损失函数,支持多专家拒答与多类拒绝决策。
- 在CIFAR-10/100、SVHN上验证算法性能优于现有方法。
- 适用于高可靠性需求场景,如医疗、金融等关键任务。
大型语言模型虽表现优异,但面临幻觉和推理成本高的挑战。利用多个专家:将不确定输入转给更强大专家可提升可靠性,简单问题交由小型精简模型处理则提高效率。这引出了多专家拒答的学习问题。本文系统研究该问题及相关的拒答学习问题,并提供强一致性保障。首先针对拒答(拒答的特例),分析基于得分与预测器-拒绝器的多分类框架,提出新的代理损失族,并证明其具有强非渐近、特定假设集的一致性,解决了两个开放问题。同时研究单阶段与实用双阶段设置,在CIFAR-10、CIFAR-100与SVHN数据集上实验表明新算法表现更优。其次,针对一般多专家拒答分类,设计适用于单阶段与双阶段的新代理损失,并证明其具备强$H$-一致性界。对于双阶段场景,证明其在恒定成本函数下为可实现$H$-一致,从而导出有效新算法。最后,提出回归任务中拒答的全新框架,适用于连续标签空间,兼容多专家与多种成本结构,支持单阶段与双阶段方法,涵盖近期拒答回归工作。提出具$H$-一致性保证的新代理损失,并实证验证算法有效性。
原文摘要 · Abstract (English)
Large language models (LLMs) have achieved remarkable performance but face critical challenges: hallucinations and high inference costs. Leveraging multiple experts offers a solution: deferring uncertain inputs to more capable experts improves reliability, while routing simpler queries to smaller, distilled models enhances efficiency. This motivates the problem of learning with multiple-expert deferral. This thesis presents a comprehensive study of this problem and the related problem of learning with abstention, supported by strong consistency guarantees. First, for learning with abstention (a special case of deferral), we analyze score-based and predictor-rejector formulations in multi-class classification. We introduce new families of surrogate losses and prove strong non-asymptotic, hypothesis set-specific consistency guarantees, resolving two existing open questions. We analyze both single-stage and practical two-stage settings, with experiments on CIFAR-10, CIFAR-100, and SVHN demonstrating the superior performance of our algorithms. Second, we address general multi-expert deferral in classification. We design new surrogate losses for both single-stage and two-stage scenarios and prove they benefit from strong $H$-consistency bounds. For the two-stage scenario, we show that our surrogate losses are realizable $H$-consistent for constant cost functions, leading to effective new algorithms. Finally, we introduce a novel framework for regression with deferral to address continuous label spaces. Our versatile framework accommodates multiple experts and various cost structures, supporting both single-stage and two-stage methods. It subsumes recent work on regression with abstention. We propose new surrogate losses with proven $H$-consistency and demonstrate the empirical effectiveness of the resulting algorithms.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。