Wenjie Sun (孙文杰)

Incoming Ph.D. student in Computer Science, University of British Columbia
Wenjie Sun

I am a Research Assistant at the Chinese University of Hong Kong, Shenzhen, supervised by Mengnan Du. I will join the University of British Columbia as a Computer Science Ph.D. student in Fall 2026.

My research broadly studies the mechanisms behind model behavior, including training dynamics, feature learning, decision shortcuts, and generalization. I am especially interested in connecting mechanistic explanations with empirical behaviors in modern neural networks.

Previously, I earned a Master's degree in Electronic and Information Engineering from SUSTech in the Fall of 2025, and obtained a Bachelor's degree in Applied Chemistry from Northeastern University in 2022.

Links: CV

News

May 2026
Jan 2026

Our paper "The Price of Amortized Inference in Sparse Autoencoders" has been accepted as a poster by ICLR 2026.

October 2025

Our paper "DiagECG: Discretized ECG Tokenization for Diagnostic Reasoning" has been accepted as a poster by AAAI 2026.

Aug 2025

I received my master's degree from Southern University of Science and Technology (SUSTech).

May 2025

New preprint on our paper, Sparsification and Reconstruction from the Perspective of Representation Geometry, the last paper during my master's program.

Dec 2024

Two of my journal papers have been accepted for publication in IEEE Transactions on Artificial Intelligence(CCF-B) and Energy(IF=9.4), respectively! The former ( CauseTerML) is my first paper since changing my research focus.

Nov 2024

I am honored to be awarded the China National Scholarship!(Top0.3%)

Jun 2024

My first journal paper was accpted by Journal of Energy Storage(IF=9.8)!


Publications

2026

  1. Law of Neural Interaction: Depth-Width Shape, Interaction Efficiency, and Generalization
    Wenjie Sun, Jinning Yang, Shuai Zhang, Mengnan Du
    arXiv preprint, 2026
  2. The Price of Amortized Inference in Sparse Autoencoders
    Wenjie Sun, Di Wang, Lijie Hu
    ICLR, 2026 (Poster)
  3. DiagECG: Discretized ECG Tokenization for Diagnostic Reasoning
    Jinning Yang, Wenjie Sun, Shi Wen
    AAAI, 2026 (Poster)

2025

  1. Sparsification and Reconstruction from the Perspective of Representation Geometry
    Wenjie Sun, Bingzhe Wu, Zhile Yang, Chengke Wu
    arXiv, 2025
  2. CauseTerML: Causal Learning via Term Mining for Assessing Review Discrepancies
    Wenjie Sun, Chengke Wu ... Zhile Yang
    IEEE Transactions on Artificial Intelligence (TAI), 2025
  3. Fine-tuning enables state of health estimation for lithium-ion batteries via a time series foundation model
    Wenjie Sun, Chengke Wu ... Wenjiao Yao
    Energy, 2025

2024

  1. State-of-charge estimation of sodium-ion batteries: A fusion deep learning approach
    Wenjie Sun, Huan Xu ... Zhile Yang
    Journal of Energy Storage, 2024