Physics-Informed Representation and Learning: Control and Risk Quantification
Zhuoyuan Wang, Reece Keller, Xiyu Deng, Kenta Hoshino, Takashi Tanaka, Yorie Nakahira
TL;DR
This work tackles efficient stochastic optimal and safety-critical control for high-dimensional systems by combining a comparison-theorem–driven dimensionality reduction with physics-informed neural networks to solve reduced PDEs for the value function and long-horizon safety probability. The key idea is to identify low-dimensional feature coordinates that exactly capture the evolution of the cost and barrier quantities, transform the problem into low-dimensional PDEs via the Feynman-Kac framework, and solve these PDEs with PINNs that integrate data and physics. A dedicated autoencoder-like network automatically discovers meaningful features that satisfy the required bounding assumptions, enabling scalable generalization to unseen regions and longer horizons. Empirical results on synthetic high-dimensional systems show substantially better sample efficiency than path-integral Monte Carlo and successful feature learning, underscoring the approach’s potential for real-world high-dimensional control and risk quantification tasks.
Abstract
Optimal and safety-critical control are fundamental problems for stochastic systems, and are widely considered in real-world scenarios such as robotic manipulation and autonomous driving. In this paper, we consider the problem of efficiently finding optimal and safe control for high-dimensional systems. Specifically, we propose to use dimensionality reduction techniques from a comparison theorem for stochastic differential equations together with a generalizable physics-informed neural network to estimate the optimal value function and the safety probability of the system. The proposed framework results in substantial sample efficiency improvement compared to existing methods. We further develop an autoencoder-like neural network to automatically identify the low-dimensional features of the system to enhance the ease of design for system integration. We also provide experiments and quantitative analysis to validate the efficacy of the proposed method. Source code is available at https://github.com/jacobwang925/path-integral-PINN.
