DiffuSolve: Diffusion-based Solver for Non-convex Trajectory Optimization

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Li, Anjian, Ding, Zihan, Dieng, Adji Bousso, Beeson, Ryne
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866910631665336320
author Li, Anjian
Ding, Zihan
Dieng, Adji Bousso
Beeson, Ryne
author_facet Li, Anjian
Ding, Zihan
Dieng, Adji Bousso
Beeson, Ryne
contents Optimal trajectory design is computationally expensive for nonlinear and high-dimensional dynamical systems. The challenge arises from the non-convex nature of the optimization problem with multiple local optima, which usually requires a global search. Traditional numerical solvers struggle to find diverse solutions efficiently without appropriate initial guesses. In this paper, we introduce DiffuSolve, a general diffusion model-based solver for non-convex trajectory optimization. An expressive diffusion model is trained on pre-collected locally optimal solutions and efficiently samples initial guesses, which then warm-starts numerical solvers to fine-tune the feasibility and optimality. We also present DiffuSolve+, a novel constrained diffusion model with an additional loss in training that further reduces the problem constraint violations of diffusion samples. Experimental evaluations on three tasks verify the improved robustness, diversity, and a 2$\times$ to 11$\times$ increase in computational efficiency with our proposed method, which generalizes well to trajectory optimization problems of varying challenges.
format Preprint
id arxiv_https___arxiv_org_abs_2403_05571
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle DiffuSolve: Diffusion-based Solver for Non-convex Trajectory Optimization
Li, Anjian
Ding, Zihan
Dieng, Adji Bousso
Beeson, Ryne
Robotics
Machine Learning
Optimal trajectory design is computationally expensive for nonlinear and high-dimensional dynamical systems. The challenge arises from the non-convex nature of the optimization problem with multiple local optima, which usually requires a global search. Traditional numerical solvers struggle to find diverse solutions efficiently without appropriate initial guesses. In this paper, we introduce DiffuSolve, a general diffusion model-based solver for non-convex trajectory optimization. An expressive diffusion model is trained on pre-collected locally optimal solutions and efficiently samples initial guesses, which then warm-starts numerical solvers to fine-tune the feasibility and optimality. We also present DiffuSolve+, a novel constrained diffusion model with an additional loss in training that further reduces the problem constraint violations of diffusion samples. Experimental evaluations on three tasks verify the improved robustness, diversity, and a 2$\times$ to 11$\times$ increase in computational efficiency with our proposed method, which generalizes well to trajectory optimization problems of varying challenges.
title DiffuSolve: Diffusion-based Solver for Non-convex Trajectory Optimization
topic Robotics
Machine Learning
url https://arxiv.org/abs/2403.05571