FairPy: A Toolkit for Evaluation of Prediction Biases and their Mitigation in Large Language Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Viswanath, Hrishikesh, Zhang, Tianyi
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916689433591808
author Viswanath, Hrishikesh
Zhang, Tianyi
author_facet Viswanath, Hrishikesh
Zhang, Tianyi
contents Recent studies have demonstrated that large pretrained language models (LLMs) such as BERT and GPT-2 exhibit biases in token prediction, often inherited from the data distributions present in their training corpora. In response, a number of mathematical frameworks have been proposed to quantify, identify, and mitigate these the likelihood of biased token predictions. In this paper, we present a comprehensive survey of such techniques tailored towards widely used LLMs such as BERT, GPT-2, etc. We additionally introduce Fairpy, a modular and extensible toolkit that provides plug-and-play interfaces for integrating these mathematical tools, enabling users to evaluate both pretrained and custom language models. Fairpy supports the implementation of existing debiasing algorithms. The toolkit is open-source and publicly available at: \href{https://github.com/HrishikeshVish/Fairpy}{https://github.com/HrishikeshVish/Fairpy}
format Preprint
id arxiv_https___arxiv_org_abs_2302_05508
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle FairPy: A Toolkit for Evaluation of Prediction Biases and their Mitigation in Large Language Models
Viswanath, Hrishikesh
Zhang, Tianyi
Computation and Language
Artificial Intelligence
Machine Learning
Recent studies have demonstrated that large pretrained language models (LLMs) such as BERT and GPT-2 exhibit biases in token prediction, often inherited from the data distributions present in their training corpora. In response, a number of mathematical frameworks have been proposed to quantify, identify, and mitigate these the likelihood of biased token predictions. In this paper, we present a comprehensive survey of such techniques tailored towards widely used LLMs such as BERT, GPT-2, etc. We additionally introduce Fairpy, a modular and extensible toolkit that provides plug-and-play interfaces for integrating these mathematical tools, enabling users to evaluate both pretrained and custom language models. Fairpy supports the implementation of existing debiasing algorithms. The toolkit is open-source and publicly available at: \href{https://github.com/HrishikeshVish/Fairpy}{https://github.com/HrishikeshVish/Fairpy}
title FairPy: A Toolkit for Evaluation of Prediction Biases and their Mitigation in Large Language Models
topic Computation and Language
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2302.05508