← Search

Ramesha Karunasena

2 accepted papers

2023

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

AAAI 2023technical

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…

2023

Generative Modelling of Stochastic Actions with Arbitrary Constraints in Reinforcement Learning

NeurIPS 2023poster

Many problems in Reinforcement Learning (RL) seek an optimal policy with large discrete multidimensional yet unordered action spaces; these include problems in randomized allocation of resources such as placements of multiple security resources and emergency response units, etc. A challenge in this…