Wenjie Du

Welcome to my homepage!🤗 I'm Wenjie Du (杜文杰), a MSc student at Nanyang Technological University (2026.01 - now). Previously, I obtained BEng in Computer Science and Technology from Sichuan University in 2024.

I have been working as a research intern in ENCODE Lab of Prof. Huan Wang at Westlake University since Fall 2025. I was also fortunate to work as a research assistant at the Hong Kong University of Science and Technology in 2025, a research intern at the Institute for AI Industry Research (AIR), Tsinghua University in 2024.

I work closely with Yuhao at NTU and Keda at Westlake. I am grateful to all my supervisors, mentors, and friends for their kindness and support throughout my journey!

Email  /  CV  /  Scholar  /  Linkedin  /  Github

I am seeking a PhD position starting in Fall 2027.

profile photo

Research

My research interests lie in large language models and machine learning systems. I focus on understanding the inner mechanisms of these models to improve their efficiency and effectiveness.

clean-usnob Which Heads Matter for Reasoning? RL-Guided KV Cache Compression
Wenjie Du, Li Jiang, Keda Tao, Xue Liu, Huan Wang,
arXiv, 2025
page / arXiv

RLKV uses RL as a probe to reveal head-level sparsity in reasoning LLMs, enabling 20-50% KV cache reduction with minimal performance loss.


This webpage is built upon the source code of Jon Barron.