← Search

Tianyue H. Zhang

5 accepted papers

2025

Addressing Concept Mislabeling in Concept Bottleneck Models Through Preference Optimization

ICML 2025poster

Concept Bottleneck Models (CBMs) propose to enhance the trustworthiness of AI systems by constraining their decisions on a set of human understandable concepts. However, CBMs typically rely on datasets with assumedly accurate concept labels—an assumption often violated in practice which we show can…

Cited by 0SourcePDFScholar
2025

Connecting Thompson Sampling and UCB: Towards More Efficient Trade-offs Between Privacy and Regret

ICML 2025poster

We address differentially private stochastic bandit problems by leveraging Thompson Sampling with Gaussian priors and Gaussian differential privacy (GDP). We propose DP-TS-UCB, a novel parametrized private algorithm that enables trading off privacy and regret. DP-TS-UCB satisfies $ \tilde{O} \l…

Cited by 0SourcePDFScholar
2025

Understanding Adam Requires Better Rotation Dependent Assumptions

NeurIPS 2025poster

Despite its widespread adoption, Adam's advantage over Stochastic Gradient Descent (SGD) lacks a comprehensive theoretical explanation. This paper investigates Adam's sensitivity to rotations of the parameter space. We observe that Adam's performance in training transformers degrades under random ro…

Cited by 0SourceScholar
2024

On PI Controllers for Updating Lagrange Multipliers in Constrained Optimization

ICML 2024poster

Constrained optimization offers a powerful framework to prescribe desired behaviors in neural network models. Typically, constrained problems are solved via their min-max Lagrangian formulations, which exhibit unstable oscillatory dynamics when optimized using gradient descent-ascent. The adoption o…

2023

Optimistic Thompson Sampling-based algorithms for episodic reinforcement learning

UAI 2023poster

We propose two Thompson Sampling-like, model-based learning algorithms for episodic Markov decision processes (MDPs) with a finite time horizon. Our proposed algorithms are inspired by Optimistic Thompson Sampling (O-TS), empirically studied in Chapelle and Li [2011], May et al. [2012] for stochas…

Cited by 6SourcePDFScholar