Tropical Reinforcement Learning
cs.AI
Submitted: 2026-10-01
Updated: 2026-10-01
Code: https://github.com/cameronfreer/lean4-skills
License: http://creativecommons.org/licenses/by/4.0/
The gist: Reinforcement learning for large language models typically maximizes expected return, adding up the probabilities of all successful trajectories.
Terminology
Abstract
Reinforcement learning for large language models typically maximizes expected return, adding up the probabilities of all successful trajectories. However, the classical sum formulation can only report how often the model policy succeeds, not which solution actually worked, and because probabilities sum to one, reinforcing one solution can make the model forget another that was never shown to be wrong. This makes expected return a poor fit for compositional reasoning, where a solution must be assembled from reasoning steps that the model produces in separate, often failed, attempts but rarely produces together. To address this, we propose Tropical Reinforcement Learning, which rests on a simple change of algebra: instead of adding the probabilities of alternative solutions, we take their maximum, which yields the tropical semiring. The value of a state then becomes the log-probability of its most likely verified solution, together with an explicit path that can be replayed and reused. This enables true composition, since the best prefix and the best suffix meeting at a shared state can be joined even when they come from different rollouts. To put this into practice, we introduce TROPIC, a training algorithm for deterministic, resettable environments with verifiable outcomes. On four agentic tasks (Sokoban, Countdown, FrozenLake, WebShop), TROPIC outperforms the strongest on-policy baselines by up to 16 percentage points. Changing the algebra of reinforcement learning, not just its estimators, can thus substantially improve compositional reasoning in language models
Sources
- RL Post-Training Builds Compositional Reasoning Strategies
- Max-Plus Matching Pursuit for Deterministic Markov Decision Processes
- OpenAI Gym
- Large Language Monkeys: Scaling Inference Compute with Repeated Sampling
- The Entropy Mechanism of Reinforcement Learning for Reasoning Language Models
- DeepSeek-V4: Towards Highly Efficient Million-Token Context Intelligence
- Group-in-Group Policy Optimization for LLM Agent Training
- Reinforcement Learning for Compositional Generalization with Outcome-Level Optimization
- GLM-5: from Vibe Coding to Agentic Engineering
- VinePPO: Refining Credit Assignment in RL Training of LLMs
- Kimi k1.5: Scaling Reinforcement Learning with LLMs
- From Reasoning Traces to Reusable Modules: Understanding Compositional Generalization in Language Model Reasoning
- Reason in Chains, Learn in Trees: Self-Rectification and Grafting for Multi-turn Agent Policy Optimization
- Qwen2.5 Technical Report
- Credit Assignment with Resets in Language Model Reasoning
- Proximal Policy Optimization Algorithms
- DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models
- Maximum Likelihood Reinforcement Learning
- Optimizing Language Models for Inference Time Objectives using Reinforcement Learning
- Solving math word problems with process- and outcome-based feedback
Related papers
- MAVEN-T: Reinforced Heterogeneous Distillation for Real-Time Multi-Agent Trajectory Prediction
- Model Discovery Agent: LLM-assisted Bayesian experiment design for data-efficient discovery of mechanistic world models
- Agents' Last Exam
- The Clinician's Veto: Navigating Trust, Liability, and Uncertainty in Autonomous AI Prescribing
- MindHelper: Closed-Loop Embodied Mental-State Reasoning for Precision Intervention
- Incumbent Advantage: Brand Bias and Cognitive Manipulation Dynamics in LLM Recommendation Systems