Falsification of Learning-Based Controllers through Multi-Fidelity Bayesian Optimization. - Zahra Shahrooei; Mykel J. Kochenderfer; Ali Baheri (ECC, 2023)

Optimal Transport-Assisted Risk-Sensitive Q-Learning. - Zahra Shahrooei; Ali Baheri (CoRR, 2024)

Optimizing Falsification for Learning-Based Control Systems: A Multi-Fidelity Bayesian Approach. - Zahra Shahrooei; Mykel J. Kochenderfer; Ali Baheri (CoRR, 2024)

WAVE: Wasserstein Adaptive Value Estimation for Actor-Critic Reinforcement Learning. - Ali Baheri; Zahra Shahrooei; Chirayu Salgarkar (L4DC, 2025)

Risk-Averse Reinforcement Learning: An Optimal Transport Perspective on Temporal Difference Learning. - Zahra Shahrooei; Ali Baheri (CoRR, 2025)

Wasserstein Barycenter Soft Actor-Critic. - Zahra Shahrooei; Ali Baheri (CoRR, 2025)