← Search

Milad Aghajohari

5 accepted papers

2026

The Markovian Thinker

ICLR 2026poster

Reasoning LLMs suffer from quadratic compute growth as their context length increases, making reinforcement learning with verifiable rewards (RLVR) and test-time scaling prohibitively expensive. Prior work has tried to lighten the computational burden by shortening reasoning traces through pruning,…

Cited by 0SourceScholar
2025

Advantage Alignment Algorithms

ICLR 2025oral

Artificially intelligent agents are increasingly being integrated into human decision-making: from large language model (LLM) assistants to autonomous vehicles. These systems often optimize their individual objective, leading to conflicts, particularly in general-sum games where naive reinforcement…

Cited by 0SourcePDFScholar
2025

VinePPO: Refining Credit Assignment in RL Training of LLMs

ICML 2025poster

Large language models (LLMs) are increasingly applied to complex reasoning tasks that require executing several complex steps before receiving any reward. Properly assigning credit to these steps is essential for enhancing model performance. Proximal Policy Optimization (PPO), a common reinforcement…

Cited by 0SourcePDFScholar
2024

LOQA: Learning with Opponent Q-Learning Awareness

ICLR 2024poster

In various real-world scenarios, interactions among agents often resemble the dynamics of general-sum games, where each agent strives to optimize its own utility. Despite the ubiquitous relevance of such settings, decentralized machine learning algorithms have struggled to find equilibria that maxim…

Cited by 2SourcePDFScholar