FlowDrive: moderated flow matching with data balancing for trajectory planning

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Wang, Lingguang, Taş, Ömer Şahin, Steiner, Marlon, Stiller, Christoph
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866917277320871936
author Wang, Lingguang
Taş, Ömer Şahin
Steiner, Marlon
Stiller, Christoph
author_facet Wang, Lingguang
Taş, Ömer Şahin
Steiner, Marlon
Stiller, Christoph
contents Learning-based planners are sensitive to the long-tailed distribution of driving data. Common maneuvers dominate datasets, while dangerous or rare scenarios are sparse. This imbalance can bias models toward the frequent cases and degrade performance on critical scenarios. To tackle this problem, we compare balancing strategies for sampling training data and find reweighting by trajectory pattern an effective approach. We then present FlowDrive, a flow-matching trajectory planner that learns a conditional rectified flow to map noise directly to trajectory distributions with few flow-matching steps. We further introduce moderated, in-the-loop guidance that injects small perturbation between flow steps to systematically increase trajectory diversity while remaining scene-consistent. On nuPlan and the interaction-focused interPlan benchmarks, FlowDrive achieves state-of-the-art results among learning-based planners and approaches methods with rule-based refinements. After adding moderated guidance and light post-processing (FlowDrive*), it achieves overall state-of-the-art performance across nearly all benchmark splits. Our code is available at https://github.com/einsteinguang/flow_drive_planner.
format Preprint
id arxiv_https___arxiv_org_abs_2509_21961
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle FlowDrive: moderated flow matching with data balancing for trajectory planning
Wang, Lingguang
Taş, Ömer Şahin
Steiner, Marlon
Stiller, Christoph
Robotics
Artificial Intelligence
Machine Learning
Learning-based planners are sensitive to the long-tailed distribution of driving data. Common maneuvers dominate datasets, while dangerous or rare scenarios are sparse. This imbalance can bias models toward the frequent cases and degrade performance on critical scenarios. To tackle this problem, we compare balancing strategies for sampling training data and find reweighting by trajectory pattern an effective approach. We then present FlowDrive, a flow-matching trajectory planner that learns a conditional rectified flow to map noise directly to trajectory distributions with few flow-matching steps. We further introduce moderated, in-the-loop guidance that injects small perturbation between flow steps to systematically increase trajectory diversity while remaining scene-consistent. On nuPlan and the interaction-focused interPlan benchmarks, FlowDrive achieves state-of-the-art results among learning-based planners and approaches methods with rule-based refinements. After adding moderated guidance and light post-processing (FlowDrive*), it achieves overall state-of-the-art performance across nearly all benchmark splits. Our code is available at https://github.com/einsteinguang/flow_drive_planner.
title FlowDrive: moderated flow matching with data balancing for trajectory planning
topic Robotics
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2509.21961