Kai-Chia Mo

A Fully First-Order Layer for Differentiable Optimization

Differentiable optimization studies how to embed a mathematical program as a differentiable layer in machine learning pipelines. However, existing approaches typically rely on implicit differentiation, involving expensive Hessian computation while differentiating through optimality conditions. To ad…