mlsanityData to 14 Oct 2026

Counterfactual Reward Model Training for Bias Mitigation in Multimodal Reinforcement Learning

arXiv 2508.19567 ↗First posted 27 Aug 2025cs.LGcited by 0

On the map: LLM alignment / RLHF / preference optimisation → Reward modeling and RLHF

Abstract (arXiv)

In reinforcement learning with human feedback (RLHF), reward models can efficiently learn and amplify latent biases within multimodal datasets, which can lead to imperfect policy optimization through flawed reward signals and decreased fairness. Bias mitigation studies have often applied passive constraints, which can fail under causal confounding. Here, we present a counterfactual reward model that introduces causal inference with multimodal representation learning to provide an unsupervised, bias-resilient reward signal. The heart of our contribution is the Counterfactual Trust Score, an aggregated score consisting of four components: (1) counterfactual shifts that decompose political framing bias from topical bias; (2) reconstruction uncertainty during counterfactual perturbations; (3) demonstrable violations of fairness rules for each protected attribute; and (4) temporal reward shifts aligned with dynamic trust measures. We evaluated the framework on a multimodal fake versus true news dataset, which exhibits framing bias, class imbalance, and distributional drift. Following methodologies similar to unsupervised drift detection from representation-based distances [1] and temporal robustness benchmarking in language models [2], we also inject synthetic bias across sequential batches to test robustness. The resulting system achieved an accuracy of 89.12% in fake news detection, outperforming the baseline reward models. More importantly, it reduced spurious correlations and unfair reinforcement signals. This pipeline outlines a robust and interpretable approach to fairness-aware RLHF, offering tunable bias reduction thresholds and increasing reliability in dynamic real-time policy making.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Reward modeling and RLHF, one of 10 directions in LLM alignment / RLHF / preference optimisation. Builds on 9 of its 13 references that are in the map. First posted on arXiv 27 Aug 2025.

Builds on

9 of 13 references.

Closest ancestors · 3

Same direction, newest first

  1. 2025Towards Reward Fairness in RLHF: From a Resource Allocation Perspective29 May
  2. 2024Towards Reliable Alignment: Uncertainty-aware RLHF31 Oct
  3. 2024RewardBench: Evaluating Reward Models for Language Modeling20 Mar

Other references · 6

  1. 2023Safe RLHF: Safe Reinforcement Learning from Human Feedbackcited by 656
  2. 2024On the Algorithmic Bias of Aligning Large Language Models with RLHF: Preference Collapse and Matching Regularizationcited by 72
  3. 2023Dynamic Benchmarking of Masked Language Models on Temporal Concept Drift with Multiple Viewscited by 21
  4. 2024Unsupervised Concept Drift Detection from Deep Learning Representations in Real-timecited by 16
  5. 2024Pinpoint Counterfactuals: Reducing social bias in foundation models via localized counterfactual generationcited by 1
  6. 2025Improving Real-Time Concept Drift Detection using a Hybrid Transformer-Autoencoder Frameworkcited by 1

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025Mitigating Length Bias in RLHF through a Causal Lens
  2. 2025Bias Fitting to Mitigate Length Bias of Reward Model in RLHF
  3. 2026Reward Bias Substitution: Single-Axis Bias Mitigations Redirect Optimization Pressure
  4. 2025Debiasing Reward Models by Representation Learning with Guarantees
  5. 2025Eliminating Inductive Bias in Reward Models with Information-Theoretic Guidance
  6. 2026Mitigating Cognitive Bias in RLHF by Altering Rationality
  7. 2026Counterfactual Rollouts for Offline Contrastive Goal-Conditioned Reinforcement Learning
  8. 2024Taming Overconfidence in LLMs: Reward Calibration in RLHF