Zirui Yan

alt text 

PhD student
Department of Electrical, Computer and Systems Engineering
Rensselaer Polytechnic institute
Email: yanz11@rpi.edu/ziruiy1808@gmail.com
Tel: +1-5189612312
Curriculum Vitae

About me

I am Ph.D. student in Electrical, Computer, and System Engineering Department at Rensselaer Polytechnic Institute (RPI), working under the supervision of Prof. Ali Tajer. Before that, I graduated from University of Science and Technology of China.

Research

My current research topic is causal bandit (see github repo for brief summary) and the causality in Large Language models (LLM). I also have broad interests in the application of bandit in federated learning.

Education

News

  • 01/2023: Our paper “Nonlinear Causal Bandits: General Causal Models and Interventions” will be appear in AISTATS 2024.

  • 12/2023: I served as volunteer at NeurIPS 2023.

  • 11/2023: Our paper “Robust Causal Bandits for Linear Models” is now on arXiv.

  • 08/2023: Finished a three-month internship at IBM Thomas J. Watson Research Center.

  • 03/2023: Our paper “Optimizing Parameter Mixing under Constrained Communications in Parallel Federated Learning” was accepted by IEEE/ACM Transactions on Networking.