Daiwei Chen

University of Wisconsin-Madison
Department of Electrical and Computer Engineering

I am a third-year PhD student in Electrical and Computer Engineering at the University of Wisconsin-Madison, working on Reliability of Large Language Models under the supervision of Prof. Ramya Korlakai Vinayak.

I received my M.S. in Systems Engineering from the University of Pennsylvania (advised by Prof. Pratik Chaudhari, working on Machine Learning Theory) and my B.S. in Psychology from Zhejiang University.

Research Interests

My research mainly focuses on Improving the Reliability of Large Language Models, with particular emphasis on three core directions: Pluralism: Addressing the lack of output diversity (e.g., mode collapsing) in aligned models. Factuality: Researching methods to bridge the gap between generated overconfidence and actual correctness. Generalization: Investigating how foundational models can maintain performance and adaptability across novel tasks.

Publications

Preprints

arXiv 2025

LinkedOut: Linking World Knowledge Representation Out of Video LLM for Next-Generation Video Recommendation

H Zhang, Y Lu, L Wang, Y Li, D Chen, Y Xu, Y Fu

arXiv

Selected Publications

ICLR 2025

PAL: Sample-Efficient Personalized Reward Modeling for Pluralistic Alignment

D Chen, Y Chen, A Rege, Z Wang, RK Vinayak

Paper arXiv Project Page

ICLR 2024 Blogpost Track

Unraveling the Impact of Training Samples

D Chen, J Zhang, RK Vinayak

Blogpost

arXiv 2023

Learning Capacity: A Measure of the Effective Dimensionality of a Model

D Chen, WK Chang, P Chaudhari

arXiv

News

Nov 2025 Received Research Grant Sponsorship from Lambda.AI 🎉.
Aug 2025 Completed GenAI Research Internship at Microsoft LinkedIn, working on LLM Recommendation System.
Apr 2025 Attending ICLR 2025 at Singapore EXPO, Apr 24-28. Wish to see you there! 🤗
Jan 2025 Our paper PAL accepted to ICLR 2025!
Dec 2024 Attended NeurIPS 2024 in Vancouver. PAL received a 🔥 Spotlight Talk at the Behavioral ML workshop.
Feb 2024 Our blogpost Unraveling The Impact of Training Samples accepted by ICLR 2024.

Academic Service

Reviewer

NeurIPS 2026 · ICLR 2026 · CVPR 2026 · EACL 2026

Teaching

ECE 204 Data Science & Engineering — UW-Madison
ECE 532 Matrix Methods in Machine Learning — UW-Madison
CIS 350 Software Design & Engineering — UPenn
ESE 542 Statistics for Data Science — UPenn

Awards & Achievements

2025 Research Grant Sponsorship — Lambda.AI
2023 Outstanding Research Award — University of Pennsylvania
2021 Zhejiang University Scholarship — Zhejiang University

Contact

I'm always open to discussing research collaborations. Feel free to reach out!

Email: [email protected]

Office: Wisconsin Institute for Discovery, Room 2115G-2, UW-Madison