Wenzhi Li Wenzhi Li's name in Chinese

How to pronounce my name? Wen-Jee Lee.

Hi there! I am a third-year PhD student in Computer Science at Cornell University, advised by Claire Cardie and Sainyam Galhotra.

Previously, I obtained my B.S. and M.S. degrees from Jilin University and Sun Yat-sen University respectively, both in computer science. I worked with Professor Hui Xiong from Hong Kong University of Science and Technology on weakly supervised machine learning for graph data during my master's studies. I interned as an applied scientist at AWS during 2026 summer, working on agentic Text-to-SQL.

My recent research focuses on LLM reasoning and agentic systems. I am particularly interested in how language models learn and retain reasoning capabilities, and how AI agents retrieve information, resolve uncertainty, and interact with complex data environments. My earlier work spans machine learning theory and graph representation learning. Representative research directions include:

  • LLM Reasoning & Post-Training, where I study how reinforcement learning affects reasoning diversity and develop methods to preserve diverse, correct reasoning behaviors while improving performance. [TAR]
  • Agentic Data Management Systems, where I develop LLM agents that discover relevant information across complex data lakes and resolve underspecified user requests through interactive clarification. [Octopus, PlanPool]
  • Machine Learning Theory, where I studied query-efficient approximation algorithms for metric clustering and graph optimization using noisy, weak comparison oracles. [COLT'25]
  • Machine Learning on Graphs, where I developed methods for learning effective and robust representations of graph-structured data, including imbalanced node classification, graph contrastive learning, and heterogeneous graph modeling. [GraphSHA, HomoGCL, AEGCL, StarGAT]
News
Publications
Preprints
Execution accuracy and ambiguity distributions in agentic Text-to-SQL
Beyond Correctness: Resolving Underspecification in Agentic Text-to-SQL
Wen-Zhi Li, Yue Gong, Konstantinos Kanellis, Balakrishnan Murali Narayanaswamy
arXiv preprint, 2026.
[arXiv] • [BibTeX]

System overview of Octopus
Conference & Journal Papers
Overview figure of the COLT'25 paper
Metric Clustering and Graph Optimization Problems using Weak Comparison Oracles
Rahul Raychaudhury, Wen-Zhi Li, Syamantak Das, Sainyam Galhotra, Stavros Sintos
38th Annual Conference on Learning Theory (COLT'25).
[PMLR] • [BibTeX]

Overview figure of GraphSHA
GraphSHA: Synthesizing Harder Samples for Class-Imbalanced Node Classification
Wen-Zhi Li, Chang-Dong Wang, Hui Xiong, Jian-Huang Lai
29th ACM SIGKDD Conference on Knowledge Discovery and Data Mining (KDD'23).
[Project Page] • [arXiv] • [ACM Digital Library] • [Code] • [Poster] • [Talk] • [Slides] • [BibTeX]
🌟 Oral Presentation

Overview figure of HomoGCL
HomoGCL: Rethinking Homophily in Graph Contrastive Learning
Wen-Zhi Li, Chang-Dong Wang, Hui Xiong, Jian-Huang Lai
29th ACM SIGKDD Conference on Knowledge Discovery and Data Mining (KDD'23).
[Project Page] • [arXiv] • [ACM Digital Library] • [Code] • [Poster] • [Talk] • [Slides] • [BibTeX]
🌟 Oral Presentation

Overview figure of AEGCL
Towards Effective and Robust Graph Contrastive Learning with Graph Autoencoding
Wen-Zhi Li, Chang-Dong Wang, Jian-Huang Lai, Philip S. Yu
IEEE Transactions on Knowledge and Data Engineering (TKDE).
[IEEE Xplore] • [Code] • [BibTeX]

Overview figure of StarGAT
StarGAT: Star-Shaped Hierarchical Graph Attentional Network for Heterogeneous Network Representation Learning
Wen-Zhi Li, Ling Huang, Chang-Dong Wang, Yuxin Ye
21st IEEE International Conference on Data Mining (ICDM'21).
[IEEE Xplore] • [Code] • [Slides] • [BibTeX]
Talks

Oral Presentation @ KDD 2023
Oral Presentation @ KDD 2023
Experiences
Amazon Web Services, May 2026-Aug. 2026
Applied Scientist Intern
Cornell University, Aug. 2024-present
PhD student in Computer Science
The Hong Kong University of Science and Technology, May 2022-July 2024
Research Assistant in Artificial Intelligence Thrust
Sun Yat-sen University, Sept. 2021-June 2024
M.S. in Computer Science
University of Alberta, July 2019-Aug. 2019
Visiting Student in Department of Mathematical and Statistical Sciences
Jilin University, Sept. 2017-June 2021
B.S. in Computer Science
Professional Services