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
- arXiv preprint arXiv:2607.28959, 2026
-
Impact of Positional Encoding: Clean and Adversarial Rademacher Complexity for Transformers under In-Context Regression
International Conference on Artificial Intelligence and Statistics (AISTATS), 2026 -
Unveiling privacy risks in llm agent memory
Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (ACL), 2025