SYNTOLOGY HomeExplorerAtlasCodeMethodologyAboutDevelopersFeedPricing
Paper · 2402.12238 · NeurIPS · 2024

MGF: Mixed Gaussian Flow for Diverse Trajectory Prediction

Kris Kitani, Jiangmiao Pang, Dahua Lin, Jinkun Cao, Jiahe Chen

arXiv · PDF · Open in the Atlas

Code that ran

We lifted 30 functions out of this paper's own repositories and ran 21 of them in a sandbox. "Ran" means the function executed on a synthesized input and returned a value. It is not a reproduction of the paper's results.

RepositoryRoleRan
mulplue/MGF canonical 10 of 12
mulplue/mgf — 11 of 18
FunctionStatusWhere it lives
BatchNorm Ran mulplue/mgf/src/models/TP/fastpredNF.py
code served (permissive licence) · get_code("5b716a5982a339d2")
Build_TP_model Ran mulplue/MGF/src/models/TP/TP_models.py
code served (permissive licence) · get_code("2826248c43f41a45")
CIF_step Ran mulplue/mgf/src/models/TP/fastpredNF.py
code served (permissive licence) · get_code("a0ca33c8b6e66193")
DiagonalGaussianConditionalDensity Ran mulplue/mgf/src/models/TP/fastpredNF.py
code served (permissive licence) · get_code("b9e0ff9a558438e1")
LinearMaskedCoupling Ran mulplue/mgf/src/models/TP/fastpredNF.py
code served (permissive licence) · get_code("4d0d2d2745b54b63")
MADE Ran mulplue/mgf/src/models/TP/fastpredNF.py
code served (permissive licence) · get_code("26209689fe626f49")
MaskedLinear Ran mulplue/mgf/src/models/TP/fastpredNF.py
code served (permissive licence) · get_code("ca93d4c44305509b")
cluster_GMM_Dist Ran mulplue/mgf/src/models/TP/fastpredNF.py
code served (permissive licence) · get_code("90c765ce469cc84d")
create_RealNVP_step Ran mulplue/mgf/src/models/TP/fastpredNF.py
code served (permissive licence) · get_code("920ca1a223d05c89")
create_masks Ran mulplue/MGF/src/models/TP/TFCondARFlow.py
code served (permissive licence) · get_code("5a2e1cee97642f25")
cyclical_lr Ran mulplue/MGF/src/models/TP/model_utils.py
code served (permissive licence) · get_code("ff1848a826b358a0")
diagonal_gaussian_log_prob Ran mulplue/mgf/src/models/TP/fastpredNF.py
code served (permissive licence) · get_code("507ad74b8141f11f")
diagonal_gaussian_sample Ran mulplue/mgf/src/models/TP/fastpredNF.py
code served (permissive licence) · get_code("9b77c0354d2d0aaa")
displacement_error Ran mulplue/MGF/src/metrics/TP_metrics.py
code served (permissive licence) · get_code("283aa4275dc465c0")
evaluate_helper Ran mulplue/MGF/src/metrics/TP_metrics.py
code served (permissive licence) · get_code("9bb40f2536a46a93")
evaluate_helper_mean Ran mulplue/MGF/src/metrics/TP_metrics.py
code served (permissive licence) · get_code("ddbf6d5de516a0dc")
exp_anneal Ran mulplue/MGF/src/models/TP/model_utils.py
code served (permissive licence) · get_code("6b2d284bebc1dc60")
get_mlp Ran mulplue/mgf/src/models/TP/fastpredNF.py
code served (permissive licence) · get_code("3508c00acbc73d66")
get_model_device Ran mulplue/MGF/src/models/TP/Trajectron.py
code served (permissive licence) · get_code("5a0beb315ee2d6b3")
load_tuned Ran mulplue/MGF/src/utils/common.py
code served (permissive licence) · get_code("92dfbf360b9c387e")
to_one_hot Ran mulplue/MGF/src/models/TP/model_utils.py
code served (permissive licence) · get_code("8f0b65c232680906")
Build_Metrics Not yet run mulplue/MGF/src/metrics/build_metrics.py
code served (permissive licence) · get_code("6ebdf632e0a6ad62")
FlowSequential Not yet run mulplue/mgf/src/models/TP/fastpredNF.py
code served (permissive licence) · get_code("f8f3b999a11d5d3e")
create_MAF_step Not yet run mulplue/mgf/src/models/TP/fastpredNF.py
code served (permissive licence) · get_code("70f0ce27583cb926")
fastpredNF Not yet run mulplue/mgf/src/models/TP/fastpredNF.py
code served (permissive licence) · get_code("53b654d10ae7d946")
fastpredNF_CIF_separate_cond Not yet run mulplue/mgf/src/models/TP/fastpredNF.py
code served (permissive licence) · get_code("e762f39de147d718")
fastpredNF_CIF_separate_cond_clusterGMM Not yet run mulplue/mgf/src/models/TP/fastpredNF.py
code served (permissive licence) · get_code("4effd10a00bbdc37")
fastpredNF_separate Not yet run mulplue/mgf/src/models/TP/fastpredNF.py
code served (permissive licence) · get_code("ef3d886fa6434784")
fastpredNF_separate_cond Not yet run mulplue/mgf/src/models/TP/fastpredNF.py
code served (permissive licence) · get_code("8f74fa4acbfdec65")
load_config Not yet run mulplue/MGF/src/utils/common.py
code served (permissive licence) · get_code("e158e0cb62772b3d")

Repositories linked to this paper

Some links come from the archived Papers with Code dataset (CC BY-SA 4.0): attribution and licence.

Abstract

To predict future trajectories, the normalizing flow with a standard Gaussian prior suffers from weak diversity. The ineffectiveness comes from the conflict between the fact of asymmetric and multi-modal distribution of likely outcomes and symmetric and single-modal original distribution and supervision losses. Instead, we propose constructing a mixed Gaussian prior for a normalizing flow model for trajectory prediction. The prior is constructed by analyzing the trajectory patterns in the training samples without requiring extra annotations while showing better expressiveness and being multi-modal and asymmetric. Besides diversity, it also provides better controllability for probabilistic trajectory generation. We name our method Mixed Gaussian Flow (MGF). It achieves state-of-the-art performance in the evaluation of both trajectory alignment and diversity on the popular UCY/ETH and SDD datasets. Code is available at https://github.com/mulplue/MGF.

For agents

The same record, over MCP at https://syntology.ai/mcp:

get_harvested_code_for_paper("2402.12238")
get_code_for_paper("2402.12238")
have("2402.12238")

Connect an agent — have() is free.