Learning with rejection has been a prototypical model for studying the human-AI interaction on prediction tasks. Upon the arrival of a sample instance, the model first uses a rejector to decide whether to accept and use the AI predictor to make a prediction or reject and defer the sample to humans. Learning such a model changes the structure of the original loss function and often results in undesirable non-convexity and inconsistency issues. For the classification with rejection problem, several works develop consistent surrogate losses for the joint learning of the predictor and the rejector, while there have been fewer works for the regression counterpart. This paper studies the regression with rejection (RwR) problem and investigates a no-rejection learning strategy that uses all the data to learn the predictor. We first establish the consistency for such a strategy under the weak realizability condition. Then for the case without the weak realizability, we show that the excessive risk can also be upper bounded with the sum of two parts: prediction error and calibration error. Lastly, we demonstrate the advantage of such a proposed learning strategy with empirical evidence.
翻译:带拒绝学习已成为研究人机交互在预测任务中的经典模型。当样本实例到达时,模型首先使用拒绝器决定是否接受并利用AI预测器进行预测,还是拒绝并将样本交由人类处理。学习此类模型会改变原始损失函数的结构,通常导致非凸性和不一致性等不良问题。针对带拒绝分类问题,已有若干研究为联合学习预测器与拒绝器开发了一致性代理损失函数,但回归场景下的相关研究较少。本文研究带拒绝回归(RwR)问题,探讨一种利用全部数据学习预测器的无拒绝学习策略。首先,我们在弱可实现性条件下建立了该策略的一致性。随后,针对不满足弱可实现性的情况,我们证明其过度风险可被上界约束为两部分之和:预测误差与校准误差。最后,我们通过实验证据展示了所提学习策略的优势。