Current browse context:
cs.LG
Change to browse by:
References & Citations
Computer Science > Machine Learning
Title: The Mori-Zwanzig formulation of deep learning
(Submitted on 12 Sep 2022 (v1), last revised 19 May 2023 (this version, v4))
Abstract: We develop a new formulation of deep learning based on the Mori-Zwanzig (MZ) formalism of irreversible statistical mechanics. The new formulation is built upon the well-known duality between deep neural networks and discrete dynamical systems, and it allows us to directly propagate quantities of interest (conditional expectations and probability density functions) forward and backward through the network by means of exact linear operator equations. Such new equations can be used as a starting point to develop new effective parameterizations of deep neural networks, and provide a new framework to study deep-learning via operator theoretic methods. The proposed MZ formulation of deep learning naturally introduces a new concept, i.e., the memory of the neural network, which plays a fundamental role in low-dimensional modeling and parameterization. By using the theory of contraction mappings, we develop sufficient conditions for the memory of the neural network to decay with the number of layers. This allows us to rigorously transform deep networks into shallow ones, e.g., by reducing the number of neurons per layer (using projection operators), or by reducing the total number of layers (using the decay property of the memory operator).
Submission history
From: Daniele Venturi [view email][v1] Mon, 12 Sep 2022 18:44:50 GMT (814kb,D)
[v2] Thu, 15 Sep 2022 22:52:19 GMT (813kb,D)
[v3] Sat, 18 Feb 2023 20:44:27 GMT (1600kb,D)
[v4] Fri, 19 May 2023 18:49:12 GMT (1606kb,D)
Link back to: arXiv, form interface, contact.