About

I am a fourth-year PhD student majoring in Statistics at the University of Wisconsin-Madison. My previous research experience has been primarily centered around machine learning, with a particular focus on the processing of sequential data. Currently, I am pursuing research interests in social network analysis, non-Euclidean models and large language models (LLM). I am privileged to work under the guidance of Professor Karl Rohe and Professor Frederic Sala.

Prior to that, I graduated from Zhejiang University with a B.S. in Statistics and the University of Wisconsin-Madison with a M.S. in Statistics. During my master's studies, I was lucky to work with Dr.Sebastian Raschka and co-authored Chapter 16: Transformers–Improving Natural Language Processing with Attention Mechanisms in his book Machine Learning with PyTorch and Scikit-Learn.

My full CV can be found here.

Work Experience

Amazon Alexa AI
Applied scientist intern, 2023 summer
Build a vector search system for multi-modal data including text, image and video. Proposed a two-stage LLM-augmented search method to enable search by user preference.

Amazon Alexa AI
Applied scientist intern, 2022 summer
Build a transformer-based multi-task learning model for label automation and label recommendation.

Teaching Experience

STAT 371: Introductory Applied Statistics for the Life Sciences
20 Fall, 21 Spring, 21 Summer

STAT 451: Introduction to Machine learning and Statistical Pattern Classification
21 Fall, 22 Spring, 22 Fall, 23 Spring

Services

I served as reviewers for following conferences and journals: NeurIPS 2023, ICML 2023, TMLR.