Wenjie Sun (孙文杰)

Wenjie Sun

I'm a Research Engineer at MBZUAI where I'm advised by Lijie Hu. Broadly speaking, I'm interested in any mechanisms of model behaviors. Specifically, such as training dynamics and feature learning on the mechanism level, and decision shortcuts and generalization principles on the behavioral level.

Currently, I am seeking a PhD position starting in Fall 2026 or later within these research areas. If you are interested in collaborating, please feel free to reach out—let's work on something interesting together.

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

October 2025

Our paper "DiagECG: Discretized ECG Tokenization for Diagnostic Reasoning" has benn 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. 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