A Quantile Neural Network Framework for Two-stage Stochastic Optimization

Antonio Alcántara,Carlos Ruiz,Calvin Tsay
2024-03-18
Abstract:Two-stage stochastic programming is a popular framework for optimization under uncertainty, where decision variables are split between first-stage decisions, and second-stage (or recourse) decisions, with the latter being adjusted after uncertainty is realized. These problems are often formulated using Sample Average Approximation (SAA), where uncertainty is modeled as a finite set of scenarios, resulting in a large "monolithic" problem, i.e., where the model is repeated for each scenario. The resulting models can be challenging to solve, and several problem-specific decomposition approaches have been proposed. An alternative approach is to approximate the expected second-stage objective value using a surrogate model, which can then be embedded in the first-stage problem to produce good heuristic solutions. In this work, we propose to instead model the distribution of the second-stage objective, specifically using a quantile neural network. Embedding this distributional approximation enables capturing uncertainty and is not limited to expected-value optimization, e.g., the proposed approach enables optimization of the Conditional Value at Risk (CVaR). We discuss optimization formulations for embedding the quantile neural network and demonstrate the effectiveness of the proposed framework using several computational case studies including a set of mixed-integer optimization problems.
Optimization and Control
What problem does this paper attempt to address?
This paper proposes a method to solve two-stage stochastic optimization problems using the Quantile Neural Network (QNN) framework. In two-stage stochastic optimization, decision variables are divided into first-stage decisions and second-stage (or recourse) decisions, where the second-stage decisions are adjusted after uncertainty realization. Typically, such problems are approximated using the Sample Average Approximation (SAA) method, which simulates uncertainty with a finite number of scenarios, resulting in a large "aggregate" problem. The SAA method may lead to a high computational complexity. Therefore, this paper suggests using QNN to approximate the distribution of the second-stage objective function, rather than just its expected value. QNN can capture uncertainty and support non-expected value optimization, such as Conditional Value at Risk (CVaR). The paper introduces two types of QNN structures: unconstrained feedforward neural network and output-constrained neural network that ensures non-decreasing property to estimate the quantiles of the second-stage objective function. By embedding QNN into the optimization problem, heuristic solutions to the original problem can be quickly obtained even when the scenario set is large, and the difference between QNN and SAA approximations is small in many cases. Furthermore, QNN performs better under time constraints for large scenario sets. The paper also reviews the relevant background knowledge of two-stage stochastic optimization, quantile neural network, and neural network embedding in mixed integer optimization. The process of data generation, QNN training, and embedding is proposed. Finally, the effectiveness of the proposed framework is demonstrated through multiple computational case studies.