Vehicle Behavior Prediction Using World-Model Reinforcement Learning

Resolve Bottlenecks,
Find Innovative Solutions
Generate Solutions

Solution Overview

Problem

Current reinforcement learning methods for training driving policies in autonomous vehicles face challenges in transferring simulator-trained policies to real-world scenarios due to domain differences and safety concerns, limiting their wider adoption.

Innovation Solution

The use of model-based reinforcement learning that leverages a deep neural network trained on real-world data to predict actor movements, allowing for simulation of unfeasible conditions and safe training of policies for autonomous vehicles.

Engineering Contradictions & Design Principles

VSEngineering Contradiction Analysis

1Measurement precision

If reinforcement learning is applied directly to real robotic systems to train driving policies, then the policies can be trained on actual driving conditions, but safety requirements are violated by placing drivers, operators, or objects in dangerous situations

Engineering Contradiction:
Improvetraining accuracy on real conditionsVSAvoidsafety risks to drivers and operators
Core Design Contradiction:
Measurement precisionVSObject-affected harmful factors

Solution Approach 1:

The patent creates a high-fidelity simulation environment that copies real-world driving conditions, physics, and actor behaviors. This virtual copy allows RL training to occur safely while maintaining realistic training scenarios, resolving the contradiction between needing real-world conditions and avoiding real-world dangers.

Inventive Principle:
Principle #26Copying

Solution Approach 2:

The simulation environment acts as an intermediary between the RL training process and the real world. It mediates the training by providing realistic driving scenarios without exposing actual drivers or operators to dangerous situations, allowing policy training to proceed safely.

Inventive Principle:
Principle #24Intermediary (Mediator)

2Object-affected harmful factors

If a simulator is used to train driving policies with reinforcement learning, then safety is maintained, but accurate results require simulating all driving conditions which is unfeasible

Engineering Contradiction:
Improvesafety of drivers and operatorsVSAvoidcomplexity of simulating all driving conditions
Core Design Contradiction:
Object-affected harmful factorsVSDevice complexity

Solution Approach 1:

The system pre-trains deep neural networks on extensive real-world driving data before using them in the simulation. This preliminary action allows the simulation to leverage pre-learned representations of driving conditions, reducing the need to explicitly simulate every possible scenario and making the simulator more feasible.

Inventive Principle:
Principle #10Preliminary action

Solution Approach 2:

The trained deep neural networks serve multiple functions: they predict actor movements, define simulation states, and generalize across different driving conditions. This multi-functionality allows a single simulation framework to handle diverse driving scenarios without requiring separate simulations for each condition.

Inventive Principle:
Principle #6Universality (Multi-functionality)

3Measurement precision

If deep neural networks are trained on real-world data to predict actor movements, then the predictions can define simulation states, but the transfer from simulation to real-world deployment remains challenging due to domain differences

Engineering Contradiction:
Improveprediction accuracy from real-world dataVSAvoidtransferability from simulation to real world
Core Design Contradiction:
Measurement precisionVSAdaptability or versatility

Solution Approach 1:

The system uses feedback loops where the simulation environment provides realistic interactions and consequences that mirror the real world. The RL agent learns from simulated outcomes and continuously refines its policy, improving adaptability to real-world conditions through iterative learning from simulated feedback.

Inventive Principle:
Principle #23Feedback

Solution Approach 2:

The system transitions parameters and policies learned in the simulation environment to the real-world deployment. By carefully managing parameter transfer and using the simulation to pre-adjust policies, the system bridges the domain gap between simulation and real-world operation.

Inventive Principle:
Principle #35Parameter changes

Data Source

PatentUS20220138568A1Model-based reinforcement learning for behavior prediction
Publication Date: 2022.05.05 NVIDIA CORP
  • US20220138568A1 patent drawing
  • US20220138568A1 patent drawing
  • US20220138568A1 patent drawing

AI summary

In various examples, reinforcement learning is used to train at least one machine learning model (MLM) to control a vehicle by leveraging a deep neural network (DNN) trained on real-world data by using imitation learning to predict movements of one or more actors to define a world model. The DNN may be trained from real-world data to predict attributes of actors, such as locations and/or movements, from input attributes. The predictions may define states of the environment in a simulator, and one or more attributes of one or more actors input into the DNN may be modified or controlled by the simulator to simulate conditions that may otherwise be unfeasible. The MLM(s) may leverage predictions made by the DNN to predict one or more actions for the vehicle.