Evaluating the Scalability and Adversarial Generalization of GRPO-Trained NLI Models
On the map: LLM reasoning → Reinforcement learning for reasoning
Abstract (arXiv)
Natural Language Inference (NLI) is a central task in natural language understanding with applications in fact-checking, question answering, and information retrieval. Despite its importance, current NLI systems heavily rely on supervised learning with datasets that often contain annotation artifacts and biases, limiting generalization and real-world applicability. In this work, we apply a reinforcement learning-based approach using Group Relative Policy Optimization (GRPO) for Chain-of-Thought (CoT) learning in NLI, eliminating the need for human-labeled rationales and enabling this type of training on challenging datasets such as ANLI. We fine-tune 7B, 14B, and 32B language models using parameter-efficient techniques (LoRA and QLoRA), demonstrating strong performance across standard and adversarial NLI benchmarks. At the 32B scale, GRPO-trained models generalize better than other supervised baselines in adversarial sets. With AWQ quantization, the 32B model fits within 22GB of CUDA memory. This work provides a scalable and practical framework for building robust NLI systems without sacrificing inference quality.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Placed in Reinforcement learning for reasoning, one of 17 directions in LLM reasoning. Builds on 33 of its 38 references that are in the map. First posted on arXiv 25 Apr 2025.
Builds on
33 of 38 references.
Closest ancestors · 1
Same direction, newest first
Other references · 32
- 2017Attention Is All You Needcited by 60,361
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,372
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,899
- 2023QLoRA: Efficient Finetuning of Quantized LLMscited by 3,418
- 2017A Broad-Coverage Challenge Corpus for Sentence Understanding through Inferencecited by 2,682
- 2020DeBERTa: Decoding-enhanced BERT with Disentangled Attentioncited by 2,181
- 2015A large annotated corpus for learning natural language inferencecited by 1,847
20 more
- 2021Multitask Prompted Training Enables Zero-Shot Task Generalizationcited by 1,470
- 2023AWQ: Activation-aware Weight Quantization for LLM Compression and Accelerationcited by 1,298
- 2019Right for the Wrong Reasons: Diagnosing Syntactic Heuristics in Natural Language Inferencecited by 876
- 2019Adversarial NLI: A New Benchmark for Natural Language Understandingcited by 807
- 2018Annotation Artifacts in Natural Language Inference Datacited by 671
- 2023MAmmoTH: Building Math Generalist Models through Hybrid Instruction Tuningcited by 487
- 2018e-SNLI: Natural Language Inference with Natural Language Explanationscited by 468
- 2019Learning the Difference that Makes a Difference with Counterfactually-Augmented Datacited by 439
- 2018Hypothesis Only Baselines in Natural Language Inferencecited by 295
- 2022WANLI: Worker and AI Collaboration for Natural Language Inference Dataset Creationcited by 202
- 2020Adversarial Filters of Dataset Biasescited by 160
- 2020WT5?! Training Text-to-Text Models to Explain their Predictionscited by 148
- 2021Entailment as Few-Shot Learnercited by 123
- 2023MathPile: A Billion-Token-Scale Pretraining Corpus for Mathcited by 67
- 2018Performance Impact Caused by Hidden Bias of Training Data for Recognizing Textual Entailmentcited by 59
- 2022Generating Data to Mitigate Spurious Correlations in Natural Language Inference Datasetscited by 54
- 2021Supervising Model Attention with Human Explanations for Robust Natural Language Inferencecited by 29
- 2024Inference-Time Decontamination: Reusing Leaked Benchmarks for Large Language Model Evaluationcited by 19
- 2020HypoNLI: Exploring the Artificial Patterns of Hypothesis-only Bias in Natural Language Inferencecited by 17
- 2020An Empirical Study on Model-agnostic Debiasing Strategies for Robust Natural Language Inferencecited by 16
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2026GRPO Training Dynamics for Small Language Models
- 2025Multi-Layer GRPO: Enhancing Reasoning and Self-Correction in Large Language Models
- 2026Demystifying Group Relative Policy Optimization: Its Policy Gradient is a U-Statistic
- 2025Can GRPO Help LLMs Transcend Their Pretraining Origin?
- 2026Transformation-Augmented GRPO for Enhancing Exploration in Reasoning of Large Language Models
- 2025SSPO: Subsentence-level Policy Optimization
- 2026GUPO: Gradient Uncertainty-aware Policy Optimization for Post-Training Large Language Models
- 2025Geometric-Mean Policy Optimization