-
Plasma Surrogate Modelling using Fourier Neural Operators
Authors:
Vignesh Gopakumar,
Stanislas Pamela,
Lorenzo Zanisi,
Zongyi Li,
Ander Gray,
Daniel Brennand,
Nitesh Bhatia,
Gregory Stathopoulos,
Matt Kusner,
Marc Peter Deisenroth,
Anima Anandkumar,
JOREK Team,
MAST Team
Abstract:
Predicting plasma evolution within a Tokamak reactor is crucial to realizing the goal of sustainable fusion. Capabilities in forecasting the spatio-temporal evolution of plasma rapidly and accurately allow us to quickly iterate over design and control strategies on current Tokamak devices and future reactors. Modelling plasma evolution using numerical solvers is often expensive, consuming many hou…
▽ More
Predicting plasma evolution within a Tokamak reactor is crucial to realizing the goal of sustainable fusion. Capabilities in forecasting the spatio-temporal evolution of plasma rapidly and accurately allow us to quickly iterate over design and control strategies on current Tokamak devices and future reactors. Modelling plasma evolution using numerical solvers is often expensive, consuming many hours on supercomputers, and hence, we need alternative inexpensive surrogate models. We demonstrate accurate predictions of plasma evolution both in simulation and experimental domains using deep learning-based surrogate modelling tools, viz., Fourier Neural Operators (FNO). We show that FNO has a speedup of six orders of magnitude over traditional solvers in predicting the plasma dynamics simulated from magnetohydrodynamic models, while maintaining a high accuracy (MSE in the normalised domain $\approx$ $10^{-5}$). Our modified version of the FNO is capable of solving multi-variable Partial Differential Equations (PDE), and can capture the dependence among the different variables in a single model. FNOs can also predict plasma evolution on real-world experimental data observed by the cameras positioned within the MAST Tokamak, i.e., cameras looking across the central solenoid and the divertor in the Tokamak. We show that FNOs are able to accurately forecast the evolution of plasma and have the potential to be deployed for real-time monitoring. We also illustrate their capability in forecasting the plasma shape, the locations of interactions of the plasma with the central solenoid and the divertor for the full (available) duration of the plasma shot within MAST. The FNO offers a viable alternative for surrogate modelling as it is quick to train and infer, and requires fewer data points, while being able to do zero-shot super-resolution and getting high-fidelity solutions.
△ Less
Submitted 18 June, 2024; v1 submitted 10 November, 2023;
originally announced November 2023.
-
Communication reduction in distributed optimization via estimation of the proximal operator
Authors:
Giorgos Stathopoulos,
Colin N. Jones
Abstract:
We introduce a reduced-communication distributed optimization scheme based on estimating the solution to a proximal minimization problem. Our proposed setup involves a group of agents coordinated by a central entity, altogether operating in a collaborative framework. The agents solve proximal minimization problems that are hidden from the central coordinator. The scheme enables the coordinator to…
▽ More
We introduce a reduced-communication distributed optimization scheme based on estimating the solution to a proximal minimization problem. Our proposed setup involves a group of agents coordinated by a central entity, altogether operating in a collaborative framework. The agents solve proximal minimization problems that are hidden from the central coordinator. The scheme enables the coordinator to construct a convex set within which the agents' optimizers reside, and to iteratively refine the set every time that an agent is queried. We analyze the quality of the constructed sets by showing their connections to the ε-subdifferential of a convex function and characterize their size. We prove convergence results related to the solution of such distributed optimization problems and we devise a communication criterion that embeds the proposed scheme in the Alternating Direction Method of Multipliers (ADMM). The developed scheme demonstrates significant communication reduction when applied to a microgrid setting.
△ Less
Submitted 7 July, 2018; v1 submitted 19 March, 2018;
originally announced March 2018.
-
An Inertial Parallel and Asynchronous Fixed-Point Iteration for Convex Optimization
Authors:
Giorgos Stathopoulos,
Colin N. Jones
Abstract:
Two characteristics that make convex decomposition algorithms attractive are simplicity of operations and generation of parallelizable structures. In principle, these schemes require that all coordinates update at the same time, i.e., they are synchronous by construction. Introducing asynchronicity in the updates can resolve several issues that appear in the synchronous case, like load imbalances…
▽ More
Two characteristics that make convex decomposition algorithms attractive are simplicity of operations and generation of parallelizable structures. In principle, these schemes require that all coordinates update at the same time, i.e., they are synchronous by construction. Introducing asynchronicity in the updates can resolve several issues that appear in the synchronous case, like load imbalances in the computations or failing communication links. However, and to the best of our knowledge, there are no instances of asynchronous versions of commonly-known algorithms combined with inertial acceleration techniques. In this work we propose an inertial asynchronous and parallel fixed-point iteration from which several new versions of existing convex optimization algorithms emanate. Departing from the norm that the frequency of the coordinates' updates should comply to some prior distribution, we propose a scheme where the only requirement is that the coordinates update within a bounded interval. We prove convergence of the sequence of iterates generated by the scheme at a linear rate. One instance of the proposed scheme is implemented to solve a distributed optimization load sharing problem in a smart grid setting and its superiority with respect to the non-accelerated version is illustrated.
△ Less
Submitted 6 March, 2018; v1 submitted 31 May, 2017;
originally announced June 2017.
-
Constrained LQR Using Online Decomposition Techniques
Authors:
L. Ferranti,
G. Stathopoulos,
C. N. Jones,
T. Keviczky
Abstract:
This paper presents an algorithm to solve the infinite horizon constrained linear quadratic regulator (CLQR) problem using operator splitting methods. First, the CLQR problem is reformulated as a (finite-time) model predictive control (MPC) problem without terminal constraints. Second, the MPC problem is decomposed into smaller subproblems of fixed dimension independent of the horizon length. Thir…
▽ More
This paper presents an algorithm to solve the infinite horizon constrained linear quadratic regulator (CLQR) problem using operator splitting methods. First, the CLQR problem is reformulated as a (finite-time) model predictive control (MPC) problem without terminal constraints. Second, the MPC problem is decomposed into smaller subproblems of fixed dimension independent of the horizon length. Third, using the fast alternating minimization algorithm to solve the subproblems, the horizon length is estimated online, by adding or removing subproblems based on a periodic check on the state of the last subproblem to determine whether it belongs to a given control invariant set. We show that the estimated horizon length is bounded and that the control sequence computed using the proposed algorithm is an optimal solution of the CLQR problem. Compared to state-of-the-art algorithms proposed to solve the CLQR problem, our design solves at each iteration only unconstrained least-squares problems and simple gradient calculations. Furthermore, our technique allows the horizon length to decrease online (a useful feature if the initial guess on the horizon is too conservative). Numerical results on a planar system show the potential of our algorithm.
△ Less
Submitted 19 September, 2016;
originally announced September 2016.
-
Solving the Infinite-horizon Constrained LQR Problem using Accelerated Dual Proximal Methods
Authors:
Giorgos Stathopoulos,
Milan Korda,
Colin N. Jones
Abstract:
This work presents an algorithmic scheme for solving the infinite-time constrained linear quadratic regulation problem. We employ an accelerated version of a popular proximal gradient scheme, commonly known as the Forward-Backward Splitting (FBS), and prove its convergence to the optimal solution in our infinite-dimensional setting. Each iteration of the algorithm requires only finite memory, is c…
▽ More
This work presents an algorithmic scheme for solving the infinite-time constrained linear quadratic regulation problem. We employ an accelerated version of a popular proximal gradient scheme, commonly known as the Forward-Backward Splitting (FBS), and prove its convergence to the optimal solution in our infinite-dimensional setting. Each iteration of the algorithm requires only finite memory, is computationally cheap, and makes no use of terminal invariant sets; hence, the algorithm can be applied to systems of very large dimensions. The acceleration brings in optimal convergence rates O(1/k^2) for function values and O(1/k) for primal iterates and renders the proposed method a practical alternative to model predictive control schemes for setpoint tracking. In addition, for the case when the true system is subject to disturbances or modelling errors, we propose an efficient warm-starting procedure, which significantly reduces the number of iterations when the algorithm is applied in closed-loop. Numerical examples demonstrate the approach.
△ Less
Submitted 18 January, 2015;
originally announced January 2015.
-
Guaranteeing Input Tracking For Constrained Systems: Theory and Application to Demand Response
Authors:
Tomasz T. Gorecki,
Altuğ Bitlislioğlu,
Giorgos Stathopoulos,
Colin N. Jones
Abstract:
A method for certifying exact input trackability for constrained discrete time linear systems is introduced in this paper. A signal is assumed to be drawn from a reference set and the system must track this signal with a linear combination of its inputs. Using methods inspired from robust model predictive control, the proposed approach certifies the ability of a system to track any reference drawn…
▽ More
A method for certifying exact input trackability for constrained discrete time linear systems is introduced in this paper. A signal is assumed to be drawn from a reference set and the system must track this signal with a linear combination of its inputs. Using methods inspired from robust model predictive control, the proposed approach certifies the ability of a system to track any reference drawn from a polytopic set on a finite time horizon by solving a linear program. Optimization over a parameterization of the set of reference signals is discussed, and particular instances of parameterization of this set that result in a convex program are identified, allowing one to find the largest set of trackable signals of some class. Infinite horizon feasibility of the methods proposed is obtained through use of invariant sets, and an implicit description of such an invariant set is proposed. These results are tailored for the application of power consumption tracking for loads, where the operator of the load needs to certify in advance his ability to fulfill some requirement set by the network operator. An example of a building heating system illustrates the results.
△ Less
Submitted 20 April, 2015; v1 submitted 27 October, 2014;
originally announced October 2014.
-
A hierarchical time-splitting approach for solving finite-time optimal control problems
Authors:
Georgios Stathopoulos,
Tamás Keviczky,
Yang Wang
Abstract:
We present a hierarchical computation approach for solving finite-time optimal control problems using operator splitting methods. The first split is performed over the time index and leads to as many subproblems as the length of the prediction horizon. Each subproblem is solved in parallel and further split into three by separating the objective from the equality and inequality constraints respect…
▽ More
We present a hierarchical computation approach for solving finite-time optimal control problems using operator splitting methods. The first split is performed over the time index and leads to as many subproblems as the length of the prediction horizon. Each subproblem is solved in parallel and further split into three by separating the objective from the equality and inequality constraints respectively, such that an analytic solution can be achieved for each subproblem. The proposed solution approach leads to a nested decomposition scheme, which is highly parallelizable. We present a numerical comparison with standard state-of-the-art solvers, and provide analytic solutions to several elements of the algorithm, which enhances its applicability in fast large-scale applications.
△ Less
Submitted 8 April, 2013;
originally announced April 2013.