-
Constraint-Aware Diffusion Models for Trajectory Optimization
Authors:
Anjian Li,
Zihan Ding,
Adji Bousso Dieng,
Ryne Beeson
Abstract:
The diffusion model has shown success in generating high-quality and diverse solutions to trajectory optimization problems. However, diffusion models with neural networks inevitably make prediction errors, which leads to constraint violations such as unmet goals or collisions. This paper presents a novel constraint-aware diffusion model for trajectory optimization. We introduce a novel hybrid loss…
▽ More
The diffusion model has shown success in generating high-quality and diverse solutions to trajectory optimization problems. However, diffusion models with neural networks inevitably make prediction errors, which leads to constraint violations such as unmet goals or collisions. This paper presents a novel constraint-aware diffusion model for trajectory optimization. We introduce a novel hybrid loss function for training that minimizes the constraint violation of diffusion samples compared to the groundtruth while recovering the original data distribution. Our model is demonstrated on tabletop manipulation and two-car reach-avoid problems, outperforming traditional diffusion models in minimizing constraint violations while generating samples close to locally optimal solutions.
△ Less
Submitted 3 June, 2024;
originally announced June 2024.
-
Combining Constrained Diffusion Models and Numerical Solvers for Efficient and Robust Non-Convex Trajectory Optimization
Authors:
Anjian Li,
Zihan Ding,
Adji Bousso Dieng,
Ryne Beeson
Abstract:
Motivated by the need to solve open-loop optimal control problems with computational efficiency and reliable constraint satisfaction, we introduce a general framework that combines diffusion models and numerical optimization solvers. Optimal control problems are rarely solvable in closed form, hence they are often transcribed into numerical trajectory optimization problems, which then require init…
▽ More
Motivated by the need to solve open-loop optimal control problems with computational efficiency and reliable constraint satisfaction, we introduce a general framework that combines diffusion models and numerical optimization solvers. Optimal control problems are rarely solvable in closed form, hence they are often transcribed into numerical trajectory optimization problems, which then require initial guesses. These initial guesses are supplied in our framework by diffusion models. To mitigate the effect of samples that violate the problem constraints, we develop a novel constrained diffusion model to approximate the true distribution of locally optimal solutions with an additional constraint violation loss in training. To further enhance the robustness, the diffusion samples as initial guesses are fed to the numerical solver to refine and derive final optimal (and hence feasible) solutions. Experimental evaluations on three tasks verify the improved constraint satisfaction and computational efficiency with 4$\times$ to 30$\times$ acceleration using our proposed framework, which generalizes across trajectory optimization problems and scales well with problem complexity.
△ Less
Submitted 26 May, 2024; v1 submitted 21 February, 2024;
originally announced March 2024.
-
Amortized Global Search for Efficient Preliminary Trajectory Design with Deep Generative Models
Authors:
Anjian Li,
Amlan Sinha,
Ryne Beeson
Abstract:
Preliminary trajectory design is a global search problem that seeks multiple qualitatively different solutions to a trajectory optimization problem. Due to its high dimensionality and non-convexity, and the frequent adjustment of problem parameters, the global search becomes computationally demanding. In this paper, we exploit the clustering structure in the solutions and propose an amortized glob…
▽ More
Preliminary trajectory design is a global search problem that seeks multiple qualitatively different solutions to a trajectory optimization problem. Due to its high dimensionality and non-convexity, and the frequent adjustment of problem parameters, the global search becomes computationally demanding. In this paper, we exploit the clustering structure in the solutions and propose an amortized global search (AmorGS) framework. We use deep generative models to predict trajectory solutions that share similar structures with previously solved problems, which accelerates the global search for unseen parameter values. Our method is evaluated using De Jong's 5th function and a low-thrust circular restricted three-body problem.
△ Less
Submitted 7 August, 2023;
originally announced August 2023.
-
Quantitative Convergence of the Filter Solution for Multiple Timescale Nonlinear Systems with Coarse-Grain Correlated Noise
Authors:
Ryne Beeson,
N. Sri Namachchivaya,
Nicolas Perkowski
Abstract:
In this paper we prove a rate of convergence for the continuous time filtering solution of a multiple timescale correlated nonlinear system to a lower dimensional filtering equation in the limit of large timescale separation. Correlation is assumed to occur between the slow signal and observation processes. Convergence is almost sure in the weak topology. An asymptotic expansion of the dual proces…
▽ More
In this paper we prove a rate of convergence for the continuous time filtering solution of a multiple timescale correlated nonlinear system to a lower dimensional filtering equation in the limit of large timescale separation. Correlation is assumed to occur between the slow signal and observation processes. Convergence is almost sure in the weak topology. An asymptotic expansion of the dual process for the solution to the Zakai equation, and probabilistic representation using backward doubly stochastic differential equations is leveraged to prove the result.
△ Less
Submitted 17 January, 2022; v1 submitted 24 November, 2020;
originally announced November 2020.
-
Approximation of the Filter Equation for Multiple Timescale, Correlated, Nonlinear Systems
Authors:
Ryne Beeson,
N. Sri Namachchivaya,
Nicolas Perkowski
Abstract:
This paper considers the approximation of the continuous time filtering equation for the case of a multiple timescale (slow-intermediate, and fast scales) that may have correlation between the slow-intermediate process and the observation process. The signal process is considered fully coupled, taking values in $\mathbb{R^{m}} \times \mathbb{R^{n}}$ and without periodicity assumptions on coefficie…
▽ More
This paper considers the approximation of the continuous time filtering equation for the case of a multiple timescale (slow-intermediate, and fast scales) that may have correlation between the slow-intermediate process and the observation process. The signal process is considered fully coupled, taking values in $\mathbb{R^{m}} \times \mathbb{R^{n}}$ and without periodicity assumptions on coefficients. It is proved that in the weak topology, the solution of the filtering equation converges in probability to a solution of a lower dimensional averaged filtering equation in the limit of large timescale separation. The method of proof uses the perturbed test function approach (method of corrector) to handle the intermediate timescale in showing tightness and characterization of limits. The correctors are solutions of Poisson equations.
△ Less
Submitted 30 October, 2020;
originally announced October 2020.