Transformer Motion Tokens for Scalable Traffic Simulation

Resolve Bottlenecks,
Find Innovative Solutions
Generate Solutions

Solution Overview

Problem

Conventional traffic modeling approaches using machine-learning models such as conditional variational autoencoders (VAEs) or generative adversarial networks (GANs) suffer from a lack of realism and poor scalability, leading to subpar simulation performance in predicting the behavior of multiple agents in complex environments.

Innovation Solution

Employing transformer-based models to tokenize agent movements, allowing for the generation of a fixed number of possible actions per timestep, which are used to train/update models to simulate realistic and scalable traffic movements by selecting and evaluating candidate token sets that best model diverse agent trajectories.

Engineering Contradictions & Design Principles

VSEngineering Contradiction Analysis

1Reliability

If conventional machine-learning models (VAEs or GANs) are used for traffic modeling, then the simulation can be performed, but the realism and scalability are poor leading to subpar simulation performance

Engineering Contradiction:
Improvesimulation performanceVSAvoidscalability
Core Design Contradiction:
ReliabilityVSAdaptability or versatility

Solution Approach 1:

The patent transforms the continuous action space into a discrete token vocabulary, fundamentally changing the parameter representation from continuous coordinates to discrete symbols. This enables the model to scale efficiently while maintaining realism, as the discrete tokens can be processed more effectively by the transformer architecture.

Inventive Principle:
Principle #35Parameter changes

Solution Approach 2:

The patent replaces conventional machine-learning models (VAEs and GANs) with a transformer-based model that uses discrete tokens to represent agent actions. This substitution improves both the realism of simulations and the scalability to handle multiple agents, directly addressing the limitations of previous approaches.

Inventive Principle:
Principle #28Mechanics substitution (Replace mechanical system)

2Productivity

If discrete actions with beam-searching are used for traffic modeling, then computation can be performed, but scalability remains poor

Engineering Contradiction:
Improvecomputation efficiencyVSAvoidscalability
Core Design Contradiction:
ProductivityVSAdaptability or versatility

Solution Approach 1:

The patent changes the parameter representation from continuous action vectors to discrete tokens from a fixed vocabulary. This transformation enables efficient processing through the transformer architecture's attention mechanism, improving computational efficiency while maintaining the ability to scale to multiple agents through the discrete token framework.

Inventive Principle:
Principle #35Parameter changes

3Adaptability or versatility

If continuous action spaces are modeled, then movement flexibility is maintained, but computational complexity increases reducing scalability

Engineering Contradiction:
Improvemovement flexibilityVSAvoidcomputational complexity
Core Design Contradiction:
Adaptability or versatilityVSDevice complexity

Solution Approach 1:

The patent discretizes the continuous action space into a finite vocabulary of tokens representing common movements. This parameter transformation maintains movement flexibility by capturing essential motion patterns while dramatically reducing computational complexity, enabling the model to scale efficiently to simulate multiple agents simultaneously.

Inventive Principle:
Principle #35Parameter changes

Data Source

PatentUS20250111109A1Generating motion tokens for simulating traffic using machine learning models
Publication Date: 2025.04.03 NVIDIA CORP
  • US20250111109A1 patent drawing
  • US20250111109A1 patent drawing
  • US20250111109A1 patent drawing

AI summary

In various examples, systems and methods are disclosed relating to generating tokens for traffic modeling. One or more circuits can identify trajectories in a dataset, and generate actions from the identified trajectories. The one or more circuits can generate, based at least on the plurality of actions and at least one trajectory of the plurality of trajectories, a set of tokens representing actions to generate trajectories of one or more agents in a simulation. The one or more circuits may update a transformer model to generate simulated actions for simulated agents based at least on tokens generated from the trajectories in the dataset.