ARMOR-PRM: Adversarially Robust Process Reward Models via Counterfactual Margins and Invariance Regularization

Hongyu Chen, Guangjia Chai, Haoxian Nan, Ruixue Wang, Peilin Jiang


Abstract
Process Reward Models (PRMs) are vulnerable to reward hacking: policies exploit surface cues such as confident phrasing or structured formatting in place of logical correctness, which yields an inverted-U collapse under optimization pressure. We propose ARMOR-PRM, which augments standard PRM training with three regularizers: a counterfactual margin loss, an adversarial invariance loss with low-score anchoring, and a gradient penalty on the reward head. On ProcessBench (three splits; "math" excluded due to training data overlap), ARMOR-PRM improves Detection Rate by 14.9 percentage points and AUROC by 4.1 percentage points over Vanilla PRM, and is the only variant whose Best-of-N reranking accuracy improves monotonically through N=128. A domain-shift diagnostic further shows that objective-level robustness and data coverage are distinct but complementary axes: ARMOR-PRM responds best to iterative hard-negative retraining, suggesting a practical two-stage deployment path.
Anthology ID:
2026.findings-emnlp.1287
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
27250–27264
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.1287/
DOI:
Bibkey:
Cite (ACL):
Hongyu Chen, Guangjia Chai, Haoxian Nan, Ruixue Wang, and Peilin Jiang. 2026. ARMOR-PRM: Adversarially Robust Process Reward Models via Counterfactual Margins and Invariance Regularization. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 27250–27264, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
ARMOR-PRM: Adversarially Robust Process Reward Models via Counterfactual Margins and Invariance Regularization (Chen et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.1287.pdf