acceptodds
Under review as a conference paper at ICLR 2027

UMARM: Unified Multi-Agent Reward Matching

Abstract

Behavior cloning, contrastive representation learning, and preference optimization are independently-motivated approaches behavioral modeling. We show that under a player-conditioned reward , these objectives share common structure: the behavior cloning optimum recovers the InfoNCE-optimal critic up to action-independent terms (Theorem 1), a principled adaptive schedule with super-linear decay guarantees convergence to the behavioral and preference objectives alone (Theorem 2), and the combined loss is equivalent to a MaxEnt goal-conditioned RL problem whose optimal reward recovers the player-conditioned data distribution (Theorem 3). UMARM (Unified Multi-Agent Reward Matching) instantiates this framework for multi-agent behavioral modeling, jointly learning a continuous player latent and a conditioned policy from raw trajectories alone — no skill ratings, agent identifiers, or per-agent parameters are required. Applied to chess across million players spanning novice to grandmaster, UMARM achieves top-1 move prediction accuracy, outperforming the parameter-matched Maia-3 model. Constructing zero-shot from moves in the current game, accuracy reaches overall and after 40 moves of context. ELO ratings are recoverable from the latent geometry () despite never being supplied, and the preference head enables match outcome prediction between unrelated opponents. All capabilities emerge from the unified objective without task-specific supervision.

open until 14 Dec 2026

est. 32% chance this paper gets accepted at ICLR 2027.

Reject 68%Accept 32%

What do you think this paper will get?

All positions stay anonymous.

Related papers

Loading the map…

Discussion (0)

Sign in to comment.