3 papers
cs.LG2025
Adapting Prediction Sets to Distribution Shifts Without Labels
Kevin Kasa, Zhiyu Zhang, Heng Yang +1
Recently there has been a surge of interest to deploy confidence set predictions rather than point predictions in machine learning. Unfortunately, the effectiveness of such predict…
stat.ML2025
The Benefit of Being Bayesian in Online Conformal Prediction
Zhiyu Zhang, Zhou Lu, Heng Yang
Based on the framework of Conformal Prediction (CP), we study the online construction of confidence sets given a black-box machine learning model. By converting the target confiden…
cs.LG2024
Fast TRAC: A Parameter-Free Optimizer for Lifelong Reinforcement Learning
Aneesh Muppidi, Zhiyu Zhang, Heng Yang
A key challenge in lifelong reinforcement learning (RL) is the loss of plasticity, where previous learning progress hinders an agent's adaptation to new tasks. While regularization…