Coverage-Guaranteed Prediction Sets for Out-of-Distribution Data

التفاصيل البيبلوغرافية
العنوان: Coverage-Guaranteed Prediction Sets for Out-of-Distribution Data
المؤلفون: Zou, Xin, Liu, Weiwei
المصدر: AAAI (2024) Vol. 38, No. 15, pages 17263-17270
سنة النشر: 2024
المجموعة: Computer Science
مصطلحات موضوعية: Computer Science - Machine Learning
الوصف: Out-of-distribution (OOD) generalization has attracted increasing research attention in recent years, due to its promising experimental results in real-world applications. In this paper,we study the confidence set prediction problem in the OOD generalization setting. Split conformal prediction (SCP) is an efficient framework for handling the confidence set prediction problem. However, the validity of SCP requires the examples to be exchangeable, which is violated in the OOD setting. Empirically, we show that trivially applying SCP results in a failure to maintain the marginal coverage when the unseen target domain is different from the source domain. To address this issue, we develop a method for forming confident prediction sets in the OOD setting and theoretically prove the validity of our method. Finally, we conduct experiments on simulated data to empirically verify the correctness of our theory and the validity of our proposed method.
نوع الوثيقة: Working Paper
DOI: 10.1609/aaai.v38i15.29673
URL الوصول: http://arxiv.org/abs/2403.19950
رقم الأكسشن: edsarx.2403.19950
قاعدة البيانات: arXiv
الوصف
DOI:10.1609/aaai.v38i15.29673