将贝叶斯思想融入在线置信预测,实现无假设下多置信度精准校准。
The Benefit of Being Bayesian in Online Conformal Prediction
- 用贝叶斯先验正则化经验分布,构建鲁棒信念更新机制。
- 无需独立同分布假设,任意置信水平误差低且无单调性问题。
- 数据服从独立同分布时自动获得正确覆盖率,适合实际部署。
基于置信预测(CP)框架,研究在黑箱机器学习模型下在线构建置信集的方法。通过将目标置信水平转化为分位数水平,问题可简化为对顺序呈现数据序列的分位数进行事后预测。此前有两种方法:(i) 假设数据为独立同分布或可交换,维护观测数据的经验分布作为算法信念,直接预测其分位数;(ii) 为避免统计假设脆弱性,近年趋势转向非分布、对抗设置,采用一阶在线优化算法处理移动分位数损失。但该方法需已知目标分位数水平,且因损失线性化导致此前未被注意的单调性问题。本文提出一种自适应CP算法,融合两者优势:无需统计假设即可应对任意置信水平查询,低遗憾且克服单调性缺陷。若数据确实为独立同分布,则同一算法自动具备正确覆盖率保证。关键技术在于用贝叶斯先验正则化前述算法信念(经验分布),通过模拟非线性化的跟随规则领导者(FTRL)算法实现鲁棒更新。该信念更新机制由不同置信水平的预测头共享,带来类似最近提出的U校准(Kleinberg et al., 2023)的实际优势。
原文摘要 · Abstract (English)
Based on the framework of Conformal Prediction (CP), we study the online construction of confidence sets given a black-box machine learning model. By converting the target confidence levels into quantile levels, the problem can be reduced to predicting the quantiles (in hindsight) of a sequentially revealed data sequence. Two very different approaches have been studied previously: (i) Assuming the data sequence is iid or exchangeable, one could maintain the empirical distribution of the observed data as an algorithmic belief, and directly predict its quantiles. (ii) Due to the fragility of statistical assumptions, a recent trend is to consider the non-distributional, adversarial setting and apply first-order online optimization algorithms to moving quantile losses. However, it requires the oracle knowledge of the target quantile level, and suffers from a previously overlooked monotonicity issue due to the associated loss linearization. This paper presents an adaptive CP algorithm that combines their strengths. Without any statistical assumption, it is able to answer multiple arbitrary confidence level queries with low regret, while also overcoming the monotonicity issue suffered by first-order optimization baselines. Furthermore, if the data sequence is actually iid, then the same algorithm is automatically equipped with the "correct" coverage probability guarantee. To achieve such strengths, our key technical innovation is to regularize the aforementioned algorithmic belief (the empirical distribution) by a Bayesian prior, which robustifies it by simulating a non-linearized Follow the Regularized Leader (FTRL) algorithm on the output. Such a belief update backbone is shared by prediction heads targeting different confidence levels, bringing practical benefits analogous to the recently proposed concept of U-calibration (Kleinberg et al., 2023).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。