학술논문

Neural Schrödinger Bridge With Sinkhorn Losses: Application to Data-Driven Minimum Effort Control of Colloidal Self-Assembly
Document Type
Periodical
Source
IEEE Transactions on Control Systems Technology IEEE Trans. Contr. Syst. Technol. Control Systems Technology, IEEE Transactions on. 32(3):960-973 May, 2024
Subject
Signal Processing and Analysis
Communication, Networking and Broadcast Technologies
Computing and Processing
Robotics and Control Systems
Stochastic processes
Artificial neural networks
Optimal control
Bridges
Mathematical models
Standards
Self-assembly
Colloidal self-assembly (SA)
physics-informed neural networks (PINNs)
Schrödinger bridge
Sinkhorn loss
stochastic optimal control
Language
ISSN
1063-6536
1558-0865
2374-0159
Abstract
We show that the minimum effort control of colloidal self-assembly (SA) can be naturally formulated in the order-parameter space as a generalized Schrödinger bridge problem (GSBP)—a class of fixed-horizon stochastic optimal control problems that originated in the works of Erwin Schrödinger in the early 1930s. In recent years, this class of problems has seen a resurgence of research activities in the control and machine-learning communities. Different from the existing literature on the theory and computation for such problems, the controlled drift and diffusion coefficients for colloidal SA are typically nonaffine in control and are difficult to obtain from physics-based modeling. We deduce the conditions of optimality for such generalized problems and show that the resulting system of equations is structurally very different from the existing results in a way that standard computational approaches no longer apply. Thus motivated, we propose a data-driven learning and control framework, named “neural Schrödinger bridge,” to solve such generalized Schrödinger bridge problems by innovating on recent advances in neural networks (NNs). We illustrate the effectiveness of the proposed framework using a numerical case study of colloidal SA. We learn the controlled drift and diffusion coefficients as two NNs using molecular dynamics (MD) simulation data and then use these two to train a third network with Sinkhorn losses designed for distributional endpoint constraints, specific for this class of control problems.