arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2510.17543cs.LGeess.SPstat.ML
收藏

基于共形对齐的边缘-云端模型级联中的可靠推理

Reliable Inference in Edge-Cloud Model Cascades via Conformal Alignment

King’s College London(伦敦国王学院);Centre for Intelligent Information Processing Systems(智能信息处理系统中心);Department of Engineering(工程系);Department of Informatics(信息学系);King’s Communications, Learning & Information Processing (KCLIP) lab(国王学院通信、学习与信息处理实验室)

Jiayi Huang, Sangwoo Park, Nicola Paoletti, Osvaldo Simeone

AI总结:

本文针对边缘-云端模型级联的可靠推理问题,提出基于共形对齐(CAb)的级联机制,可保证边缘预测的条件覆盖率,减少云端卸载量,在CIFAR-100和TeleQnA基准上验证了有效性。

摘要:

边缘智能通过紧凑的设备端模型实现低延迟推理,但确保推理可靠性仍是挑战。本文研究边缘-云端级联系统,要求其满足条件覆盖率:当边缘模型返回预测集合时,该集合需以用户指定概率包含真实标签,且与云端模型的输出效果一致。我们针对云端预测分布对条件覆盖率进行形式化定义,提出一种基于共形对齐(Conformal Alignment,CA)的级联机制(CAb),可在用户指定风险水平下保证该性质。该方法将边缘模型到云端模型的升级过程转化为多假设检验(Multiple-Hypothesis Testing,MHT)问题,通过定制化共形对齐选择可由边缘安全处理的输入。所提CAb模型级联方法对满足云端级条件覆盖率的边缘决策平均占比提供统计保证,适用于任意边缘预测集合,包括共形预测(Conformal Prediction,CP)的各类变体,且可在覆盖率、弃权(不执行)率与集合大小间进行可调权衡。在CIFAR-100图像分类任务和TeleQnA问答(Question-Answering,QA)基准上的实验表明,所提CAb级联方法可维持边缘预测的目标条件覆盖率,同时大幅减少向云端的卸载量,且预测集合大小仅出现适度增长。

英文摘要:

Edge intelligence enables low-latency inference via compact on-device models, but assuring reliability remains challenging. We study edge-cloud cascades that must preserve conditional coverage: whenever the edge returns a prediction set, it should contain the true label with a user-specified probability, as if produced by the cloud model. We formalize conditional coverage with respect to the cloud predictive distribution, and introduce a conformal alignment-based (CAb) cascading mechanism that certifies this property with user control over the risk level. Our method casts escalation from edge to cloud models as a multiple-hypothesis testing (MHT) problem, tailoring conformal alignment (CA) to select which inputs can be safely handled at the edge. The proposed CAb model cascading method yields statistical guarantees on the average fraction of edge decisions that satisfy cloud-level conditional coverage. The procedure applies to arbitrary edge prediction sets, including variants of conformal prediction (CP), and exposes a tunable trade-off among coverage, deferral rate, and set size. Experiments on CIFAR-100 image classification and the TeleQnA question-answering (QA) benchmark show that the proposed CAb cascade maintains the target conditional coverage for edge predictions while substantially reducing offloading to the cloud and incurring modest increases in prediction-set size.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑