Massively Scalable Inverse Reinforcement Learning in Google Maps

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Barnes, Matt, Abueg, Matthew, Lange, Oliver F., Deeds, Matt, Trader, Jason, Molitor, Denali, Wulfmeier, Markus, O'Banion, Shawn
Natura: Preprint
Pubblicazione: 2023
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866913255440515072
author Barnes, Matt
Abueg, Matthew
Lange, Oliver F.
Deeds, Matt
Trader, Jason
Molitor, Denali
Wulfmeier, Markus
O'Banion, Shawn
author_facet Barnes, Matt
Abueg, Matthew
Lange, Oliver F.
Deeds, Matt
Trader, Jason
Molitor, Denali
Wulfmeier, Markus
O'Banion, Shawn
contents Inverse reinforcement learning (IRL) offers a powerful and general framework for learning humans' latent preferences in route recommendation, yet no approach has successfully addressed planetary-scale problems with hundreds of millions of states and demonstration trajectories. In this paper, we introduce scaling techniques based on graph compression, spatial parallelization, and improved initialization conditions inspired by a connection to eigenvector algorithms. We revisit classic IRL methods in the routing context, and make the key observation that there exists a trade-off between the use of cheap, deterministic planners and expensive yet robust stochastic policies. This insight is leveraged in Receding Horizon Inverse Planning (RHIP), a new generalization of classic IRL algorithms that provides fine-grained control over performance trade-offs via its planning horizon. Our contributions culminate in a policy that achieves a 16-24% improvement in route quality at a global scale, and to the best of our knowledge, represents the largest published study of IRL algorithms in a real-world setting to date. We conclude by conducting an ablation study of key components, presenting negative results from alternative eigenvalue solvers, and identifying opportunities to further improve scalability via IRL-specific batching strategies.
format Preprint
id arxiv_https___arxiv_org_abs_2305_11290
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Massively Scalable Inverse Reinforcement Learning in Google Maps
Barnes, Matt
Abueg, Matthew
Lange, Oliver F.
Deeds, Matt
Trader, Jason
Molitor, Denali
Wulfmeier, Markus
O'Banion, Shawn
Machine Learning
Inverse reinforcement learning (IRL) offers a powerful and general framework for learning humans' latent preferences in route recommendation, yet no approach has successfully addressed planetary-scale problems with hundreds of millions of states and demonstration trajectories. In this paper, we introduce scaling techniques based on graph compression, spatial parallelization, and improved initialization conditions inspired by a connection to eigenvector algorithms. We revisit classic IRL methods in the routing context, and make the key observation that there exists a trade-off between the use of cheap, deterministic planners and expensive yet robust stochastic policies. This insight is leveraged in Receding Horizon Inverse Planning (RHIP), a new generalization of classic IRL algorithms that provides fine-grained control over performance trade-offs via its planning horizon. Our contributions culminate in a policy that achieves a 16-24% improvement in route quality at a global scale, and to the best of our knowledge, represents the largest published study of IRL algorithms in a real-world setting to date. We conclude by conducting an ablation study of key components, presenting negative results from alternative eigenvalue solvers, and identifying opportunities to further improve scalability via IRL-specific batching strategies.
title Massively Scalable Inverse Reinforcement Learning in Google Maps
topic Machine Learning
url https://arxiv.org/abs/2305.11290