CORE: Collaborative Reasoning via Cross Teaching

작성자

카테고리:

← 피드로
arXiv cs.AI · Kshitij Mishra, Mirat Aubakirov, Martin Takac, Nils Lukas, Salem Lahlou · 2026-08-04 AI

[Submitted on 29 Jan 2026 (v1), last revised 3 Aug 2026 (this version, v2)]

View PDF HTML (experimental)

Abstract:Large language models exhibit complementary reasoning errors: on the same instance, one model may succeed with a particular decomposition while another fails. We propose Collaborative Reasoning (CORE), a training-time collaboration framework that converts peer success into a learning signal via a cross-teaching protocol. Each problem is solved in two stages: a cold round of independent sampling, followed by a contexted rescue round in which models that failed receive hint extracted from a successful peer. CORE optimizes a combined reward that balances (i) correctness, (ii) a lightweight DPP-inspired diversity term to reduce error overlap, and (iii) an explicit rescue bonus for successful recovery. We evaluate CORE across four standard reasoning datasets GSM8K, MATH, AIME, and GPQA. With only 1,000 training examples, a pair of small open source models (3B+4B) reaches Pass@2 of 99.54% on GSM8K and 92.08% on MATH, compared to 82.50% and 74.82% for single-model training. On harder datasets, the 3B+4B pair reaches Pass@2 of 77.34% on GPQA (trained on 348 examples) and 79.65% on AIME (trained on 792 examples), using a training-time budget of at most 1536 context tokens and 3072 generated tokens. Overall, these results show that training-time collaboration can reliably convert model complementarity into large gains without scaling model size.

Submission history

From: Kshitij Mishra [view email]
[v1] Thu, 29 Jan 2026 12:07:54 UTC (4,481 KB)
[v2] Mon, 3 Aug 2026 07:18:11 UTC (4,455 KB)

원문에서 계속 ↗

추출 본문 · 출처: arxiv.org · https://arxiv.org/abs/2601.21600

코멘트

답글 남기기

이메일 주소는 공개되지 않습니다. 필수 필드는 *로 표시됩니다