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

  • L2UL was accepted to WSDM 2027 Findings
    Machine Unlearning
  • Released one paper about Distributional Machine Learning (DiFuMVC).
  • Wrapped up my five-month internship at Alibaba (AliExpress).
  • Released one paper about Machine Unlearning (L2UL).
  • Released one paper about Self-Improving AI Systems (SIRI).
  • Started an internship at Alibaba (AliExpress).
  • UniCorn was accepted to ACL 2026
    Self-Improving AI Systems
  • Released one paper about Distributional Machine Learning (GCSC).
  • Released one paper about Self-Improving AI Systems (UniCorn).
  • Released one paper about Distributional Machine Learning (DC).
  • TIV was accepted to AAAI 2026
    Efficient Language Models
  • Wrapped up my three-month internship at Alibaba (Quark).
  • Released one paper about Distributional Machine Learning (DAE).
  • LLM4MS was accepted by Communications Chemistry
    AI for Science
  • Released one paper about Efficient Language Models (TIV).
  • PIP was accepted to EMNLP 2025 Findings
    Efficient Language Models
  • Started an internship at Alibaba (Quark).
  • Released one paper about AI for Science (LLM4MS).
  • Released one paper about Efficient Language Models (PIP).
  • Released one paper about Distributional Machine Learning (DBSC).
  • Released one paper about Distributional Machine Learning (D-Spec).
  • Started my master's program at 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, Yaokun Fang, Hang Zhang, Vincent Henric, Chang Yi, Jianhui Ji, Rong Xiao, Xiaoyi Zeng, Kai Ming Ting
Self-Improving AI Systems · 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
Distributional Machine Learning · Preprint 2026 · [arXiv]

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 Language Models · EMNLP 2025 Findings · [Pdf] · [arXiv]

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 Language Models · AAAI 2026 · [Pdf]

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
AI for Science · Communications Chemistry 2025 · [Pdf] · [Code]

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
Self-Improving AI Systems · ACL 2026 · [Pdf] · [arXiv] · [Code]

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.

L2UL Paper Image
Learning to Unlearn: Machine Unlearning via Learning the Unlearning Behaviors
Hang Zhang, Kai-Feng Zhang, Yi-Xiao Ma, Wei-Jie Xu, Ye Zhu, Kai Ming Ting
Machine Unlearning · WSDM 2027 Findings · [arXiv]

We propose L2UL, a learning-based and model-agnostic framework for machine unlearning that learns the unlearning behavior instead of manually designing an unlearning function. By representing data distributions with kernel mean embeddings, L2UL achieves accuracy comparable to retraining while providing substantial efficiency gains, especially in data-intensive scenarios.

Experience

Alibaba Logo
Algorithm Engineer Intern  |  Apr. 2026 - Sep. 2026
Agent Evolving, Context Engineering
  • AliExpress, Alibaba International Digital Commerce Group (AIDC), Alibaba Group
  • Developed SIRI, a self-improving framework for e-commerce Query-Item relevance auto-labeling
  • Designed module-decoupled multi-agent debate and dual-memory context evolution for robust relevance judgment
Alibaba Logo
Algorithm Engineer Intern  |  Jul. 2025 - Oct. 2025
Deep Search, Agentic RL
  • Qwen Business Unit, Alibaba Token Hub (ATH), Alibaba Group
  • Built and improved the post-training framework for agentic RL
  • Assisted in building a data synthesis framework

Award & Service

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