Zhiwei Xu

Email: zhiweixu AT umich DOT edu

Hi! I am Zhiwei, 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 statistical inference and theoretical understanding of foundation models, especially some surprising generalization phenomena.

Education

University of Michigan

Ann Arbor, MI, USA
Ph.D. in Statistics
Advisor: Wei Hu & Yixin Wang
2022 - Present

University of Science and Technology of China

Hefei, Anhui, China
B.S. in Statistics
Advisor: Weiping Zhang
2018 - 2022

Preprints/Publications

Benign Overfitting in Single-Head Attention

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

Submitted.

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*

Preprint, 2023.

Under review

Teaching

STATS/DATASCI 451: Bayesian Data Analysis

UMich
Graded homework and conducted office hours
FALL 2023

STATS 250: Introduction to Statistics and Data Analysis

UMich
Taught weekly lab classes, graded homework and lab, and conducted office hours
WINTER 2023

Real Analysis

USTC
Taught weekly lab classes and graded homework
SPRING 2021