Coopshift: Post-Training Amplifies The Memory Curse In LLM Cooperation
Organizations: Carnegie Mellon University · Foundations of Cooperative AI Lab (FOCAL) · The University of Texas at Austin
Abstract
How does post-training change cooperation during repeated interaction? Within each of seven released model lineages, we compare pretrained (Base), supervised fine-tuned (SFT), and subsequent preference- or reward-optimized (Aligned) checkpoints across seven social dilemmas. Copies of each checkpoint play one another for 200 rounds, with visible history limited to the last two or forty rounds. The memory curse refers to lower cooperation when agents can see more past rounds. Our central finding is that post-training amplifies this decline. Averaged across the six binary-action games, expanding visible history reduces cooperation at every SFT and Aligned checkpoint. At each post-trained stage, these declines exceed Base's in six of seven lineages. Most of this average amplification is already present at SFT. With long history, both post-trained stages also cooperate less than Base on average in each binary-action game. In the symmetric binary games, their fitted cooperation rates fall more steeply on average than Base's as per-round cooperation costs rise, benefits fall, or noncooperation becomes more rewarding. The difference in cooperation rates between Aligned and SFT also changes from one-shot decisions to repeated play. When given the same past records, their choices depend on whether the instructions describe a final round or possible future interaction. Post-training evaluation should therefore go beyond isolated choices to assess whether cooperation is sustained in repeated interaction and how it depends on payoffs and visible history.
Figures & tables
| Game | Decision and tension | Meaning of the raw main parameter |
|---|---|---|
| PGG | Contribute a unit to a shared pool or retain it. | The marginal per-capita return (MPCR), , is each player’s return per contributed unit, where is the pool multiplier and the group size. Higher lowers private cost . |
| CPR | Restrain extraction or extract more; each unit earns less as total extraction rises. | Low extraction is 1 and high extraction is . Higher increases high extraction’s private advantage at fixed ; payoffs depend on current-round extraction. |
| Donation-PD | Pay to give to each other player, or give nothing. | is the donor’s cost and the benefit to each of two others. Higher , at fixed , raises external returns; private cost stays . |
| Volunteer | Supply a shared benefit or rely on another volunteer; one suffices. | Each player receives if anyone volunteers; every volunteer pays , so extra volunteers duplicate costs. Higher , at fixed , increases volunteering cost. |
| Assurance | Join a threshold project or take a sure outside payoff. | A successful contributor earns a fraction more than the sure outside payoff. Success requires at least contributors, including itself; unsuccessful contributors receive zero. Higher raises the reward for successful coordination. |
| TD | Claim an amount; the lower claimant gets a bonus, the higher a penalty. | is the number of legal claim levels. Higher , at fixed , raises bonus/penalty and one-step undercutting gain. Outcome: claim normalized to . |
| Game / stage | Drop | ||||
|---|---|---|---|---|---|
| PGG : MPCR | 0.8 | 0.6 | 0.5 | 0.4 | – |
| Base (B) | |||||
| SFT (S) | |||||
| Aligned (A) | |||||
| CPR : | 0.2 | 0.4 | 0.6 | 0.8 | – |
| Base (B) |
| History change or paired difference | Mean SD | 95% interval | Negative |
|---|---|---|---|
| Base: HL40 minus HL2 | 2/7 | ||
| SFT: HL40 minus HL2 | 7/7 | ||
| Aligned: HL40 minus HL2 | 7/7 | ||
| SFT minus Base history change | 6/7 | ||
| Aligned minus Base history change | 6/7 | ||
| Aligned minus SFT history change | 6/7 |
| Game | One-shot | Repeated (HL40) | Change | Lineages with |
|---|---|---|---|---|
| Aligned SFT | Aligned SFT | Repeated one-shot | a decrease | |
| PGG | 5/7 | |||
| CPR | 5/7 | |||
| Donation-PD | 4/7 | |||
| Volunteer | 7/7 | |||
| Assurance | 3/7 |
| Game | Final | Continue | Mean | Base | SFT | Aligned |
|---|---|---|---|---|---|---|
| PGG | ||||||
| CPR | ||||||
| Donation-PD | ||||||
| Volunteer | ||||||
| Assurance | ||||||
| TD |
| Future instruction | Stage | Real | Masked | Real Masked |
|---|---|---|---|---|
| Continue | SFT (S) | |||
| Aligned (A) | ||||
| Final | SFT (S) | |||
| Aligned (A) |
Appendix figures & tables16 assets
Supplementary material from the paper’s appendix.
Appendix
| Lineage | Stage | Hugging Face repository |
|---|---|---|
| OLMo-3-7B | Base | allenai/Olmo-3-1025-7B |
| OLMo-3-7B | SFT | allenai/Olmo-3-7B-Instruct-SFT |
| OLMo-3-7B | Aligned (DPO) | allenai/Olmo-3-7B-Instruct-DPO |
| SmolLM2-1.7B | Base | HuggingFaceTB/SmolLM2-1.7B |
| SmolLM2-1.7B | SFT | HuggingFaceTB/SmolLM2-1.7B-sft-only |
| SmolLM2-1.7B | Aligned (DPO) | HuggingFaceTB/SmolLM2-1.7B-Instruct |
| Lineage | ||||||
|---|---|---|---|---|---|---|
| OLMo | ||||||
| SmolLM2 | ||||||
| DeepSeekMath | ||||||
| InternLM2 | ||||||
| Nemotron | ||||||
| Tulu |
| Game | Total | Between lineage | Within lineage/config |
|---|---|---|---|
| PGG | 1586.55 | 1439.81 | 146.74 |
| CPR | 829.71 | 745.27 | 84.44 |
| Game / stage | Fitted drop | ||||
|---|---|---|---|---|---|
| PGG : MPCR | 0.8 | 0.6 | 0.5 | 0.4 | – |
| Base (B) | |||||
| SFT (S) | |||||
| Aligned (A) | |||||
| SFT Base | |||||
| Aligned Base |
| Lineage / stage | Fitted drop | ||||
|---|---|---|---|---|---|
| PGG : MPCR | 0.8 | 0.6 | 0.5 | 0.4 | – |
| OLMo-3-7B | |||||
| Base (B) | |||||
| SFT (S) | |||||
| Aligned (A) | |||||
| SFT Base | |||||
| Lineage / stage | Fitted drop | ||||
|---|---|---|---|---|---|
| CPR : | 0.2 | 0.4 | 0.6 | 0.8 | – |
| OLMo-3-7B | |||||
| Base (B) | |||||
| SFT (S) | |||||
| Aligned (A) | |||||
| SFT Base | |||||
| Lineage / stage | Fitted drop | ||||
|---|---|---|---|---|---|
| Donation-PD : | 2.4 | 2 | 1.4 | 1.1 | – |
| OLMo-3-7B | |||||
| Base (B) | |||||
| SFT (S) | |||||
| Aligned (A) | |||||
| SFT Base | |||||
| Lineage / stage | Fitted drop | ||||
|---|---|---|---|---|---|
| Volunteer : | 0.125 | 0.375 | 0.625 | 0.875 | – |
| OLMo-3-7B | |||||
| Base (B) | |||||
| SFT (S) | |||||
| Aligned (A) | |||||
| SFT Base | |||||
| Lineage / stage | Fitted drop | ||||
|---|---|---|---|---|---|
| Assurance : | 0.4 | 0.2 | 0.1 | 0.05 | – |
| OLMo-3-7B | |||||
| Base (B) | |||||
| SFT (S) | |||||
| Aligned (A) | |||||
| SFT Base | |||||
| Lineage / stage | Fitted drop | ||||
|---|---|---|---|---|---|
| TD : | 0.025 | 0.05 | 0.1 | 0.2 | – |
| OLMo-3-7B | |||||
| Base (B) | |||||
| SFT (S) | |||||
| Aligned (A) | |||||
| SFT Base | |||||
| Checkpoint | Training completed at this step |
|---|---|
| Base | Qwen3-8B pretrained model, before Nemotron-Cascade post-training. |
| SFT | Multi-stage supervised fine-tuning on example responses spanning general dialogue, reasoning, coding, and other domains. |
| RLHF | Reinforcement learning using a learned human-preference reward. This is our main panel’s Aligned checkpoint. |
| IFRL | Instruction-following reinforcement learning, rewarding adherence to explicit response constraints; applied in non-thinking mode for this unified model. |
| MathRL | Reinforcement learning on mathematical problems, using answer-correctness rewards. |
| CodeRL | Reinforcement learning on programming problems, using code-execution tests as rewards. |
| Stage | OS | HL2 | HL40 | ||
|---|---|---|---|---|---|
| Base | |||||
| SFT | |||||
| RLHF | |||||
| IFRL | |||||
| MathRL | |||||
| CodeRL |
| Stage | OS | HL2 | HL40 |
|---|---|---|---|
| Base | |||
| SFT | |||
| RLHF | |||
| IFRL | |||
| MathRL | |||
| CodeRL |
| Stage | OS | HL2 | HL40 | ||
|---|---|---|---|---|---|
| Base | |||||
| SFT | |||||
| RLHF | |||||
| IFRL | |||||
| MathRL | |||||
| CodeRL |
LLM Post-Training
Large language model post-training adapts pretrained models into more useful, reliable, and specialized assistants through supervised and preference-based learning.
See all 246 papers