Hello! I’m Weiyi He, a third-year Ph.D. student at Michigan State University (MSU), dual-majoring in Statistics and Computer Science. I am fortunate to be advised by Prof. Yue Xing and Prof. Jiliang Tang. I received my B.S. in Statistics from the University of Science and Technology of China (USTC).

My research focuses on Trustworthy AI, particularly LLM robustness and Adversarial Training. I am interested in developing statistical tools to understand modern AI systems and principled methods to improve their reliability. I was an Applied Scientist Intern at Amazon AGI, where I worked with Giuseppe Vietri, Dionysis Manousakas, and Dmitriy Bespalov on understanding and evaluating LLM robustness to input and parameter perturbations.

*Bold indicates first or co-first authorship.

News

  • I joined Amazon AGI as an Applied Scientist Intern in Arlington, VA.

  • I presented my first poster at AISTATS 2026 in Tangier, Morocco.

  • One paper is accepted to AISTATS 2026!

  • I have passed the preliminary exams in both Statistics and Probability!

  • One paper is accepted to ACL 2025!

Selected Publications

View all publications →