Inference-Time Selective Debiasing

التفاصيل البيبلوغرافية
العنوان: Inference-Time Selective Debiasing
المؤلفون: Kuzmin, Gleb, Yadav, Neemesh, Smirnov, Ivan, Baldwin, Timothy, Shelmanov, Artem
سنة النشر: 2024
المجموعة: Computer Science
مصطلحات موضوعية: Computer Science - Computation and Language, Computer Science - Artificial Intelligence
الوصف: We propose selective debiasing -- an inference-time safety mechanism that aims to increase the overall quality of models in terms of prediction performance and fairness in the situation when re-training a model is prohibitive. The method is inspired by selective prediction, where some predictions that are considered low quality are discarded at inference time. In our approach, we identify the potentially biased model predictions and, instead of discarding them, we debias them using LEACE -- a post-processing debiasing method. To select problematic predictions, we propose a bias quantification approach based on KL divergence, which achieves better results than standard UQ methods. Experiments with text classification datasets demonstrate that selective debiasing helps to close the performance gap between post-processing methods and at-training and pre-processing debiasing techniques.
نوع الوثيقة: Working Paper
URL الوصول: http://arxiv.org/abs/2407.19345
رقم الأكسشن: edsarx.2407.19345
قاعدة البيانات: arXiv