Wai Tuck Wong

Beyond NaN: Resiliency of Optimization Layers in the Face of Infeasibility

Prior work has successfully incorporated optimization layers as the last layer in neural networks for various problems, thereby allowing joint learning and planning in one neural network forward pass. In this work, we identify a weakness in such a set-up where inputs to the optimization layer lead t…