Wei-Jie Xu

Currently I am a master student of School of Artificial Intelligence in Nanjing University and a member of LAMDA group, led by Prof. Zhi-Hua Zhou.

In June 2024, I received my B.Eng. degree from School of Computer Science and Technology, Soochow University. In the same year, I was admitted to study for a master's degree in School of Artificial Intelligence, Nanjing University under the supervision of Prof. Kai Ming Ting.

Email: xuwj@lamda.nju.edu.cn
Office: B306, Big Data and Artificial Intelligence Research Building, Nanjing University Xianlin Campus

News

  • [Aug.2026] Release one paper about Agent Evolving (SIRI).
  • [Apr.2026] Start an Internship in Alibaba (AliExpress).
  • [Apr.2026] One paper about MLLM (UniCorn) has been accepted by ACL 2026 !
  • [Jan.2026] Release one paper about Spectral Clustering (GCSC).
  • [Jan.2026] Release one paper about MLLM (UniCorn).
  • [Dec.2025] Release one paper about Deep Clustering.
  • [Nov.2025] One paper about Efficient LLM (TIV) has been accepted by AAAI 2026 !
  • [Oct.2025] Spend three wonderful months in Alibaba (Quark).
  • [Sep.2025] Release one paper about Dimension Reduction (DAE).
  • [Aug.2025] One paper about AI4Sci (LLM4MS) has been accepted by Communication Chemistry !
  • [Aug.2025] Release one paper about Efficient LLM (TIV).
  • [Aug.2025] One paper about Efficient LLM (PIP) has been accepted by EMNLP 2025 Findings !
  • [Jul.2025] Start an Internship in Alibaba (Quark).
  • [Apr.2025] Release one paper about LLM4Sci (LLM4MS).
  • [Jan.2025] Release one paper about Efficient LLM (PIP).
  • [Nov.2024] Release one paper about Spectral Clustering (DBSC).
  • [Oct.2024] Release one paper about Spectral Clustering (D-Spec).
  • [Sep.2024] Enroll as a master student in Nanjing University.

Preprint

These papers are currently surviving the peer-review gauntlet — send thoughts and prayers 😂.

SIRI Paper Image
SIRI: A Self-Improving Framework for Query-Item Relevance via Iterative Context Evolution
Wei-Jie Xu, Hang Zhang, Yaokun Fang, Vincent Henric, Chang Yi, Jianhui Ji, Rong Xiao, Xiaoyi Zeng, Kai Ming Ting
Agent Evolving · Preprint 2026

We propose SIRI, a two-stage self-improving agentic framework for e-commerce Query-Item relevance auto-labeling. By combining module-decoupled multi-agent debate with dual-memory, bi-level context evolution, SIRI improves average judgment accuracy from 90.53 to 94.24 while reducing reliance on costly manual annotation.

DC Paper Image
How to Achieve the Intended Aim of Deep Clustering Now, without Deep Learning
Kai Ming Ting, Wei-Jie Xu, Hang Zhang
Deep Clustering · Preprint 2026

We investigate whether deep clustering, especially DEC, can overcome the fundamental limitations of k-means clustering on arbitrary shapes, varied sizes, and densities. The study shows that current deep clustering methods fall short without distributional information and effective learned centroid-based representations, while a non-deep approach can better achieve this goal.

Research

PIP Paper Image
PIP: Perturbation-based Iterative Pruning for Large Language Models
Yi Cao, Wei-Jie Xu, Yucheng Shen, Weijie Shi, Chi-Min Chan, Jiajie Xu
Efficient LLM · EMNLP 2025 Findings

We propose PIP (Perturbation-based Iterative Pruning), a method that iteratively prunes parameters based on the distinction between unperturbed and perturbed views. Experimental results show that PIP reduces parameter count by approximately 20% while retaining over 85% of the original accuracy.

TIV Paper Image
TIV: Thought Injection via Vectors for Efficient Reasoning in Large Reasoning Models
Yi Cao, Weijie Shi, Wei-Jie Xu, Yucheng Shen, Yue Cui, Hanghui Guo, Shimin Di, Ziyi Liu, Jiaming Li, Alexander Zhou, Jia Zhu, Jiajie Xu
Efficient LLM · AAAI 2026

We propose TIV, an innovative framework that compresses token-level reasoning into compact vectors without sacrificing performance. Rather than generating explicit thoughts, TIV injects learnable vectors into the post-attention hidden states of the final token across Transformer layers, enabling implicit and lightweight reasoning.

LLM4MS Paper Image
A large language model for deriving spectral embeddings for accurate compound identification in mass spectrometry
Yang Xu, Yi-Xiao Ma, Wei-Jie Xu, Zu-Liang Yang, Kai Ming Ting

We propose LLM4MS, a method leveraging expert knowledge from large language models to generate discriminative spectral embeddings for improved compound identification. Experimental results show a 13.7% improvement in accuracy over existing methods on a million-scale library, with a query speed of nearly 15,000 queries per second.

UniCorn Paper Image
UniCorn: Towards Self-Improving Unified Multimodal Models through Self-Generated Supervision
Ruiyan Han, Zhen Fang, XinYu Sun, Yuchen Ma, Ziheng Wang, Yu Zeng, Zehui Chen, Lin Chen, Wenxuan Huang, Wei-Jie Xu, Yi Cao, Feng Zhao
MLLM · ACL 2026

We propose UniCorn, a simple yet elegant self-improvement framework that eliminates the need for external data or teacher supervision. By partitioning a single UMM into three collaborative roles: Proposer, Solver, and Judge, UniCorn generates high-quality interactions via self-play and employs cognitive pattern reconstruction to distill latent understanding into explicit generative signals.

Experience

Alibaba Logo
Algorithm Engineer Intern
Quark, Alibaba Group  |  Jul. 2025 – Oct. 2025
Deep Search, Reinforcement Learning, VeRL
  • Quark, Intelligent Information, Alibaba Group
  • Built and improved the post-training framework for agentic RL
  • Assisted in building a data synthesis framework

Award & Service

  • New Point Software Scholarship, 2021–2022, School of Computer Science and Technology, Soochow University.
  • Teaching Assistant, Introduction to Data Mining (with Prof. Kai Ming Ting), 2026 Spring, School of Artificial Intelligence, Nanjing University.