← Search

Zhouhao Yang

3 accepted papers

2024

Memory-Efficient Gradient Unrolling for Large-Scale Bi-level Optimization

NeurIPS 2024poster

Bi-level optimizaiton (BO) has become a fundamental mathematical framework for addressing hierarchical machine learning problems. As deep learning models continue to grow in size, the demand for scalable bi-level optimization has become increasingly critical. Traditional gradient-based bi-level opti…

2023

Distributionally Robust Policy Gradient for Offline Contextual Bandits

AISTATS 2023poster

Learning an optimal policy from offline data is notoriously challenging, which requires the evaluation of the learning policy using data pre-collected from a static logging policy. We study the policy optimization problem in offline contextual bandits using policy gradient methods. We employ a distr…