Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

COFT: Counterfactual-Conformal Decoding for Fair Chain-of-Thought Reasoning in Large Language Models

About

Large language models (LLMs) can reveal and amplify societal biases during chain-of-thought (CoT) generation. We present COFT (Chain of Fair Thought), a training-free decoding method that applies token-level fairness control at decode time, with distribution-free marginal validity guarantees (under exchangeability) for any frozen causal language model. COFT operates in three stages. First, it creates a masked counterfactual prompt by replacing sensitive spans with neutral tokens. Second, it compares the factual and masked logit distributions through lightweight logit fusion to attenuate attribute-driven biases. Third, it uses dual-branch split-conformal calibration to certify per-step candidate token sets at a user-chosen risk level. We evaluate COFT across six models and multiple bias benchmarks. Our method reduces standard bias metrics by 30-55% (median 38%) while preserving task utility and language quality. Reasoning accuracies remain unchanged within run-to-run noise margins. The computational overhead is modest, equivalent to one additional cached forward pass (<=11%). COFT offers a clear, auditable path to safer CoT generation with significant bias reduction, negligible utility loss, and no requirement for retraining, auxiliary classifiers, or weight access.

Arya Fayyazi, Mehdi Kamal, Massoud Pedram• 2026

Related benchmarks

TaskDatasetResultRank
Math ReasoningGSM8K
Accuracy51
254
Toxicity EvaluationBoLD--
26
Commonsense ReasoningStrategyQA
Accuracy (%)73.6
24
Science QAARC Easy
Accuracy77.8
17
Fairness evaluationUtrecht
Utrecht DP11.2
10
Fairness evaluationCOMPAS
COMPAS Gap0.113
10
Stereotypical Bias EvaluationCrowS-Pairs (CP)
CP Accuracy64.7
10
Language ModelingWikiText-2
Perplexity (PPL)13.9
10
Text generation qualityOpenAI Summaries
MAUVE0.81
10
Physical Commonsense ReasoningPIQA
Accuracy (PIQA)79.5
10
Showing 10 of 14 rows

Other info

Follow for update