Consistency Group Relative Policy Optimization (Con-GRPO)
- Con-GRPO is a reinforcement learning algorithm that optimizes consistency across semantically equivalent queries using group-based reward computation.
- It employs within-group normalized advantage estimation and a clipped PPO surrogate loss to enforce output agreement in retrieval-augmented generation systems.
- Empirical benchmarks show significant gains in lexical and LLM-judge consistency as well as accuracy on both short- and long-form QA tasks.
Consistency Group Relative Policy Optimization (Con-GRPO) is a reinforcement learning (RL) algorithmic framework developed to directly optimize consistency objectives across groups of semantically equivalent inputs. The primary context of its application is language generation systems, particularly Retrieval-Augmented Generation (RAG), where ensuring that paraphrased queries yield consistent (i.e., stable in informational content) answers is essential for reliability, trust, and compliance in high-stakes settings. Con-GRPO instantiates Group Relative Policy Optimization (GRPO), where advantage estimation and optimization occur not for individual samples, but in a groupwise context with rewards reflecting cross-sample agreement, thereby enabling direct control of consistency properties (Hamman et al., 5 Oct 2025).
1. Formal Setting, Definitions, and Notation
Con-GRPO is typically instantiated in RAG architectures, comprising:
- A retriever that maps a query to a document set .
- A generator (parametric policy) , which yields a distribution over output sequences conditional on the query and retrieved documents:
- A paraphrase set (group) , representing semantically equivalent variants of canonical query .
For each paraphrase , 0 rollouts are sampled, generating 1, with 2 for 3, 4.
The principal consistency goal is for all outputs corresponding to any paraphrase in 5 to convey the same core information, regardless of input phrasing or retriever variability.
2. Group Similarity Rewards and Computation
Central to Con-GRPO is the group similarity reward, which incentivizes statistical agreement among the outputs for all members of the paraphrase set. Pairwise similarity between outputs is measured using a token-level function, most commonly BLEU-6 (unigram for short-form, bigram for long-form QA):
7
Each rollout's reward is its average similarity to all rollouts from other paraphrases in the group. For short-form QA where a reference 8 exists, an accuracy term 9 (e.g., token F1 or exact match) is added, forming the final reward:
0
where hyperparameters 1 (typically 2 for short-form, 3 for long-form/open-ended).
To address the 4 cost of all-pairs group reward computation, a scalable unbiased approximation samples only a subset of other paraphrases (5) and rollouts (6):
7
with 8 and 9.
In practice, 0, 1, 2, 3 suffice, resulting in linear computational scaling.
3. Policy Optimization and GRPO Objective
Rather than estimate a state-value critic, Con-GRPO employs a within-group normalization for advantage estimation. For each paraphrase 4:
- Compute the reward mean 5 and standard deviation 6:
7
- The normalized advantage is
8
for numerical stability.
The GRPO surrogate loss (PPO-style) is:
9
where 0 is the (possibly token-level) importance weight, 1 the clipping parameter, and 2 a KL-regularization coefficient to avoid policy drift.
In effect, this structure:
- Increases the probability of outputs whose normalized agreement within their paraphrase group is high.
- Penalizes significant divergence from the reference (pre-trained) policy.
4. Training Procedure, Implementation, and Hyperparameters
The Con-GRPO procedure is organized as follows:
- For each batch, sample a set of canonical queries.
- For each canonical query, generate 3 paraphrases.
- Retrieve documents and sample 4 rollouts per paraphrase.
- Compute approximate group similarity rewards per rollout.
- Normalize rewards within paraphrase groups to compute advantages.
- Accumulate the clipped GRPO loss and update parameters via backpropagation.
Key hyperparameters:
- Paraphrase set size: 5
- Rollouts per paraphrase: 6
- BLEU-7 for short-form, BLEU-8 for long-form
- Reward weights: 9, 0 (short-form), 1 (open-ended)
- KL penalty: 2 (supervised), 3 (open-ended)
- Optimizer: AdamW, learning rate 4
- Decoding temperature at training matches inference (commonly 5 for deterministic rollouts)
- Batch size: 1–2 canonical queries per GPU (dependent on model/memory)
5. Empirical Outcomes and Benchmarks
Con-GRPO, instantiated as "Information Consistent RAG" (Con-RAG), demonstrates pronounced consistency and accuracy gains relative to standard RAG and strong RL baselines (e.g., DRAG, CoT-RAG, SFT):
Short-form QA (LLaMA-3.1-8B, TriviaQA):
| Metric | RAG Baseline | Con-RAG |
|---|---|---|
| End-to-end lexical consistency (%) | 53.0 | 87.3 |
| End-to-end LLM-judge consistency (%) | 77.8 | 91.3 |
| Generator lexical consistency (%) | 67.3 | 91.2 |
| Generator LLM-judge consistency (%) | 88.5 | 93.0 |
| Exact Match accuracy (%) | 56.0 | 77.0 |
| Token F1 (%) | 66.1 | 81.0 |
Long-form QA (ELI5, LLaMA-3.1-8B):
| Metric | RAG Baseline | Con-RAG |
|---|---|---|
| End-to-end lexical consistency (%) | 8.6 | 14.6 |
| End-to-end LLM-judge consistency (%) | 62.8 | 72.7 |
| Generator lexical consistency (%) | 15.1 | 21.7 |
| Generator LLM-judge consistency (%) | 74.2 | 80.8 |
| ROUGE accuracy | 21.9 | 24.2 |
| LLM-judge accuracy (%) | 74.0 | 78.0 |
Con-GRPO continues to outperform supervised-fine-tuning (SFT) even when ground truth is unavailable, confirming the efficacy of group similarity rewards for open-ended or reference-free tasks (Hamman et al., 5 Oct 2025).
6. Relationship to Broader GRPO Paradigm and Extensions
Con-GRPO is a specialization of GRPO for information consistency, leveraging group reward computation and normalization within paraphrase sets. Related GRPO variants extend the paradigm to other objectives:
- Constrained GRPO imposes explicit behavioral constraints via Lagrangian relaxation and scalarized advantage construction, outperforming naive scalarization approaches in both theory and practice (Girgis et al., 5 Feb 2026).
- Consensus GRPO distills Minimum Bayes Risk decoding into a policy optimized using only groupwise consensus (e.g., BLEURT) as utility, eliminating reliance on gold references or preference labels (Ichihara et al., 3 Feb 2026).
- Continuous Control Con-GRPO applies group-based normalization to trajectory clusters and state-aware credit assignment for high-dimensional, continuous-action tasks (Khanda et al., 25 Jul 2025).
These extensions maintain the core principle of intra-group normalized advantage estimation, with variations in reward construction and applicability.
7. Limitations and Future Directions
While Con-GRPO has demonstrated clear empirical gains in consistency and accuracy across QA and RAG tasks, limitations and open research directions include:
- Reward functions are dependent on lexical similarity (e.g., BLEU), which may not fully capture semantic agreement.
- Approximate reward computation, while efficient, introduces stochastic variance not present in exhaustive all-pairs evaluation.
- Extension to more diverse and challenging paraphrase sets, dialogue, non-English queries, and non-retrieval settings remains to be explored.
- Integration of richer semantic similarity metrics or human-aligned utility functions to minimize the risk of overfitting to lexical form rather than true informational content.
The framework provides a reproducible and theoretically grounded approach for aligning large-scale neural generation systems with consistency-centric deployment constraints (Hamman et al., 5 Oct 2025).