We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: DDPNOpt: Differential Dynamic Programming Neural Optimizer

Abstract: Interpretation of Deep Neural Networks (DNNs) training as an optimal control problem with nonlinear dynamical systems has received considerable attention recently, yet the algorithmic development remains relatively limited. In this work, we make an attempt along this line by reformulating the training procedure from the trajectory optimization perspective. We first show that most widely-used algorithms for training DNNs can be linked to the Differential Dynamic Programming (DDP), a celebrated second-order method rooted in the Approximate Dynamic Programming. In this vein, we propose a new class of optimizer, DDP Neural Optimizer (DDPNOpt), for training feedforward and convolution networks. DDPNOpt features layer-wise feedback policies which improve convergence and reduce sensitivity to hyper-parameter over existing methods. It outperforms other optimal-control inspired training methods in both convergence and complexity, and is competitive against state-of-the-art first and second order methods. We also observe DDPNOpt has surprising benefit in preventing gradient vanishing. Our work opens up new avenues for principled algorithmic design built upon the optimal control theory.
Comments: Accepted in International Conference on Learning Representations (ICLR) 2021 as Spotlight
Subjects: Machine Learning (cs.LG); Neural and Evolutionary Computing (cs.NE); Optimization and Control (math.OC)
Journal reference: 10th International Conference on Learning Representations (ICLR 2021)
Cite as: arXiv:2002.08809 [cs.LG]
  (or arXiv:2002.08809v3 [cs.LG] for this version)

Submission history

From: Guan-Horng Liu [view email]
[v1] Thu, 20 Feb 2020 15:42:15 GMT (4055kb,D)
[v2] Mon, 29 Jun 2020 16:51:50 GMT (5172kb,D)
[v3] Sat, 8 May 2021 21:47:35 GMT (3902kb,D)

Link back to: arXiv, form interface, contact.