Zhiwei Xu

Zhiwei Xu

Ph.D. Candidate in Statistics

University of Michigan, Ann Arbor

Hi! I am Zhiwei, a fourth-year Ph.D. candidate in Statistics at University of Michigan, Ann Arbor, gratefully advised by Prof. Wei Hu and Prof. Yixin Wang. My research interest is centered around theoretical and scientific understanding of foundation models.

Recently I'm interested in designing more efficient model architecture and more effective training strategies for agentic models. Feel free to reach out if you share the same interests!

Education

University of Michigan

Ph.D. in Statistics

Advisors: Wei Hu & Yixin Wang

2022 - Present

University of Science and Technology of China

B.S. in Statistics

Advisor: Weiping Zhang

2018 - 2022

Publications & Preprints

Recommendations Beyond Catalogs: Diffusion Models for Personalized Generation

Gabriel Patron, Zhiwei Xu, Ishan Kapnadak, Felipe Maia Polo

Submitted.

Benign Overfitting in Single-Head Attention

Roey Magen*, Shuning Shang*, Zhiwei Xu, Spencer Frei, Wei Hu, Gal Vardi

NeurIPS 2025.

Let Me Grok for You: Accelerating Grokking via Embedding Transfer from a Weaker Model

Zhiwei Xu*, Zhiyu Ni*, Yixin Wang, Wei Hu

ICLR 2025.

Benign Overfitting and Grokking in ReLU Networks for XOR Cluster Data

Zhiwei Xu, Yutong Wang, Spencer Frei, Gal Vardi, Wei Hu

ICLR 2024.

Inference of Dependency Knowledge Graph for Electronic Health Records

Zhiwei Xu, Ziming Gan, Doudou Zhou, Shuting Shen, Junwei Lu*, Tianxi Cai*

Journal of the Royal Statistical Society, Series B, 2025

ARCH: Large-scale Knowledge Graph via Aggregated Narrative Codified Health Records Analysis

Ziming Gan*, Doudou Zhou*, et al.

Journal of Biomedical Informatics, 2025

Teaching

STATS 500: Statistical Learning

Fall 2024

Graduate Student Instructor, University of Michigan

Graded homework and conducted office hours

STATS 413: Applied Regression Analysis

Winter 2024

Graduate Student Instructor, University of Michigan

Taught weekly lab classes, graded homework, and conducted office hours

STATS/DATASCI 451: Bayesian Data Analysis

Fall 2023

Graduate Student Instructor, University of Michigan

Graded homework and conducted office hours

STATS 250: Introduction to Statistics and Data Analysis

Winter 2023

Graduate Student Instructor, University of Michigan

Taught weekly lab classes, graded homework and lab, and conducted office hours

Real Analysis

Spring 2021

Teaching Assistant, USTC

Taught weekly lab classes and graded homework

Service