Enhancing LLM Metacognition via Cognitive Pairwise Training

Weitao Li, Hao Zhou, Xuanyu Lei, Fandong Meng, Yuanhang Liu +9 more
5/30/2026
cs.LG

Official code: Tsinghua-dhy/CPT (7 stars, Apache-2.0, last commit 2026-06-02).

Abstract

Reinforcement learning with verifiable rewards (RLVR) has become central to LLM reasoning, but its outcome-level rewards can make models more willing to give confident answers when evidence or reasoning is unreliable. Existing SFT or RL methods mainly teach LLMs to refuse or express uncertainty at the response level, which can overfit abstention behavior rather than improve reasoning reliability. To address this limitation, we propose Cognitive Pairwise Training (CPT), a cognitive mid-training alignment stage that turns pairwise comparisons over reasoning traces into a reusable alignment signal. By learning to distinguish trustworthy from flawed reasoning, CPT encourages the model to internalize a reasoning-quality discrimination boundary rather than memorize surface refusal patterns. Across five model scales and three model families, CPT improves the reasoning--metacognition trade-off. At 14B, CPT+RL outperforms the standard SFT+RL pipeline by +2.2 math-average points and +5.2 abstention-F1 points. Further analyses show that CPT improves trace quality and exhibits strong robustness and scalability across evaluation and training settings. Code and models are released at https://github.com/Tsinghua-dhy/CPT.

View on arXivView PDF

Code Implementations(1)

Tsinghua-dhy/CPTOfficial100%
Apache-2.0

CPT: Enhancing LLM Metacognition via Cognitive Pairwise Training

70Shell, PythonMay 19, 20264 months agoApache-2.0

Similar Papers

Cite this paper

@article{li2026enhancing,
  title  = {Enhancing LLM Metacognition via Cognitive Pairwise Training},
  author = {Weitao Li and Hao Zhou and Xuanyu Lei and Fandong Meng and Yuanhang Liu and Jingyi Ren and Ante Wang and Xiaolong Wang and Yuanchi Zhang and Fuwen Luo and Guangwen Yang and Lin Gan and Weizhi Ma and Yang Liu},
  year   = {2026},
  eprint = {2606.00869},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2606.00869v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2606.00869.svg)](https://codeofpaper.com/paper/2606.00869)

Discussion