Structural Reinforcement Learning for Heterogeneous Agent Macroeconomics

Yucheng Yang, Chiyuan Wang, Andreas Schaab, Benjamin Moll
12/21/2025
econ.THcs.AIcs.LG

No confident code implementation found for this paper.

Abstract

We present a new approach to formulating and solving heterogeneous agent models with aggregate risk. We replace the cross-sectional distribution with low-dimensional prices as state variables and let agents learn equilibrium price dynamics directly from simulated paths. To do so, we introduce a structural reinforcement learning (SRL) method which treats prices via simulation while exploiting agents' structural knowledge of their own individual dynamics. Our SRL method yields a general and highly efficient global solution method for heterogeneous agent models that sidesteps the Master equation and handles problems traditional methods struggle with, in particular nontrivial market-clearing conditions. We illustrate the approach in the Krusell-Smith model, the Huggett model with aggregate shocks, and a HANK model with a forward-looking Phillips curve, all of which we solve globally within minutes.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{yang2025structural,
  title  = {Structural Reinforcement Learning for Heterogeneous Agent Macroeconomics},
  author = {Yucheng Yang and Chiyuan Wang and Andreas Schaab and Benjamin Moll},
  year   = {2025},
  eprint = {2512.18892},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2512.18892v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2512.18892.svg)](https://codeofpaper.com/paper/2512.18892)

Discussion