自動運転車が安全かつ効果的に運転するには、歩行者や他の車など、シーン内の他のエージェントの確率的な将来の軌道を予測する必要があります。このような複雑なマルチモーダル分布を予測するには、強力な確率論的アプローチが必要です。フローの正規化は、このような分布をモデル化するための魅力的なツールとして最近登場しました。ただし、フローモデルから軌道予測を生成する場合の主な欠点は、独立したサンプルが基礎となる分布のすべてのモードを適切にキャプチャできないことが多いことです。事前にトレーニングされたフローモデルからの軌道サンプルの品質と多様性を改善する方法である、フローの多様性サンプリング(DSF)を提案します。 DSFは、個々のサンプルを生成するのではなく、一連の軌跡を1回のショットで生成します。事前にトレーニングされた予測フローモデルを前提として、モデルの勾配を使用してDSFをトレーニングし、軌道間の高い空間分離とともに、予測セット内の個々の軌道の可能性が高い目的関数を最適化します。 DSFは実装が簡単で、いくつかの既存のフローベースの予測モデルに単純なプラグインの改善を提供し、2つの挑戦的な車両と歩行者の予測ベンチマークで最先端の結果を達成することを示します。
For autonomous cars to drive safely and effectively, they must anticipate the stochastic future trajectories of other agents in the scene, such as pedestrians and other cars. Forecasting such complex multi-modal distributions requires powerful probabilistic approaches. Normalizing flows have recently emerged as an attractive tool to model such distributions. However, when generating trajectory predictions from a flow model, a key drawback is that independent samples often do not adequately capture all the modes in the underlying distribution. We propose Diversity Sampling for Flow (DSF), a method for improving the quality and the diversity of trajectory samples from a pre-trained flow model. Rather than producing individual samples, DSF produces a set of trajectories in one shot. Given a pre-trained forecasting flow model, we train DSF using gradients from the model, to optimize an objective function that rewards high likelihood for individual trajectories in the predicted set, together with high spatial separation between trajectories. DSF is easy to implement, and we show that it offers a simple plug-in improvement for several existing flow-based forecasting models, achieving state-of-art results on two challenging vehicle and pedestrian forecasting benchmarks.