Certifying Counterfactual Bias in LLMs

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Chaudhary, Isha, Hu, Qian, Kumar, Manoj, Ziyadi, Morteza, Gupta, Rahul, Singh, Gagandeep
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908330317840384
author Chaudhary, Isha
Hu, Qian
Kumar, Manoj
Ziyadi, Morteza
Gupta, Rahul
Singh, Gagandeep
author_facet Chaudhary, Isha
Hu, Qian
Kumar, Manoj
Ziyadi, Morteza
Gupta, Rahul
Singh, Gagandeep
contents Large Language Models (LLMs) can produce biased responses that can cause representational harms. However, conventional studies are insufficient to thoroughly evaluate biases across LLM responses for different demographic groups (a.k.a. counterfactual bias), as they do not scale to large number of inputs and do not provide guarantees. Therefore, we propose the first framework, LLMCert-B that certifies LLMs for counterfactual bias on distributions of prompts. A certificate consists of high-confidence bounds on the probability of unbiased LLM responses for any set of counterfactual prompts - prompts differing by demographic groups, sampled from a distribution. We illustrate counterfactual bias certification for distributions of counterfactual prompts created by applying prefixes sampled from prefix distributions, to a given set of prompts. We consider prefix distributions consisting random token sequences, mixtures of manual jailbreaks, and perturbations of jailbreaks in LLM's embedding space. We generate non-trivial certificates for SOTA LLMs, exposing their vulnerabilities over distributions of prompts generated from computationally inexpensive prefix distributions.
format Preprint
id arxiv_https___arxiv_org_abs_2405_18780
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Certifying Counterfactual Bias in LLMs
Chaudhary, Isha
Hu, Qian
Kumar, Manoj
Ziyadi, Morteza
Gupta, Rahul
Singh, Gagandeep
Artificial Intelligence
Machine Learning
Large Language Models (LLMs) can produce biased responses that can cause representational harms. However, conventional studies are insufficient to thoroughly evaluate biases across LLM responses for different demographic groups (a.k.a. counterfactual bias), as they do not scale to large number of inputs and do not provide guarantees. Therefore, we propose the first framework, LLMCert-B that certifies LLMs for counterfactual bias on distributions of prompts. A certificate consists of high-confidence bounds on the probability of unbiased LLM responses for any set of counterfactual prompts - prompts differing by demographic groups, sampled from a distribution. We illustrate counterfactual bias certification for distributions of counterfactual prompts created by applying prefixes sampled from prefix distributions, to a given set of prompts. We consider prefix distributions consisting random token sequences, mixtures of manual jailbreaks, and perturbations of jailbreaks in LLM's embedding space. We generate non-trivial certificates for SOTA LLMs, exposing their vulnerabilities over distributions of prompts generated from computationally inexpensive prefix distributions.
title Certifying Counterfactual Bias in LLMs
topic Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2405.18780