The present invention pertains to planning in autonomous vehicles and other mobile robots.
A rapidly emerging technology is autonomous vehicles (AVs) that can navigate by themselves on urban roads. Such vehicles must not only perform complex manoeuvres among people and other vehicles, but they must often do so while guaranteeing stringent constraints on the probability of adverse events occurring, such as collision with these other agents in the environments. An autonomous vehicle, also known as a self-driving vehicle, refers to a vehicle which has a sensor system for monitoring its external environment and a control system that is capable of making and implementing driving decisions automatically using those sensors. This includes in particular the ability to automatically adapt the vehicle's speed and direction of travel based on perception inputs from the sensor system. A fully-autonomous or “driverless” vehicle has sufficient decision-making capability to operate without any input from a human driver. However, the term autonomous vehicle as used herein also applies to semi-autonomous vehicles, which have more limited autonomous decision-making capability and therefore still require a degree of oversight from a human driver. Other mobile robots are being developed, for example for carrying freight supplies in internal and external industrial zones. Such mobile robots would have no people on board and belong to a class of mobile robot termed UAV (unmanned autonomous vehicle). Autonomous air mobile robots (drones) are also being developed.
Recent work in the field has considered planning formulated as a constrained non-linear optimization problem, though mainly in the context of assistive (not fully-autonomous) driving. W. Schwarting, J. Alonso-Mora, L. Paull, S. Karaman, and D. Rus, “Safe nonlinear trajectory generation for parallel autonomy with a dynamic vehicle model,” IEEE Transactions on Intelligent Transportation Systems, no. 99, pp. 1-15, 2017, denoted by [C-1] herein, and incorporated herein by reference in its entirety, discloses a “Parallel Autonomy” framework which allows deviation from human driver inputs to maintain safety, but seeks to minimize extreme intervention. A receding horizon planner is formulated as Nonlinear Model Predictive Control (NMPC), subject to a set of hard constraints, namely (1) respecting a transition model of the system, including kinematic and dynamic constraints (2) maintaining the vehicle within the limits of the road and (3) avoiding other traffic participants in the sense of guaranteeing a probability of collision below pϵ. A cost function penalizes deviation from a current acceleration and steering angle specified by a human driver—a form of “soft constraint” or “objective” according to the terminology used herein.
The techniques of [C-1] formulate planning as a non-linear constrained optimization problem, and seek to solve that problem using a receding horizon formulation (the solution being a set of control inputs that optimize the cost function). It is possible to extend this framework to more complex cost functions, for example with explicit comfort objectives, or to include more sophisticated vehicle (or, more generally, mobile robot) models. However, a challenge with this approach is that convergence to an optimal set of control inputs is both slow and uncertain, particularly as the complexity of the cost function and/or the mobile robot model increases, i.e. the optimizer may take a long time to converge to a solution, or may never successfully converge. Another challenge is that non-linear constrained optimization solvers tend to be local in nature and thus have a tendency to converge to local optima which may be far from the globally optimal solution. This can significantly impact mobile robot performance.
The present invention also formulates mobile robot planning as a constrained optimization problem. Given a scenario with a desired goal, the problem is to find a series of control actions (“policy”) that substantially (exactly or approximately) globally optimises a defined cost function for the scenario and the desired goal.
A core issue addressed herein is speed of convergence in constrained optimization trajectory planning, i.e. for planners that formulate trajectory planning (synthesis) as a constrained optimization subject to a set of hard constraints. Implementing constrained optimization-based planners that can not only provide highly quality trajectories but also converge to an acceptable solution (trajectory) in real-time is challenging using current hardware and state of the art solvers.
In the context of constrained optimization planning, hard constraints are constraints that a planned trajectory must be guaranteed to satisfy. These could, for example, be constraints on collision avoidance (avoiding collisions with static or moving obstacles), permitted area constraints (e.g. constraining planned trajectories to keep within a road layout or other permitted area), or comfort constraints for autonomous vehicles.
One aspect herein provides a computer system for planning mobile robot trajectories, the computer system comprising: an input configured to receive a set of scenario description parameters describing a scenario and a desired goal for the mobile robot therein; a runtime optimizer configured to compute a final mobile robot trajectory that substantially optimizes a cost function for the scenario, subject to a set of hard constraints that the final mobile robot trajectory is guaranteed to satisfy; and a trained function approximator configured to compute, from the set of scenario description parameters, initialization data defining an initial mobile robot trajectory; wherein the computer system is configured to initialize the runtime optimizer with the initialization data, in order to guide the optimizer from the initial mobile robot trajectory to the final mobile robot trajectory that satisfies the hard constraints, the function approximator having been trained on example sets of scenario description parameters and ground truth initialization data for the example sets of scenario description parameters.
The function approximator may, for example, take the form of a neural network. Function approximators may require significant amounts of data to train, and therefore require significant resources in training. However, once trained, a function approximator is efficient to implement, and can be applied to scenarios in order to generate initialization data quickly, even on resource-constrained platforms.
The initialization data provides a starting point for the search by the optimizer for a globally optimal solution (the final trajectory). It could, for example, take the form of an initial trajectory and/or an initial sequence of control actions that define the initial vehicle trajectory in conjunction with an ego dynamics model(s) (or some other initialization data derived from one of both of those).
The initial trajectory is not necessarily guaranteed to satisfy any hard constraints (even if such constrains have been imposed during training—see below). However, this is not an issue, because the initial trajectory is only used to initialize the runtime optimizer. The final trajectory determined by the runtime optimizer is guaranteed to satisfy the hard constraints.
In addition to speed/convergence time, the present disclosure addresses local optima problems that arise in the context of planning based on constrained optimization.
In general, the full constrained optimization problem to be solved is non-linear. In embodiments, the runtime planner may therefore take the form of a constrained non-linear optimizer. Non-linear optimizers are particularly vulnerable to convergence to locally but non-globally optimal solutions. However, a high-quality initialization that is reasonably close to the global optima can significantly instances of convergence to non-local optima.
In embodiments, the training data used to train the function approximator may also be generated using a constrained optimization-based planner, i.e. the function approximator may be trained to approximate (imitate) a constrained non-linear optimization.
For example, in certain embodiments, the function approximator may be trained to a multi-stage constrained optimisation planner, in which a first stage is formulated as an optimization problem that is similar to, but simpler than a more complex planning problem that ultimately needs to be solved. For example, the first stage may use a linear cost function and linear robot dynamics model. Such a problem is generally less susceptible to local optima. The solution of the first stage is then used to initialize the second stage, in which the “full” planning problem is solved. The solution of the first stage will be more likely to be close to a globally optimal solution to the full planning problem—up to up to some acceptable level of error introduced by the simplification assumptions of the first stage—and therefore reduces the tendency of the second stage to converge to local optima far from the global solution.
This two-stage approach is highly effective at preventing unwanted convergence to non-local optima. However, it is challenging to implement in real-time at present.
Note, the first and second optimization stages form part of a planner that the function approximator is trained to imitate—referred to herein as a “reference” or “expert” planner. Those stages do not need to be implemented at runtime, because the trained function approximator is implemented at runtime instead. The expert planner is therefore not required to be able to operate in real-time in order to achieve real-time operation at runtime.
The runtime optimizer is separate from both of the above constrained optimization stages—although, in some embodiments, it may implement the same or similar logic to the second stage of the multi-stage planner. In contrast to the multi-stage planner, the runtime optimizer is implemented at runtime. Because the runtime optimizer is provided with a high-quality initialization from the trained function approximator, it is significantly less likely to converge to local optima, and can also complete its search for the final mobile robot trajectory in real time.
In embodiments, the function approximator may have been trained to approximate a reference planner, the ground truth initialization data having been generated by applying the reference planner to the example training scenarios.
The runtime optimizer may be configured to determine a series of control actions, and compute the final mobile robot trajectory by applying a robot dynamics model to the series of control actions.
The initialization data may comprise an initial sequence of control actions defining the initial mobile robot trajectory.
The hard constraints may comprise one or more collision avoidance constraints for one or more static or moving obstacles in the scenario, and location(s) of the static or moving obstacles may be encoded in the set of scenario description parameters for use by the function approximator.
The hard constraints may comprise one or more permitted area constraints for keeping the mobile robot within a permitted area, and the permitted area may be encoded in the set of scenario description parameters for use by the function approximator.
The goal may be defined relative to a reference path, and the cost function may encourage achievement of the goal by penalizing at least one of lateral deviation from the reference path, and longitudinal deviation from a reference location on the reference path.
The initial and final trajectories may be represented in a frame of reference defined by the reference path.
The function approximator may have a neural network architecture.
The function approximator may have a convolutional neural network (CNN) architecture.
The computer system may be configured to transform the set of scenario description parameters into an input tensor comprising one or more images visualizing the permitted area and/or the location(s) of the obstacles.
The input tensor may comprise multiple images visualizing predicted locations of the obstacles at different time instants.
The image(s) may encode the reference path.
For example, in the case that initial and final trajectories are represented in a frame of reference defined by the reference path, the image(s) may encode the reference path by visualizing the permitted area and/or the location(s) of the obstacles in the frame of reference defined by the reference path.
The function approximator may encode the initial mobile robot trajectory as a set of smooth function parameters.
A second aspect herein provides a method of configuring a mobile robot planner, the method comprising:
A third aspect herein a method of training a function approximator to imitate a reference planner (expert planner), the method comprising:
In embodiments of the first or second aspects, the reference planner may be a multi-stage optimization-based planner, and the training data set may be generated by, for each example set of scenario description parameters:
Note, this “initialization data” is internal to the multi-stage planner, and is separate from the “initialization data” provided by the trained function approximator at runtime. It will be clear in context which is meant. Where useful to distinguish explicitly, the former may be referred to as “internal initialization data”, and the latter may be referred to as “runtime initialization data”.
The function approximator may be trained using Dataset Aggregation, by applying the function approximator in simulation to determine additional sets of scenario description parameters, applying the reference planner to the additional sets of scenario description parameters to compute ground truth initialization data for the new sets of scenario description parameters, and re-training the function approximator based thereon.
The runtime optimizer may be a non-linear optimizer.
The “full” multi-stage optimization approach mentioned above may be implemented as follows, in order to generate training data for training the function approximator.
As noted, in embodiments the runtime optimizer may implement the same or similar logic to the second constrained optimization stage of the multi-stage optimization-based planner. All disclose below pertaining to the second constrained optimization phase applies equally to the runtime optimizer in such embodiments.
The two-stage optimization approach may be implemented as computer-implemented method of determining control actions for controlling a mobile robot, the method comprising:
A less-complex cost function and dynamics model can be used in the first stage, whilst still providing an effective initialization to the second stage. In the present context, the term complexity refers to the form of the cost function and the model, in the space of variables over which they are defined.
A robot dynamics model is a predictive component that predicts how the mobile robot will actually move in the scenario in response to a given sequence of control actions, i.e. it models the mobile robot's response to control actions. A higher-complexity model, as used in the second stage, can model that response more realistically. The lower-complexity model is free to use highly-simplifying assumptions about the behaviour of the mobile robot but these may be relatively unrealistic. Depending on the simplifying assumptions applied in the first stage, the first predicted trajectory may not even be fully dynamically realisable.
A higher-complexity cost function and model, as used in the second stage, can provide superior trajectories, which may be of sufficient quality that they can be used as a basis for effective planning and/or control decisions. However, generally speaking, higher-quality trajectories will be obtained when convergence to an approximately globally optimal solution (i.e. at or near a global optima of the full cost function) is achieved. As the complexity of the full cost function and model increases, achieving such convergence becomes increasingly dependent on the quality of the initialization.
By contrast, the simplifying assumptions applied in the first stage make it inherently less susceptible to the problem of non-local optima, i.e. the ability of the first optimizer to converge to an approximately globally optimal solution is far less dependent on any initialization of the first optimization phase. The output of the simplified first stage is unlikely to be of sufficient quality to use as a basis for such planning decisions directly, and the trajectories it produces may not even be full dynamically realisable (depending on the simplifying assumptions that are made in the first stage). Nevertheless, provided the solution of the first stage is reasonably close to the globally optimal solution of the second stage, the initialization data of the first stage can still facilitate faster and more reliable convergence to an approximately globally optimal solution in the second stage, which will correspond to a dynamically realisable trajectory.
The present invention thus benefits from the high-quality output of the more complex second stage, whilst avoiding (or at least mitigating) the issues of local optima convergence that would otherwise come with it, through the provision of an effective initialization to the second stage.
The described embodiments consider a two-stage constrained optimization. However, other embodiments may use more than two stages. In that case, the first constrained optimization stage that is applied to determine the initialization data could, itself, be a multi-stage optimization. In that case, two or more preliminary cost functions may be optimized in the first stage, with at least one of the preliminary cost functions being optimized in order to initialize another of the preliminary cost functions, before ultimately determining the initialization data to the above-mentioned second constrained optimization stage.
In embodiments, the computed trajectory may be determined, based on an initial mobile robot state, as a series of subsequent mobile robot states. Each mobile robot state of the first computed trajectory may be determined by applying the full robot dynamics model to at least the previous mobile robot state of the first computed trajectory and a corresponding control action of the first series of control actions. Each mobile robot state of the second computed trajectory may be determined by applying the full robot dynamics model to at least the previous mobile robot state of the second computed trajectory and a corresponding control action of the second series of control actions.
The preliminary robot dynamics model may be linearly dependent on at least the previous mobile robot state of the first computed trajectory and the corresponding control action of the first series of control actions, and the full robot model may be non-linearly dependent on at least one of the previous mobile robot state of the second computed trajectory and the corresponding control action of the second series of control actions.
The preliminary cost function may be linearly dependent on the mobile robot states of the first computed trajectory, and the full cost function may be non-linearly dependent on the mobile robot states of the second computed trajectory.
The preliminary cost function may be linearly dependent on the control actions of the first series, and the full cost function may be non-linearly dependent on the control actions of the second series.
The first optimizer may be a mixed integer linear programming (MILP) optimizer, and the second optimizer may be a non-linear programming (NLP) optimizer.
The hard constraints of the first stage may comprise one or more mixed integer collision avoidance constraints for one or more static or moving obstacles in the scenario and/or one or more mixed integer permitted area constraints for keeping the mobile robot within a permitted area. The hard constraints of the second stage may comprise one or more similar collision avoidance and/or permitted area constraints formulated in terms of non-integer variables.
The first optimizer may apply a receding horizon approximation to iteratively optimize component costs of the preliminary cost function, and thereby determine the first series of control actions, and the second optimizer may not use any receding horizon approximation and may instead optimize the full loss function as a whole.
The goal may be defined relative to a reference path, and each cost function may encourage achievement of the goal by penalizing at least one of lateral deviation from the deference path, and longitudinal deviation from a reference location on the reference path.
Each of the computed trajectories may be represented in a frame of reference defined by the reference path.
The preliminary cost function may be linearly dependent on the above lateral and/or longitudinal deviation, and the full cost function is non-linearly dependent thereon.
Both cost functions may penalize deviation from a target speed.
The method may be implemented in a planner of a mobile robot and comprise the step of using control data of at least one of: the second computed trajectory, and the second series of control actions to control motion of the mobile robot.
Embodiments and optional implementations of the invention address the problem of speed though the use of function approximation. Depending on the resources of the available hardware platform, there may be occasions when running the first and second optimizations in real-time is not feasible. As an optional optimization, one or both of the optimizers may be replaced, in a real-time context, with a function approximator training to approximate the first and/or second optimization stage as applicable.
For example, the method may be performed repeatedly for different scenarios so as to generate a first training set comprising inputs to the first optimizer and corresponding outputs computed by the first optimizer, and the training set may be used to train a first function approximator to approximate the first optimizer (training method 1).
Alternatively or additionally, the method may be performed repeatedly for different scenarios so as to generate a second training set comprising inputs to the second optimizer and corresponding outputs computed by the second optimizer, and the training set may be used to train a second function approximator to approximate the second optimizer (training method 2).
The method may comprise the step of configuring a runtime stack of mobile robot to implement one of the following combinations:
Alternatively, the method may be performed repeatedly for different scenarios so as to generate a training set comprising inputs to the first optimizer and corresponding outputs computed by the second optimizer, and the training set may be used to train a single function approximator to approximate both of the first and the second optimizers (training method 3).
The method may comprise the step of configuring a runtime stack of mobile robot to implement the single function approximator.
The runtime stack may be configured to implement one of combinations (i) and (ii), or the single function approximator, and the method may comprise an additional step of configuring the runtime stack with a verification component configured to verify an output of the second function approximator or the single function approximator.
A further aspect of the invention provides a computer-implemented method of determining control actions for controlling a mobile robot, the method comprising:
A further aspect herein provides a computer system comprising one or more optimization components configured to implement or approximate: a first optimization stage as defined in any of the above aspects or embodiments; and a second optimization stage as defined in any of the above aspects or embodiments.
The one or more optimization components may comprise a first optimization component configured to implement or approximate the first optimization stage, and a second optimization component configured to implement or approximate the second optimization stage, using initialization data provided by the first optimization component.
The first optimization component may take the form of a first function approximator trained in accordance with training method 1 above and/or the second optimization component may take the form of a second function approximator trained in accordance with training method 2 above to approximate the second optimization stage.
The second optimization component may take the form of a second function approximator, and the computer system may additionally comprise a verification component configured to verify at least one of a second trajectory and a second series of control actions computed by the second function approximator.
The one or more optimization components may comprise a single optimization component, trained in accordance with training method 3 above, to approximate both of the first and second optimization stages.
The computer system may additionally comprise a verification component configured to verify at least one of a second trajectory and a second series of control actions computed by the single function approximator.
The or each function approximator may have a neural network architecture.
In alternative embodiments, the function approximator may be trained to implement a different form of expert planner. For example, the function approximator may be trained to implement only the first constrained optimization stage (in which case only that stage need be implemented in order to generate the training data), and this can still provide an acceptable initialization.
The computer system may be embodied in an autonomous vehicle or other mobile robot, wherein the computer system may be further configured to control the motion of the mobile robot via one or more actuators of the mobile robot using control data provided by the second optimization component.
A further aspect herein provides a computer program for programming one or more computers to implement any of the methods disclosed herein.
To assist understanding of the present disclosure and to show how embodiments of the present invention may be put into effect, reference is made by way of example to the accompanying drawings, in which:
Various embodiments of the present disclosure are described below. The described embodiments provide what is referred to herein as a “PILOT” planner. This is a trajectory planner, in which a trained function approximator provides an initialization to a runtime optimizer. The PILOT planner is able to find, in real-time, a final mobile robot trajectory that is substantially optimal, in the sense optimizing one or more soft constraints for a given scenario, but which is also guaranteed to satisfy a set of hard constraints for the scenario.
In some of the described embodiments, the function approximator is trained to approximate all or part of a multi-stage optimization-based planner, which is described in detail below to provide context.
Embodiments are described in the context of an autonomous vehicle. However, the description applies equally to other forms of autonomous mobile robot.
The perception stack 102 receives sensor outputs from an on-board sensor system 110 of the AV.
The on-board sensor system 110 can take different forms but generally comprises a variety of sensors such as image capture devices (cameras/optical sensors), LiDAR and/or RADAR unit(s), satellite-positioning sensor(s) (GPS etc.), motion sensor(s) (accelerometers, gyroscopes etc.) etc., which collectively provide rich sensor data from which it is possible to extract detailed information about the surrounding environment and the state of the AV and any external actors (vehicles, pedestrians, cyclists etc.) within that environment.
Hence, the sensor outputs typically comprise sensor data of multiple sensor modalities such as stereo images from one or more stereo optical sensors, LiDAR, RADAR etc.
Stereo imaging may be used to collect dense depth data, with LiDAR/RADAR etc. proving potentially more accurate but less dense depth data. More generally, depth data collection from multiple sensor modalities may be combined in a way that respects their respective levels (e.g. using Bayesian or non-Bayesian processing or some other statistical process etc.). Multiple stereo pairs of optical sensors may be located around the vehicle e.g. to provide full 360° depth perception. This provides a much richer source of information than is used in conventional cruise control systems.
The perception stack 102 comprises multiple perception components which co-operate to interpret the sensor outputs and thereby provide perception outputs to the prediction stack 104.
The perception outputs from the perception stack 102 are used by the prediction stack 104 to predict future behaviour of the external actors.
Predictions computed by the prediction stack 104 are provided to the planner 106, which uses the predictions to make autonomous driving decisions to be executed by the AV in a way that takes into account the predicted behaviour of the external actors.
The planner 106 implements the techniques described below to plan trajectories for the AV and determine control actions for realizing such trajectories. In particular, a core function of the planner 106 is to determine a series of control actions for controlling the AV to implement a desired goal in a given scenario. In a real-time planning context, a scenario is determined using the perception stack 102 but can also incorporate predictions about other actors generated by the prediction stack 104. A scenario is represented as a set of scenario description parameters used by the planner 106. A typical scenario would define a drivable area and would also capture predicted movements of any obstacles within the drivable area (such as other vehicles) along with a goal. A goal would be defined within the scenario, and a trajectory would then need to be planned for that goal within that scenario. In the following, obstacles are represented probabilistically in a way that reflects the level of uncertainty in their perception within the perception stack 102.
The goal could for example be to enter a roundabout and leave it at a desired exit; to overtake a vehicle in front; or to stay in a current lane at a target speed (lane following). The goal may, for example, be determined by an autonomous route planner (not shown).
The controller 108 executes the decisions taken by the planner 106 by providing suitable control signals to on-board actuators 112 such as motors of the AV. In particular, the controller 108 controls the actuators in order to control the autonomous vehicle to follow a trajectory computed by the planner 106.
As described in further detail below, the planner 106 plans over acceleration (magnitude) and steering angle control actions simultaneously, which are mapped to a corresponding trajectory by modelling the response of the vehicle to those control actions. This allows constraints to be imposed both on the control actions (such as limiting acceleration and steering angle) and the trajectory (such as collision-avoidance constraints), and ensures that the final trajectories produced are dynamically realisable. The planner 106 will determine an optimal trajectory and a corresponding sequence of control actions that would result in the optimal trajectory according to whatever vehicle dynamics model is being applied. The control actions determined by the planner 106 will not necessarily be in a form that can be applied directly by the controller 108 (they may or may not be). Ultimately, the role of the planner 106 is to plan a trajectory and the role of the controller 108 is to implement that trajectory. The term “control data” is used herein to mean any trajectory information derived from one of both of the planned trajectory and the corresponding series of control actions that can be used by the controller 108 to realize the planner's chosen trajectory. For example, the controller 108 may take the trajectory computed by the planner 106 and determine its own control strategy for realizing that trajectory, rather than operating on the planner's determined control actions directly (in that event, the controller's control strategy will generally mirror the control actions determined by the planner, but need not do so exactly).
The planner 106 will continue to update the planned trajectory as a scenario develops. Hence, a trajectory determined at any time may never be realized in full, because it will have been updated before then to account for changes in the scenario that were not predicted perfectly.
These functions of the planner 106 will now be described in detail. As set out above, the present disclosure addresses the planning problem of determining an optimal series of control actions (referred to herein as a “policy”) for a given scenario and a given goal as a constrained optimization problem with two optimization stages.
In the described embodiments, the first optimization stage solves a Mixed Integer Linear Programming (MILP) problem where obstacles lead to hard constraints for the MILP. The outcome of this seeds (initializes) a subsequent nonlinear optimisation of the trajectory for dynamics and comfort constraints, in the second optimization stage. The second optimization stage solves a Non-Linear Programming (NLP) problem, initializing using the results of the MILP optimization stage. The MILP stage uses a mixed-integer formulation of collision avoidance and drivable area constraints as described later.
Although performed second, the NLP stage is described first, in order to set out the final planning problem that ultimately needs to be solved (Problem 2 below), and to provide some context for the subsequent description of how best to provide an effective initialization (seed) from the MILP stage.
Before describing either stage in detail, the description sets out a general framework applicable to both stages.
A. Notation and Definitions
Following notation from [C-1], an index k≡tk, is assumed where tk=t0+kΔt with t0 being the current time and Δt a fixed timestep. A trajectory is defined over the next N steps covering a temporal horizon of τ=NΔL Vectors are in bold. A shorthand notation ri:e={ri, . . . , re} is used for any variable r.
A vehicle whose motion is being planned is referred to as the ego vehicle. The ego vehicle's state at time k is given by Xk=(Xk, Yk, Vk, Φk)∈X, where (Xk, Yk) is the position of the vehicle, Vk is the speed and Φk is its heading in a global coordinate frame . The ego state at time 0 is given by X0, and a function of the planner 106 is to determine the ego states over the next N steps, X1:N∈XN.
Other traffic participants, such as vehicles, pedestrians and cyclists, are indexed by i∈{1, . . . , n} and their pose at time k is modelled using a Gaussian distribution having a mean Oki=(Xki, Yki, Φki)∈ and covariance τki. The set of the mean poses of all traffic participants over time is O0:N1:n, and similarly for the covariances τ0:N1:n, and both are given as inputs to the planning problem.
B. Reference Path-Based Representation
To simplify the process of defining a planning goal, the global coordinate frame is transformed to a reference path-based representation under an invertible transform T as described in Sec. II-B. This representation significantly simplifies the problem of path tracking.
A goal of the ego vehicle is defined as following a differentiable and bounded two-dimensional reference path in the global coordinate frame, ref parameterized by the distance from the start of the path ((λ), (λ)). Tangential and normal vectors of the reference path in the global coordinate frame can be obtained at any point A along the path ref as
respectively.
The reference path ref is a path which the ego vehicle is generally intending to follow, at a set target speed. However, deviation from the reference path and target speed, whilst discouraged, are permitted provided that no hard constraints (such as collision avoidance constraints) are breached. The reference path can be determined using knowledge of the road layout, which may use predetermined map data (such as an HD map of the driving area), information from the perception stack 104, or a combination of both. For complex layouts in particular (such as complex junctions or roundabouts), the reference path could be learned by monitoring the behaviour of other drivers in the area over time.
The invertible transform operates over three types of input: (1) poses, (2) velocities and (3) covariance matrices. Each of the individual operations is described next.
1) Pose transform: maps poses (X, Y, Φ) in the global coordinate frame to poses (x, y, ϕ) in the reference path frame r as shown in
As will be appreciated, the inverse transformation can be derived straightforwardly by applying the same geometric principles.
2) Velocity transform: since is defined spatially, speeds are invariant to it: ν=(V)=V.
3) Covariance transform: considering a traffic participant with pose O and covariance τ, such that (O)=[x y ϕ]T, the transformed covariance matrix in the reference path coordinate frame is given by:
Σ=(τ)=R(∠tx−ϕ)τR(<tx−ϕ)T (2)
where tx is the tangent of ref evaluated at λ=x, and R∈SO (2) is a rotation matrix.
C. Problem Statement
Following the reasoning presented in Sec. II-A, the planning problem is defined in the reference path coordinate frame.
The state of the ego vehicle at time k is given by xk=(xk, yk, vk, ϕk)∈X where (xk, yk) is the 2D position of the vehicle, νk is its speed and ϕk is its heading. The evolution of the vehicle state is given by a discrete general dynamical system:
X
k+1
=f
Δt(xk,uk), (3)
where fΔt is a discrete non-linear function with time parameter Δt, and uk=(ak, δk)∈u are an acceleration and steering angle, respectively, applied at time k. The ego vehicle is modelled as a rigid body occupying an area Se⊂2 relative to its center, and the area occupied by the ego vehicle at state xk is given by (xk)⊂2.
For other traffic participants, pose at time k is described probabilistically with mean oki=(xki, yki, ϕki)∈ and covariance Σki, for i∈{1, . . . , n}. Following the definition from [C-1], the area each traffic participant occupies is defined as i(oki, Σki, p∈)⊂2 with probability larger than p∈.
A driveable surface area ⊂2 is defined as the area in which it is safe for the ego vehicle to drive, with the unsafe area out=2\. With a cost function J(x0:N, u0:N−1, o0:N1:n, Σ0:N1:n) defined over the positions and controls of the ego vehicle and states and uncertainties of other traffic participants, a “policy synthesis” problem can be formulated.
As indicated above, a “policy” in the present context refers to a time-series of control actions, which is mapped to a corresponding vehicle trajectory using a vehicle dynamics model.
Problem 1 (Policy Synthesis). Given an initial ego state x0, and trajectories of other traffic participants (o0:N1:n, Σ0:N1:n), compute an optimal policy:
This section describes a specific solution to Problem 1, posing it as a Non-Linear Programming (NLP) problem with a set of hard constraints and a multi-objective cost function. The hard constraints comprise (1) vehicle model constraints, including kinematic ones on the transitions of the model and on the controls it allows (Sec. III-1); and (2) collision avoidance constraints with the purpose of avoiding collisions with the boundaries of the road (which can include, e.g. construction areas or parked vehicles) as well as other traffic participants (Sec. III-2). Soft constraints make up the terms of the cost function (i.e. are embodied in the cost function itself) and are presented in Sec. III-3, whereas hard constraints constrain the optimization of the cost function. Sec. III-4 describes the full problem.
1) Vehicle Model: A “kinematic bicycle” model is applied, which is a simplified car model focused on the center of the vehicle. Assuming the ego vehicle to be a rectangle with an inter-axel distance of L, this model can be discretized under Δt as:
While the kinematic bicycle model is not strictly limited by any specific values of maximum steering and acceleration, a limit is imposed on the allowed ranges of controls by enforcing |δk|≤δmax and amin≤ak≤amax, as well as limiting jerk |ak+1−ak|≤{dot over (a)}max and angular jerk |δk+1−δk|≤{dot over (δ)}max. These maintain the model within the operational domain, and also help to ensure that the resulting trajectories respect the rules of the road and passenger comfort. Finally, speed is constrained as 0≤νmin≤νk≤νmax, to guarantee forward motion within a set speed limit νmax.
2) Collision Avoidance: The collision avoidance problem is divided into two different types of avoidance: the limits of the road (which may include other features such as construction zones or parked vehicles) and other traffic participants. For both, a representation of the area of the ego-vehicle is defined as (xk) for state xk. A simplified representation is used wherein the ego vehicle boundary is represented solely by its corners. For a rectangular vehicle of width w and length l, the positions of the corners can be defined as a function of the state xk as:
where z∈Z={[1 1], [−1 1], [−1 −1], [1 −1]}, R(ϕk)∈SO(2) is the rotation matrix corresponding to the ego-vehicle's heading, and o is the Hadamard product. Following this definition, the maintenance of the ego-vehicle within the driveable surface, corresponding to the constraint defined as (xk)∩out=∅, can be reduced to the constraint
∀z∈Z:cz(xk)∈B. (7)
The driveable surface is described by limits, bl(λ), br(λ), which are continuous and differentiable functions of λ defining the left and right borders, respectively, with ∀λ: bl(λ)<br(λ). These boundaries impose a constraint on the allowed lateral deviation at position x, such that keeping the ego-vehicle within the driveable surface, (xk)∩Bout=∅, can be reduced to the set of constraints
∀z∈Z:bl(xkz)≤kz≤br(xkz), (8)
with [xkz, ykz]T defined as in (6).
For other traffic participants, the area occupied i(oik, Σki, p∈) by each participant is modelled in a similar fashion to [C-1], with the exception that the uncertainty at time k can be a non-diagonal matrix Σk for the purpose of the soft constraints introduced next in Sec. III-3.
However, for the purposes of illustration, a diagonal matrix is considered to obtain:
S
i(oki,Σki,p∈)⊂(+ashape,+bshape)=(aki,bki) (9)
where L(aki, bki) is an ellipse which conservatively inscribes vehicle i at time k up to an uncertainty p∈. Thus, the set of constraints can be expressed as:
3) Cost function: The multi-objective cost function is defined for a set of soft constraints over the states and controls of the ego, as well as the positions and uncertainty of the other traffic participants. If a soft constraint l∈ is defined via a function θl(x, u, o1:n, Σ1:n), and a weight ωl∈+, the cost function can be defined as:
The weights associated with the soft constraints determine their relative importance. Soft constraints are defined over different objectives:
4) NLP formulation: The optimization problem is formulated using the constraints and cost J
Problem 2 (Non-Linear Programming Problem). Given an initial ego state x0, and trajectories of other traffic participants (o0:N1:n, Σ0:N1:n), and the set of soft constraints , compute the optimal policy:
Due to the non-linearity of J, fΔt, bl, br and gi,z, this problem is a general (equality and inequality constraints) non-linear, non-convex, constrained optimization problem. While it is appealing to solve the problem directly, or using a receding horizon formulation as in [C-1], there are two major challenges to this approach:
In the above, the tuple (x0, O0:n1:N, Σ0:n1:N) is an example of a set of scenario description parameters, describing a dynamic scenario and a starting state of the ego vehicle within it. The dynamics of the scenario is captured in terms of the motion components of x0, O0:n1:N and Σ0:n1:N, i.e. the speed and heading dimensions within the NLP formulation.
To mitigate these NLP issues, a framework depicted schematically in
The two stage optimisation of
A motivation behind the architecture is to avoid the local optima convergence in the non-linear, non-convex constrained optimization by providing an initial solution that is closer to the global optimum, which also should lead to faster and more reliable convergence. In the two-tier optimization, the first stage 302 solves a linearized version of Problem 2 in a finite, receding horizon manner using a Mixed Integer Linear Programming (MILP) formulation (details presented in Sec. IV-A). This gives guarantees on the optimality for each stage of the receding horizon problem (see Sec. IV-A5), and, thus, acts as a proxy towards reaching the global optimum of the linearized problem. The second stage 304 uses the output of the MILP optimizer as an initial solution and solves the full Problem 2, as set out above. Given the similar representations of the linearized and non-linear problems, it is expected that this initialization improves convergence, speed and the quality of the final solution.
A. Mixed Integer Linear Programming Formulation
In this stage, Problem 2 is re-formulated as a Mixed Integer Linear Programming optimization problem, with mixed integer linear constraints and a multi-objective cost function.
To do so, a linear vehicle model with kinematic feasibility constraints is considered (Sec. IV-A1). An approach to collision avoidance is formulated which maintains the mixed integer linearity of the model (Sec. IV-A2). In Sec. IV-A3 an interpretation of the soft constraints within the MILP cost function is provided. The full MILP problem in Sec. IV-A4 and discuss the optimality of the solution in Sec. IV-A5.
The following sections make use of the non-linear operators |⋅|, max(⋅) and min(⋅), which can be enforced in MILP by considering auxiliary binary variables under the “big-M” formulation [C-8], [C-9]. For example, for a constraint C=max(A,B), assume b∈{0,1} to be a binary variable and M∈+ to be a sufficiently large value. It is then possible to define the corresponding mixed integer constraints:
C≥A
C≥B
C≤A+Mb
C≤B+M(1−b) (13)
Similar definitions can be obtained for the operators |⋅| and min
I) Linear Vehicle Model and Kinematic Feasibility: The kinematic bicycle model presented in Sec. Ill-1 is non-linear, but can be linearized around a point using a series expansion. The problem with this approach lies on the fact that this approximation is only valid around the point, yielding higher errors as the distance to the point increases. To avoid this issue, a different vehicle model with nonholonomic constraints is adopted for the first stage 302.
A state of this linear vehicle model at time k is defined as
νkx=νk cos(ϕk),νky=νk sin(ϕk) (14)
and with controls ūk=[akx aky]∈, where is the input space for the MILP formulation.
This representation is still based on the reference path ref, and is the same as the representation used in the second stage 304 in terms of the spatial dimensions. However, whereas the second stage 304 uses an angular representation of motion (i.e. velocity and acceleration are represented in terms of magnitude and angle components), the first stage uses a linear representation of velocity and acceleration.
The linear vehicle dynamics model is defined as:
k+1
=F
Δt(
where FΔt corresponds to a zero-order hold discretization of the continuous state-space system:
This nonholonomic model is highly simplified when compared to the kinematic bicycle model in (5). To introduce kinematic feasibility, the following constraint is imposed:
νx≥β|νy|
for a given constant ρ∈+. Assuming forward motion, that is
this constraint dictates that any movement in the y direction requires motion in the x direction as well.
Similar constraints are imposed as in the non-linear model of the second stage 304, in particular input bounds constraints, axmin≤axk≤axmax and aymin≤ayk≤aymax; jerk related constraints, |ak+1x−akx<ΔamaxxΔt and |ak+1y−aky|<ΔamaxyΔt; and velocity constraints νxmin≤νyk≤νxmax and νymin≤νyk≤νymax, with νxmin≥0 to guarantee forwards motion.
2) Collision Avoidance: Similarly to the non-linear formulation, collision avoidance is split into avoiding the limits of the road and avoiding other traffic participants. However, a key difference is that the state x does not explicitly model the orientation of the vehicle, so linearly approximating the calculation of the corners of the ego vehicle would induce very high errors in the model. Thus, the ego vehicle is considered to be a point pk=[xk, yk] and deal with the area it occupies in the definition of the road limits and other traffic participants. For the driveable space, piecewise-linear functions (x) and (x) are defined for the left and right road boundaries respectively, such that ∀x: (x)<(x). To take the size of the ego vehicle into account, the constraint is formulated as:
d+(xk)≤yk≤(xk)−d (18)
where d:e−> is a function of the size of the ego vehicle with respect to its point estimate. In the most conservative case, assuming a rectangular ego vehicle of width w and length l, it is possible to define d=√{square root over ((w2+l2))}/2, i.e. reduce the driveable surface to =2\(out⊕e) where ⊕ is the Minkowski-sum operator. For practical purposes, d=w/2 is considered, which is exact when ϕ=0.
For traffic participants, the ellipses (aki, bki) defined in (9) are inscribed with axis-aligned rectangles, which are then augmented to consider the point estimate of the ego vehicle's pose. With dx and dy being functions of the size of the ego vehicle with respect to its center in the x and y direction, respectively, rectangles ki are defined with the limits:
It should be noted that xk,mini, xk,maxi, yk,mini, yk,maxi can be computed in closed form from (aki, bki), dx and dy.
Then, the collision avoidance constraint is the logical implication:
x
k,min
i
≤x≤x
k,max
i
{circumflex over ( )}y≥y
k,min
i
⇒y≥y
k,max
i (20)
which can be understood as “if the ego position is aligned with the vehicle along x, then it must be outside the vehicle's borders in y”. The big-M formulation [C-8], [C-9] is used to obtain the following mixed integer constraint:
for a sufficiently large M∈+.
3) Mixed Integer Linear Cost Function: For best performance, the cost function of the MILP stage 302 should be similar to the cost function from the non-linear stage 304 to minimize the gap between the optimum obtained in both stages.
If the cost functions are similar, and subject to similar constraints, then the optimal trajectory computed in the first stage 302 should approximate the final optimal trajectory computed in the second stage 304, and therefore provide a useful initialization.
The MILP problem is defined in a receding horizon formulation, in which the cost over each state (individual cost) is also defined independently in time, with:
for a set of soft constraints , where each constraint t is defined by a function Θl(
Similarly to the non-linear stage 304, soft constraints are defined over different objectives:
with the auxiliary constraints:
such that minimizing risk corresponds to minimizing the function ΘΣ=Σi=1n
4) MILP Problem Definition: With constraints and cost function , the planning problem of the first stage 302 is formulated as a MILP problem with a receding horizon of K steps.
In the end, a full trajectory
Problem 3 (Receding Horizon MILP). Given an initial ego state x0, trajectories of other traffic participants (o0:N1:n, Σ0:N1:n), a set of soft constraints , and planning step (iteration) 0≤m≤N−K of the receding horizon, compute the optimal policy:
For planning step m=0, the known initial vehicle state
In both the MILP and the NLP stages, optimization occurs in the continuous space; no spatial/grid discretization occurs at either of the stages. In particular, in the MILP stage, the state vectors of both the ego vehicles and the other actors, and control vectors are not restricted to integer values, or otherwise discretised to any fixed grid. Not discretizing over a grid is one of the advantages of the method, as it allows for smoother trajectories.
The integer variables in the MILP stage are used to enforce the constraints related to collision avoidance and driveable surface only.
The trajectory
Consistent with the above terminology, the linear cost function of Equation (22) and the linear dynamics model of Equations (15) and (16) are examples of a “preliminary” cost function and model respectively, whilst the non-linear cost function of Equation (11) and the non-linear dynamics model of Equation (5) are examples a “final” or, equivalently, “full” cost function and model respectively, where the terms “final” and “full” are, again, only used in the specific context of the two-stage optimization, and does not imply absolute finality.
It is noted that, which the above paragraph considers seed and final trajectories, it could be that the control actions ū0:N−1 of the MILP stage 302 are alternatively or additionally used to seed the NLP stage 304 (and they may be referred to as seed control actions in that event); similarly, as noted above, data of one or both of the final trajectory x0:N and the final series of control actions u0:N−1 may be provided to the controller 108 for the purpose of generating “actual” control signals for controlling the autonomous vehicle.
Whereas in the NLP stage 304, the scenario description parameters (x0, O0:n1:N, Σ0:n1:N) have a non-linear form (because they are formulated in terms of heading ϕk), in the MILP stage 302, the scenario description parameters (x0, O0:n1:N, Σ0:n1:N) have a linear form (because motion is formulated in terms of components νkx, νky instead of speed and heading νk, ϕk), with
Regarding notation,
It will be appreciated that the duration Δt between time steps in
An overtaking goal is defined by way of a suitably distant reference location (not shown) on the reference path ref, ahead of the forward vehicle 402. The effect of the progress constraints is to encourage the ego vehicle 400 to reach that reference location as soon as it can, subject to the other constraints, whilst the effect of the collision avoidance constraints is to prevent the ego vehicle 402 from pulling out until the oncoming vehicle stops being a collision risk.
On the left-hand side,
As can be seen on the left-hand side of
By contrast, as shown on the right-hand side of
5) On the optimality of the MILP formulation: The goal of the formulation of Problem 3 is to obtain an optimal solution which can be used as initialization to Problem 2 to minimize the problems of slow convergence and local optima that arise from solving the NLP directly. In this section, the optimality of the solution to Problem 3 is discussed.
A solution to Problem 3 can be obtained using Branch and Bound, a divide and conquer algorithm first introduced and applied to mixed integer linear programming by Land and Doig in [C-10]. This solution is proven to be the globally optimal one [C-7], [C-11]. In practice, modern solvers (e.g. Gurobi or CPLEX) may fail to find the global solution due to rounding errors and built-in tolerances [C-11]. On top of this, the receding horizon formulation of the problem, introduced for the sake of computational tractability, generates suboptimality by definition [C-12], [C-13]. Due to these factors, no strong theoretical guarantee can be given regarding the solution of the MILP stage. However, despite the lack of theoretical guarantees at this level, a solution that is close to the global optimum at each receding horizon step acts as a proxy towards a final solution to be obtained in the second stage 304 that is close to the global optimum and, in turn, initializing the NLP stage using this solution is expected to improve the quality of the solution at that second stage 304.
In some contexts, it may be that the speed increase provided by the two-stage approach will admit direct application on an autonomous vehicle in real-time.
However, to provide an additional speed increase, which may be needed to ensure real-time performance in a resource-constrained environment, it is also possible to implement an approximation of one or both of the first and second stages 302, 304, using a trained function approximator, such as a neural network.
Either one or both of the first and second stages 302, 304 could be replaced with a function approximator in a practical implementation.
By way of example, the following section describes, with reference to
Each training example 601 of the NLP training set is made up of an input, which in turn comprises a particular set of scenario description parameters (x0, O0:n1:N, Σ0:n1:N) together with a corresponding seed trajectory
Once trained, given such an input, i.e. a set of scenario description parameters (x0, O0:n1:N, Σ0:n1:N) and a seed trajectory
If the approximate final trajectory {tilde over (x)}0:N satisfies the trajectory verification constraints, it can be passed to the controller 108 to implement. In the case that the approximate final trajectory fails to satisfy at least one of the trajectory verification constraints, then it can modified, either at the planning level (by the planner 106) or control level (by the controller 108) so that it does. Assuming the NLP approximator 702 has been adequately trained such modifications, if required, should generally be relatively small.
The different scenarios could, for example, be simulated scenarios, generated in a simulator.
That is to say, whereas the seed trajectory
The same principles generally apply, but now a function approximator 702 is trained on the MILP training set 700 to approximate the MILP stage 602. Such a trained MILP approximator 702 can then be used in online processing to compute an approximate seed trajectory {tilde over (
Regarding notation, {tilde over (x)}0:N and {tilde over (
Note that the term “approximate” is used herein in two somewhat distinct senses. When a function approximator is used to approximate the MILP stage 302 and/or the NLP stage 304, the MILP/NLP function approximator 302/304 is said to approximate the seed trajectory
The seed trajectory
Note that, whilst the description of
The above techniques be implemented in an “onboard” or “offboard” context. One example of an offboard context would be the above training performed in an offboard computer system. The above techniques can also be implemented as part of a simulated runtime stack in order to test the performance of the runtime stack in a simulator. Simulation is an increasingly crucial component of safety and other performance testing for autonomous vehicles in particular.
Note, where a function approximation approach is used, the function approximator(s) can be tested in the simulator, in order to assess their safety and performance before they are deployed on an actual vehicle.
The term “PILOT” (Planning by Imitation Learning and Optimisation) is used herein to refer to a runtime planner, in which a trained function approximator—trained via imitation learning—seeds (initializes) a runtime optimizer. A PILOT planner could be implemented in the architecture of
As another example, a PILOT planner could be implemented the architecture of
In the examples described below, the architecture of
The same applies to the alternative architecture of
A significant benefit PILOT architecture is that, not only can it provide high-quality solutions by avoiding convergence to local optima, it can do so in real-time, even on a resource-constrained platform such as an autonomous vehicle or other mobile robot computer system.
Achieving the right balance between planning quality, safety and run-time efficiency is a major challenge for autonomous driving research. With PILOT, a neural network component efficiently imitates an expensive-to-run optimisation-based planning system to enforce a baseline of planning quality, and an efficient optimisation component with a similar objective function to the expert that refines the network output to guarantee satisfaction of requirements of safety and passenger comfort. In simulated autonomous driving experiments, it has been demonstrated that the proposed framework achieves a significant reduction in runtime when compared to the optimisation-based expert it is based on, without sacrificing the output quality.
V.1 Introduction
Guaranteeing safety of decision making is a fundamental challenge in the path towards the long-anticipated adoption of autonomous vehicle technology. Attempts to address this challenge show the diversity of possible definitions of what safety means: whether it is maintaining the autonomous system inside a safe subset of possible future states [D-1], [D-2], preventing the system from breaking domain-specific constraints [D-3], [D-4], or exhibiting behaviours that match the safe behaviour of an expert [D-5], amongst others.
Typically, model-based approaches to safety are engineering-heavy and require deep knowledge of the application domain, while, on the other hand, the hands-off aspect of the data-driven approach is lucrative, hence the growing interest in the research community in exploiting techniques like imitation learning for autonomous driving [D-6], [D-7], [D-8], [D-9]. Moreover, inference using a data-driven model is usually very efficient compared to, e.g., more elaborate search- or optimisation-based approaches.
From a different angle, model-based planning approaches give a better handle on understanding system expectations through model specification and produce more interpretable plans [D-10], [D-11], [D-12], but usually at the cost of robustness [D-13] or runtime efficiency [D-4].
A simplistic attempt to leveraging learning from data in this setting (e.g. a vanilla behavioural cloning approach to imitation learning [D-14]), will likely fail to exhibit safe behaviour at deployment time due to covariate shift between the situations in the specific training data and the deployment environment [D-15], [D-16], even with a prohibitive amount of training data. Attempts to improve deployment time performance include online approaches to training that actively enrich the training data with actual experiences of the learner in the deployment environment [D-17], and offline approaches that synthesise realistic perturbed traces from the original expert dataset to include more failure cases and near-misses [D-18].
Still, in a safety-critical application like autonomous driving, exhibiting safe behaviour is not sufficient. As pure data-driven approaches struggle to certify the safety of their output at deployment time, approaches that leverage data-driven methods along with additional components that give some guarantees on the safety of the output have emerged, e.g., using control safe sets to validate acceleration and steering commands predicted by a neural network [D-19].
The embodiments described below follow the imitation learning paradigm, but instead of depending on curated, human expert data in a learning scheme, imitate traces are generated by a performant planner (referred to as the base, reference or expert planner) that is expensive-to-run. Taken as the expert, a data-driven, imitation learning technique is utilized to produce an efficient neural planner—when compared to the expert planner—that maintains its standard of planning quality. This imitation learning paradigm takes full advantage of the base planner without introducing superficial limits to its sophistication. This is in contrast to approaches like Constrained Policy Nets [D-20] in which the cost function of an optimisation planner is made into a loss signal to train a policy network from scratch, requiring careful treatment of the constraint set.
The imitation learning approach taught herein is generally applicable, and can be used to approximate any desired reference planner. The following examples use the two-stage optimisation approach of
In this context, a benefit of the multi-stage optimization-based architecture of
The following examples use an in-the-loop DAgger [D-17] approach to imitation learning to train a deep neural network to imitate the output of the expert planner. Online augmentation using DAgger enriches the learner's dataset with relevant problem settings that might be lacking in the expert planner's dataset. This benefits from the fact that (unlike a human expert) the expert planner of
To guarantee the safety of the output of the network, a constrained optimisation step is applied, that uses a similar objective function to the expert's, to smooth and improve upon the resulting trajectory from a safety perspective. This uses the architecture of
In this context, the trajectory verification component 712 receives a trajectory from the neural network 900, which is used to seed a further non-linear optimization performed by the trajectory verification component 712. The further non-linear optimization may result in a modified trajectory that is guaranteed to satisfy whatever constraints are imposed by the trajectory verification component 712. In such implementations, the trajectory verification component 712 may be referred to as a “runtime” or “post-hoc” optimizer (to distinguish from the optimization stages 302, 304 of the expert planner).
Another benefit of imitating an optimisation-based system rather than human data is a much reduced training cost, especially with regard to updating the training dataset and labelling new instances experienced by the learner with expert output. Unlike a human, an optimization-based system can also be used to generate training data for simulated scenarios.
The performance of the present imitation learning and optimisation architecture has been evaluated on sets of simulated experiments generated using a light-weight simulator and using CARLA [D-22], and compared in terms of trajectory cost and runtime to the optimisation planner it is based on, as well as to other alternative architectures that employ the same optimisation component as PILOT.
In summary, one aspect herein is an efficient and safe planning framework for autonomous driving, comprising a deep neural network followed by a constrained optimisation stage.
Another aspect is process of imitating a model-based optimiser to improve runtime efficiency.
These aspects can be combined to provide a robust framework to imitate an expensive-to-run optimiser using a deep neural network and an efficient optimiser with the same cost function (or similar cost functions). An application of this framework to the two-stage optimization-based planner of
V.2. Pilot: Planning by Imitation Learning and Optimisation
The PILOT solution improves the efficiency of expensive-to-run optimisation-based planners. The input to the planning problem is assumed to be given by s∈d, and the goal of the planning to obtain a sequence of states
such that it optimises:
where g=(g1, . . . , gL) and h=(h1, . . . , hM) are possibly nonlinear, non-convex inequality and equality constraints on the planning states, and is a cost function defined over the plan. Whilst globally solving this optimisation problem is known to be NP-hard [D-23], [D-24], there are efficient solvers that compute local solutions within acceptable times in practice assuming a sensible initial guess is provided [D-25], [D-26]. Here, ν is defined to be an efficient optimiser that solves Eq. D-1 (e.g. optimiser in [D-25]), and Ω to be an expert expensive-to-run optimisation procedure that attempts to improve upon the local optimum of Eq. D-1 found by v. Examples of Ω can include performing a recursive decomposition of the problem and taking the minimum cost [D-27] or applying other warm-starting procedures [D-4],[D-28].
The goal of PILOT is to achieve the lower cost on provided by Ω, while approximating the efficient runtime of ν. To do so, PILOT employs an imitation learning paradigm to train a deep neural network, Θ (900,
In order to achieve that, the network 900 is pre-trained on a dataset of problems labelled by the expert, 0={(si, τi)}i=1, . . . ,n. Then, using the trained network as the planner in a simulator, a DAgger-style training loop [D-17] is employed to adapt to the covariate shift between the training dataset and the learner's experience in the simulator. See Algorithm 1. In principle, a well-trained neural network could be used at deployment time as a proxy to Ω if it produces feasible trajectories similar to the outputs of the expert in problems close to the ones in the training set. However, the raw output of a neural network is not guaranteed to satisfy solution optimality and the constraints without major investments in robust training [D-29], [D-30] or post-hoc analysis [D-31]. Instead, the neural network Θ to initialise v so as to maintain safety and smoothness guarantees. See Algorithm 2.
V.3 Pilot for the Two Stage Optimisation-Based Motion Planner
For the purpose of illustration only, the base planner is implemented using the 2s-OBT framework of
A. Two-Stage Optimisation-Based Motion Planner
In 2s-OPT, projecting the world state and road user predictions into a reference path-based coordinate frame produces the input to the optimisation. The first optimisation stage 302 solves a linearised version of the planning problem using a Mixed Integer Linear Programming (MILP) solver. This minimises a cost function that encodes desirable plan features, such as passenger comfort and progress along the reference path, while satisfying hard constraints that encode safety requirements. The output of the MILP solver is fed as a warm-start initialisation to a constrained, non-linear optimiser 304. This second optimisation stage ensures that the output trajectory is smooth and feasible, while maintaining the safety guarantees.
As discussed above, although the framework produces superior outputs when compared to alternatives with regard to solution quality (measured by convergence guarantee and output cost values), it suffers from the limitation of pure optimisation approaches in solving time, as the method effectively trades off efficiency for better solution quality.
B. Implementation Details
PILOT is used with the two stage optimisation (2s-OPT) approach of
1) System Architecture:
2) Network Input Representation:
The planning problem input comprises the static road layout, road users with their predicted trajectories, and a reference path to follow. As the problem is transformed to the reference path coordinate frame, the transformed scene is automatically aligned with the area of interest—the road along the reference path. This simplifies the representation to the neural network 900.
To encode the predicted trajectories of dynamic road users, C greyscale top-down images of the scene of size W×H are produced by uniformly sampling in time the positions of road users along their predicted trajectories at times
h for the planning horizon h.
These images create an input tensor of size C×W×H, allowing convolutional layers to be used to extract semantic features of the scene and its temporal evolution. The static layout information is present on all channels.
Additional information of the planning problem that is not visualised in the top-down images (such as the initial speed of the ego vehicle) are appended as scalar inputs, along with the flattened convolutional neural network (CNN) output, to the first dense layer of the network.
3) Network Output Representation:
The output of the network is a trajectory in the reference path coordinate frame. One possibility is to output a tensor of size 2×(h·f) for a planning horizon h and planning frequency f, encoding timestamped spatial points τ={(xj, yj)}j=1, . . . ,N. To enforce output smoothness, an alternative is to train the network to produce parameters for smooth function families, e.g. polynomials and B-splines, over time, namely fx(t) and fy(t).
The post-hoc NLP optimisation stage (detailed below) expects as input a time-stamped sequence of states, each comprising: (x,y) position, speed, orientation and control inputs (steering and acceleration), all in the reference path coordinate frame. Velocities and orientations are calculated from the sequence of points produced by the network (or sampled from the smooth function output). Control input is derived from an inverse dynamics model.
4) Neural Network Training:
a) Pre-training: The neural network is trained to imitate the output of the 2s-OPT. In a supervised learning fashion, the expert data is produced by running 2s-OPT in problems generated by simulating various configurations of driving instances to create a training dataset 0={(si, τiexp)}i=1, . . . ,n. The training loss is defined as the L2 norm distance between the expert trajectory and the network output:
where θ refers to the neural network parameter vector, is the dataset of training examples, and the identifier exp indicates an expert trajectory from the dataset. An ADAM optimiser
[32] is used to determine update step sizes.
b) DAgger training: Dataset Aggregation (DAgger) [D-17] is a training regime that reduces the difference between the distribution of the problems in the expert dataset and the distribution of problems seen by the learner when interacting with its environment. It does this by augmenting the training dataset online with additional problems generated by the pre-trained learner network when interacting with the environment in simulation. A requirement of DAgger is an interactive expert that can be queried for solutions to the new problems in a scalable way. This is satisfied in our case by the 2s-OPT which labels the new problems with high quality solutions. The DAgger process alternates between problem generation in the simulation and training the network as described in Algorithm 1.
Expanding on the benefits of DAger training, typical supervised learning problems assume the data-generation process is iid (independent and identically distributed). In the case of autonomous driving, this assumption is flawed. For effective driving, a key aspect is the accumulation of errors across time. The distribution of ‘problems’ (scenario states) seen by an agent when driving depends on that agent's previous actions. If the ego's actions lead to problems that lie outside the domain of demonstrations seen by the learner, then the learner's planning relies on generalisation.
A scenario state in this context is a snapshot of a scene at a given time instant with an ego (simulated agent or real vehicle) to plan for and all other agents' trajectories having been predicted.
As per Algorithm 2, the scheme alternates between training steps and augmentation steps. The first training set is performed on a large dataset 0 of examples obtained using the reference planner Ω. In each subsequent training step, the parameters θ are tuned via training on an augmented training set , as augmented in the previous augmentation step. In each augmentation step the (partially trained) network 900 is applied to simulated scenarios in a simulator 904, by using the partially trained network 900 to plan trajectories for a simulated ego agent (typically in response to other agent(s) in the scenario). The expert planner Ω is then applied in “shadow mode”, i.e. given snapshots (x0, O0:n1:N, Σ0:n1:N) of the scenarios encountered in the most recent set of simulations, the reference planner Ω is used to generate expert trajectories at those time instants, which in turn are used to augment the training set for the next training step. Note that the evolution of each simulated scenario is determined, at least in part, by the actions of the partially trained network 900 (for example, at some time into the scenario, x0 would generally depend on earlier actions by the partially trained network 900, as may the other parameters O0:n1:N, Σ0:n1:N e.g. in the event the other agents are reacting to the ego agent); the reference planner Ω then provides expert trajectories for those scenarios (but, unlike the network 900, does not control the ego agent or influence the simulated scenarios, because the aim is to expose the reference planner Ω to scenarios in which errors by the partially trained network 900 might have accumulated over time).
5) Post-Hoc Optimisation Stage:
In the post-hoc optimizer 712, the design of the constrained, non-linear optimisation stage of 2s-OPT is followed to smooth and rectify the trajectory generated by the imitation learning network. More specifically, a discrete bicycle model for the ego vehicle is used:
where zk=[xk, yk, ϕk, νk]T describes the state of the ego at time k with (xk, yk) being the reference-path projected position, ϕk is the vehicle's yaw angle, and νk is its speed. (ak, δk) represent the acceleration and steering controls applied to the vehicle at time k, and L is the distance between axles.
Constraints are enforced that maintain acceleration and steering values within the permitted ranges, limit maximum jerk and angular jerk, and maintain speed within the allowed speed limit for the road. Also, constraints are added that ensure that the corners of the ego vehicle's footprint are at all times bounded within the road surface, and that prevent collision with other road users.
The post-hoc optimizer 712 optimizes a cost function defined as
where ωi∈ are scalar weights, and θi(zk, uk) are soft constraints that measure deviation from the desired speed, the reference path and the end target location, and that control the norms of acceleration and steering control inputs. Parameters of the optimisation are fine-tuned using grid-search in the parameter space. See Annex B for a more detailed formulation of the optimisation problem.
The output of the imitation learning neural network is used as an initialisation to this optimisation problem. As the NLP solver converges faster when initialised with a feasible solution, as demonstrated in Annex A, the initialisation trajectory is processed to cap abnormal values of calculated speed, acceleration and steering that might break some of the hard constraints. The process is detailed in Annex C.
The described approach poses no restrictions on the design of the optimisation objective that the planner output is desired to respect, while in settings such as [D-20] the optimisation problem should be carefully designed to ensure that the optimisation constraints are differentiable in order for them to be usable to train the planner network.
In general, a function approximator takes the form of one or more trainable models than can be trained to imitate an expert. Those models can take any form, including but not limited to neural network(s). The expert may be a computer-implemented planner (not necessarily required to operate in real-time), as in the above examples. In that case, the one or more models are chosen so that they can be applied with fewer computing resources or memory resources than the expert planner (preferably in real-time). This generally means trading efficiency (time and/or memory efficiency) for accuracy (i.e. the accuracy with which the function approximator is able to imitate the expert)—in the embodiments described above, the trade of is specifically that the trained function approximator may no longer be guaranteed to satisfy the hard constraints on the NLP stage. However, this issue is addressed by implementing the NLP logic at runtime, to refine the initial trajectory produced by the function approximator, and in this context it is not necessary for the trained function approximator to perfectly imitate the expert.
As will be appreciated, whilst the above considers a particular form of multistage optimization-based planner as an expert to be imitated, the function approximator can be trained to implement other types of planner. There are particular benefits when the expert planner has the ability to produce high-quality trajectories, but requires significant computational and/or memory resources to do so. It also feasible to use a human expert(s) to provide at least some of the training data for the function approximator.
References herein to components, functions, modules and the like, including the optimizers of the first and second stages 302, 304, the function approximators 702, 604 and 900, and the verification component 712, denote functional components of a computer system which may be implemented at the hardware level in various ways. A computer system comprises one or more computers that may be programmable or non-programmable. A computer comprises one or more processors which carry out the functionality of the aforementioned functional components. A processor can take the form of a general-purpose processor such as a CPU (Central Processing unit) or accelerator (e.g. GPU) etc. or more specialized form of hardware processor such as an FPGA (Field Programmable Gate Array) or ASIC (Application-Specific Integrated Circuit). That is, a processor may be programmable (e.g. an instruction-based general-purpose processor, FPGA etc.) or non-programmable (e.g. an ASIC). A computer system may be implemented “onboard” a mobile robot (such as an AV) or “offboard” for example in a simulator context.
For example, an offboard implementation of the full multi-stage optimization could be implemented for the purpose of training on-board components, but also for other purposes, such as safety testing or other performance testing, verification etc. Performance testing could involve implementing any of the above trained function approximators 702, 604, 900 and (where applicable) the verification component 712 in an offboard computer system as part of the performance testing, applied to simulated scenarios.
In an on-board implementation, the multi-phase optimization may be implemented in full, or one or more (including all) optimization stages may be replaced with one or more function approximators.
Annex E documents experimental results that demonstrate the efficacy of the PILOT architecture (as described in Section V.).
Results are given in Annex A to demonstrate the efficacy of the 2s-OPT of Section IV.
Annex A—Results and Evaluation for 2s-OPT Planner
The 2s-OPT methodology of section IV. has been subject to empirical testing to further demonstrate is efficacy.
Reference is made to
The bottom half of
In this section we will show that:
1) our method is general and can be applied to a diversity of driving situations
2) the MILP stage provides the NLP stage with a better initialization when compared to simpler heuristics, leading to higher convergence rates, faster solving times and more optimal solutions;
3) our method leads to solutions that outperform a Nonlinear Model Predictive Control (NMPC) approach similar to the one presented in [E-25] in progress and comfort metrics.
To that end, we implement the first stage (MILP; Problem 3) using Gurobi 8.1 [E-15], and the second stage (NLP; Problem 2) using IPOPT [29]. Both solvers have a timeout of 25s, after which the optimization is stopped. We use N=40 and Δt=0.2s for a trajectory horizon of 8s (Other parameters are listed in Appendix A). Without loss of generality, we assume left-hand traffic where drivers are expected to be on the left hand side of the road, that a route planner is available to generate a reference path the satisfies the planning goal in the local coordinate frame, and consider a constant velocity model for the prediction of dynamic agents. In the simulator, the behavior of other dynamic vehicles is based on the Intelligent Driver Model [E-27].
VI.1 Generality:
VI.2 Initialization Comparison:
We consider four heuristic initializations as alternatives to our MILP stage: ZEROS, in which all states and controls are initialized to zero; C.VEL, where the ego-vehicle is assumed to maintain its speed throughout the solution; C.ACC, where the ego-vehicle maintains a constant acceleration of 1 ms−2 until the maximum speed is achieved; and C.DEC where the ego-vehicle maintains a constant acceleration of −1 ms−2 until it stops.
A dataset of 1000 examples per scenario was procedurally generated, solved using our method as well as suing only the NLP stage for the same constrains initialized by each of the four heuristics. Table D-I presents the percentage of solved examples within the allocated time (25s) per initialization method and scenario. As it can be observed, our method either performs similarly or outperforms other methods in all the scenarios considered. In
We see that, in general, other initialization methods obtain higher costs on average than ours. While C.ACC is able to achieve a lower average cost in some scenarios, it leads to a substantial number of unsolved examples, particularly for DO and DO+OV (see
VI.3 NMPC Baseline Comparison:
We compare our method to an NMPC baseline which optimizes Problem 2 with the same soft and hard constraint but in a receding-horizon fashion similar to [25]. This baseline uses the same interior-point optimization method as our NLP stage to solve each of the horizon steps. To evaluate the quality of the methods, we introduce the following metrics that compare our solution, (x1:NOurs, u1:N−1Ours), to the baseline's, (x1:NNMPC, u1:N−1NMPC):
ΔP=max(x1:NOurs)−max(x1:NNMPC)
We generate 1000 examples per scenario and solve them using our method and the NMPC baseline. The top of Table III presents the percentage of the examples that are solved by both methods. In all scenarios, our method solves more examples than the baseline. For the problems that are solved by both we show the metrics per scenario in the bottom of Table III. We achieve significantly higher progress and better velocity matching across scenarios, and similar or slightly smaller jerk values and deviation from the reference path. These results validate our claim that our method is better in progress and comfort metrics than the baseline that has a similar formulation to [E-25].
A. Optimization Parameters
The parameters used in the solving of Problem 2 and 3 in the context of Sec. VI are defined in Table A1
B. Generalization of Scenario Examples
For the randomly generated scenarios presented in Sec. V, we assume the ego vehicle has length 4.8 m and width 1.9 m. the example used in this work were preceduirally generated by uniform sampling the parameters of the scenarios, following the ranges defined in Tables B1, B2, B3, B4, and B5.
Annex B. Nonlinear Programming Problem Formulation
For initial ego-vehicle state z0, predictions of other road users (o0:N1:n, Σ0:N1:n) and the cost function J in (5), compute:
under a discrete dynamical system (4) for timestep k∈{0, . . . , N}, and the set of constraints:
The parameters of the optimisation are detailed in Table D-III
Annex C. Output Transformation Checks
The network produces a sequence of spatial positions, and the rest of the required input of the optimiser are computed from that sequence. A number of checks of upper and lower limits are applied to tame abnormalities in the network output and to improve the input to the optimiser.
Annex D. CARLA Dataset
We used Town02 and Town01 for generating a training dataset and a benchmarking dataset, respectively. To create the planning problems, we ran a planner to guide the ego vehicle in CARLA simulations and collected problem instances at a rate of 1.5 Hz in runs of 12 seconds. We added a maximum of 40 other vehicles (static or driven by CARLA's Autopilot) to each scenes, spawned randomly in each simulation. The training dataset has 20604 problems, and the benchmarking dataset has 1000 problems. We solved all the problems using 2s-OPT for the baseline.
Annex E: PILOT Experiments
We show results of PILOT compared to 2s-OPT; the two stage optimisation planner it was trained to imitate. Also, we present an ablation study which compares the imitation learning network output as an initialisation to the NLP solver compared to alternatives.
A. Setup
We follow the implementation in [D-4] for 2s-OPT, and we base the design of the post-hoc optimisation stage on the second stage of 2s-OPT using IPOPT [D-25] as a solver.
The first experiment uses a training dataset of 70,000 procedurally-generated problems of two-lane, urban driving in a residential setting. The problems are generated from 33 manually-designed base scenarios that contain a mixture of static and moving vehicles. The ego vehicle's position, heading and speed are perturbed at different timesteps of the original scenarios to create new configurations. Each of these configurations is considered a planning problem and solved by 2s-OPT, creating a high-quality expert dataset. We use an 80-20 split for training and testing datasets, respectively.
After initial training of the neural network, the training dataset is augmented with new problems generated interactively by the trained network when used in a driving simulator on new problems similar to the ones in the dataset. The new problems are solved by 2s-OPT as well to provide the expert trajectories. In a DAgger fashion, we generate 64 new problems and add them to the training dataset every training epoch.
The second experiment uses the CARLA simulator [D-22] where we generate 20,000 planning instances by running a baseline planner in randomly-generated scenarios in Town02, and compute their 2s-OPT solutions, to train PILOT. Then we run DAgger training on Town02 using the CARLA simulator and CARLA's Autopilot to control all non-ego agents. Lastly, we benchmark PILOT on a dataset generated in Town01 using 2s-OPT. More details about the process are in Appendix D.
B. Results
We report results on the quality of PILOT when compared to 2s-OPT using two metrics:
We report these metrics for the procedurally-generated experiment and for the CARLA experiment by sampling a benchmarking 1000 problems from the respective test dataset and computing the two metrics for PILOT and 2s-OPT in problems solved by both methods. These problems are not seen by the neural network at training time. The results for the procedurally-generated experiment are shown in Table D-I(a), and the results for the CARLA experiment are shown in Table D-I(b).
These results vindicate our approach of combining an imitation learning with an optimisation stage, resulting in an efficient approach to planning safe and comfortable trajectories. As the results show, time efficiency has a clear advantage (saving of ˜78% and 84% respectively) using PILOT with no significant deterioration of the solution quality (drop of ˜3% and 4% respectively). By testing in a different town from the training set in the CARLA experiment, we show our framework has at least reasonably generalised to this new environment.
Next, we present an ablation study on the quality of initialising the NLP stage using the output of the imitation learning network compared to alternatives. We use the same benchmarking problems from the procedurally-generated experiment above. The alternative initialisations we compare against include: None initialisation which sets x, y, v and θ to zero at all timesteps; ConstVel a constant velocity initialisation that keeps θ constant while the vehicle moves with a constant speed; and ConstAccel/ConstDecel constant acceleration and deceleration initialisations are similar but the speed is changed with a constant rate until it reaches the allowed speed limit or 0, respectively.
We compare these alternatives relative to the original MILP initialisation of the 2s-OPT framework, as a baseline. We use three metrics:
The results are shown in Table D-II.
As the results show, PILOT's neural network initialisation produces trajectories that are easier to optimise (as reflected in the reduction in NLP solving time) with a small increase in the final cost in average when compared to the MILP initialisation of 2s-OPT. One of the alternatives (ConstAccel) has a slight advantage in final NLP cost, but it takes more time when it converges and it fails to converge in 9% more examples to a solution.
Reference is also made in the above to the following, each of which is incorporated herein by reference in its entirety:
Number | Date | Country | Kind |
---|---|---|---|
2001200.1 | Jan 2020 | GB | national |
2001202.7 | Jan 2020 | GB | national |
2001277.9 | Jan 2020 | GB | national |
2017252.4 | Oct 2020 | GB | national |
Filing Document | Filing Date | Country | Kind |
---|---|---|---|
PCT/EP2021/052036 | 1/28/2021 | WO |