JAX加速多智能体宏观模型训练至秒级

01 摘要

JaxMARL-BC 是一个 JAX 加速的模拟器，用于训练多智能体强化学习宏观经济模型。作者报告其训练速度比参考实现快 75 至 724 倍，可扩展到 200,000 个独立学习家庭，并在两小时内完成原本需一个多月的校准实验。论文验证了模型在已知 DSGE 极限下的正确性，并提出了简单的成本缩放规律。

02 关键点

1. 训练速度较参考实现提升75至724倍，可扩展到20万独立学习家庭。
2. 模型在特定极限下退化为经典DSGE模型，并以此验证正确性。
3. 通过异质性资本回报扩展模型的校准概念验证，将耗时从超过一个月缩短至两小时。