In this paper, we introduce a conformal prediction method to construct prediction sets in a oneshot federated learning setting. More specifically, we define a quantile-of-quantiles estimator and prove that for any distribution, it is possible to output prediction sets with desired coverage in only one round of communication. To mitigate privacy issues, we also describe a locally differentially private version of our estimator. Finally, over a wide range of experiments, we show that our method returns prediction sets with coverage and length very similar to those obtained in a centralized setting. Overall, these results demonstrate that our method is particularly well-suited to perform conformal predictions in a one-shot federated learning setting.
翻译:在本文中,我们提出了一种共形预测方法,用于在一次性联邦学习场景中构建预测集。具体而言,我们定义了一种分位数之分的估计量,并证明对于任意分布,仅通过一轮通信即可输出具有期望覆盖率的预测集。为缓解隐私问题,我们还描述了该估计量的局部差分隐私版本。最后,通过大量实验表明,我们的方法返回的预测集在覆盖率和长度上与集中式场景中获得的预测集非常相似。总体而言,这些结果证明该方法特别适用于一次性联邦学习环境中的共形预测。