Learning to Drive Anywhere

  • 2023-09-21 18:55:36
  • Ruizhao Zhu, Peng Huang, Eshed Ohn-Bar, Venkatesh Saligrama
  • 0

Abstract

Human drivers can seamlessly adapt their driving decisions acrossgeographical locations with diverse conditions and rules of the road, e.g.,left vs. right-hand traffic. In contrast, existing models for autonomousdriving have been thus far only deployed within restricted operational domains,i.e., without accounting for varying driving behaviors across locations ormodel scalability. In this work, we propose AnyD, a single geographically-awareconditional imitation learning (CIL) model that can efficiently learn fromheterogeneous and globally distributed data with dynamic environmental,traffic, and social characteristics. Our key insight is to introduce ahigh-capacity geo-location-based channel attention mechanism that effectivelyadapts to local nuances while also flexibly modeling similarities among regionsin a data-driven manner. By optimizing a contrastive imitation objective, ourproposed approach can efficiently scale across inherently imbalanced datadistributions and location-dependent events. We demonstrate the benefits of ourAnyD agent across multiple datasets, cities, and scalable deployment paradigms,i.e., centralized, semi-supervised, and distributed agent training.Specifically, AnyD outperforms CIL baselines by over 14% in open-loopevaluation and 30% in closed-loop testing on CARLA.

 

Quick Read (beta)

loading the full paper ...