Jiaxing Wang
Author directory2026
MixEmb: Importance-Aware Quantization and Caching Co-Design for Industrial Decentralized Embedding Training
Zhida Jiang | Zhen Chen | Zhaolong Xing | Jiaxing Wang | Saixing Gao | Baopeng Yuan | Hao Wang | Yin Xu | Lin Liu | Dongyue Wang | Ke Zhang
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing: Industry Track
Zhida Jiang | Zhen Chen | Zhaolong Xing | Jiaxing Wang | Saixing Gao | Baopeng Yuan | Hao Wang | Yin Xu | Lin Liu | Dongyue Wang | Ke Zhang
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing: Industry Track
Industrial recommendation training increasingly relies on decentralized multi-GPU infrastructure, yet sparse embedding tables remain a major bottleneck due to significant storage, lookup, and communication overhead. Most existing works typically optimize only one of these bottlenecks or are designed for parameter server architectures, making them less effective in industrial decentralized training stacks. To tackle these challenges, we propose MixEmb, a practical framework for efficient decentralized embedding training through importance-aware quantization and caching. We first design a novel importance metric, i.e., embedding score, which can distinguish the contribution of embedding vectors based on spatial heterogeneity and temporal dynamics. Guided by embedding scores, MixEmb employs mixed precision quantization and hierarchical caching at fine-grained vector level. Besides, our key finding is that quantization and caching decisions are tightly coupled. Instead of naive combination, we delicately integrate them into a unified training framework through coupled co-design. A theoretical convergence analysis is also provided for MixEmb. We implement and evaluate MixEmb in real-world training scenarios. Across three workloads, MixEmb attains up to 1.55× speedup and 3.54× compression ratio compared to state-of-the-art methods.
FlowTrain: Flow-Based Decoupled Training for Industrial-Grade Vision-Language Models
Zhida Jiang | Zhaolong Xing | Pei Yang | Xiaolong Chen | Yuanhang Xiao | Chengzhi Huang | Xiyu Liu | Haopeng Liu | Qingyuan Sang | Lingfeng Zhou | Jiaxing Wang | Zicheng Zhang | Wenzhe Wang | Liuxinyu | Yan Li | Zhen Chen | Ke Zhang
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing: Industry Track
Zhida Jiang | Zhaolong Xing | Pei Yang | Xiaolong Chen | Yuanhang Xiao | Chengzhi Huang | Xiyu Liu | Haopeng Liu | Qingyuan Sang | Lingfeng Zhou | Jiaxing Wang | Zicheng Zhang | Wenzhe Wang | Liuxinyu | Yan Li | Zhen Chen | Ke Zhang
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing: Industry Track
Industrial-grade distributed training of vision-language models (VLMs) remains far less efficient than that of unimodal LLMs. Existing solutions either follow a monolithic design that assigns uniform parallelism to heterogeneous modules or adopt a disaggregated deployment that separates modules while executing them as a batch-synchronized pipeline. In this paper, we highlight that the above solutions are still not sufficient, and VLM training can be further decoupled. To this end, we present FlowTrain, a flow-based decoupled training framework that reformulates VLM training as a producer-consumer dataflow coordinated through a unified memory pool. The encoder and backbone can progress independently over a global virtual address space. Since this execution decoupling fundamentally changes the optimization objective of allocation and scheduling, FlowTrain further introduces a heterogeneous parallel allocator that assigns module-specific parallelism strategies by solving a throughput matching problem. The dynamic packing scheduler is used to construct balanced microbatches at runtime according to the actual LLM-side computation cost. Extensive experiments on real-world workloads show that FlowTrain achieves over 50% MFU and up to 1.7× throughput improvement, narrowing the efficiency gap to LLM-only training.
2025
Logic-of-Thought: Injecting Logic into Contexts for Full Reasoning in Large Language Models
Tongxuan Liu | Wenjiang Xu | Weizhe Huang | Yuting Zeng | Jiaxing Wang | Xingyu Wang | Hailong Yang | Jing Li
Proceedings of the 2025 Conference of the Nations of the Americas Chapter of the Association for Computational Linguistics: Human Language Technologies (Volume 1: Long Papers)
Tongxuan Liu | Wenjiang Xu | Weizhe Huang | Yuting Zeng | Jiaxing Wang | Xingyu Wang | Hailong Yang | Jing Li
Proceedings of the 2025 Conference of the Nations of the Americas Chapter of the Association for Computational Linguistics: Human Language Technologies (Volume 1: Long Papers)
Large Language Models (LLMs) have demonstrated remarkable capabilities across various tasks but their performance in complex logical reasoning tasks remains unsatisfactory. Although some prompting methods, such as Chain-of-Thought, can improve the reasoning ability of LLMs to some extent, they suffer from an unfaithful issue where derived conclusions may not align with the generated reasoning chain. To address this issue, some studies employ the approach of propositional logic to further enhance logical reasoning abilities of LLMs. However, the potential omissions in the extraction of logical expressions in these methods can cause information loss in the logical reasoning process, thereby generating incorrect results. To this end, we propose Logic-of-Thought (LoT) prompting which employs propositional logic to generate expanded logical information descriptions and utilizes them as an additional augmentation to original contexts, thereby ensuring information completeness and enhancing logical reasoning ability. LoT is orthogonal to existing prompting methods and can be seamlessly integrated with them. Extensive experiments demonstrate that LoT boosts the performance of various prompting methods with a striking margin across five logical reasoning tasks. In particular, LoT enhances Chain-of-Thought’s performance on the ReClor dataset by +4.35%, improves Chain-of-Thought with Self-Consistency’s performance on the RuleTaker dataset by +3.52%, and boosts performance of Tree-of-Thoughts on the ProofWriter dataset by +8%.
Beyond Logits: Aligning Feature Dynamics for Effective Knowledge Distillation
Guoqiang Gong | Jiaxing Wang | Jin Xu | Deping Xiang | Zicheng Zhang | Leqi Shen | Yifeng Zhang | Junhua Shu | Zhaolong Xing | Zhen Chen | Pengzhang Liu | Ke Zhang
Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Guoqiang Gong | Jiaxing Wang | Jin Xu | Deping Xiang | Zicheng Zhang | Leqi Shen | Yifeng Zhang | Junhua Shu | Zhaolong Xing | Zhen Chen | Pengzhang Liu | Ke Zhang
Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Knowledge distillation (KD) compresses large language models (LLMs), known as teacher models, into lightweight versions called student models, enabling efficient inference and downstream applications. However, prevailing approaches accomplish this by predominantly focusing on matching the final output distributions of student/teacher models. Drawing on the perspective that transformers can be viewed as discretizing ordinary differential equation (ODEs) on integer time steps (corresponding to layer indices), where intermediate features evolve across layers, we argue that effective KD requires aligning the entire feature dynamics between teacher and student models, which we call feature dynamics distillation (FDD). This alignment involves matching both the feature trajectory and its first-order derivative, rather than just the final states. Our approach extends the original KD objective with two additional loss terms: layer-wise feature KD, which matches discretized feature trajectory, and layer feature delta KD, which matches first-order changes in features across adjacent layers. Extensive experiments on various tasks validate the effectiveness of our distillation method.
2024
Breaking the Hourglass Phenomenon of Residual Quantization: Enhancing the Upper Bound of Generative Retrieval
Zhirui Kuai | Zuxu Chen | Huimu Wang | Mingming Li | Dadong Miao | Wang Binbin | Xusong Chen | Li Kuang | Yuxing Han | Jiaxing Wang | Guoyu Tang | Lin Liu | Songlin Wang | Jingwei Zhuo
Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing: Industry Track
Zhirui Kuai | Zuxu Chen | Huimu Wang | Mingming Li | Dadong Miao | Wang Binbin | Xusong Chen | Li Kuang | Yuxing Han | Jiaxing Wang | Guoyu Tang | Lin Liu | Songlin Wang | Jingwei Zhuo
Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing: Industry Track
Generative retrieval (GR) has emerged as a transformative paradigm in search and recommender systems, leveraging numeric-based identifier representations to enhance efficiency and generalization. Notably, methods like TIGER, which employ Residual Quantization-based Semantic Identifiers (RQ-SID), have shown significant promise in e-commerce scenarios by effectively managing item IDs. However, a critical issue termed the “Hourglass” phenomenon, occurs in RQ-SID, where intermediate codebook tokens become overly concentrated, hindering the full utilization of generative retrieval methods. This paper analyses and addresses this problem by identifying data sparsity and long-tailed distribution as the primary causes. Through comprehensive experiments and detailed ablation studies, we analyze the impact of these factors on codebook utilization and data distribution. Our findings reveal that the “Hourglass” phenomenon substantially impacts the performance of RQ-SID in generative retrieval. We propose effective solutions to mitigate this issue, thereby significantly enhancing the effectiveness of generative retrieval in real-world E-commerce applications.
2001
Improving the Effectiveness of Information Retrieval with Clustering and Fusion
Jian Zhang | Jianfeng Gao | Ming Zhou | Jiaxing Wang
International Journal of Computational Linguistics & Chinese Language Processing, Volume 6, Number 1, February 2001: Special Issue on Natural Language Processing Researches in MSRA
Jian Zhang | Jianfeng Gao | Ming Zhou | Jiaxing Wang
International Journal of Computational Linguistics & Chinese Language Processing, Volume 6, Number 1, February 2001: Special Issue on Natural Language Processing Researches in MSRA
Search
Fix author
Co-authors
- Zhen Chen 3
- Zhaolong Xing 3
- Ke Zhang 3
- Zhida Jiang 2
- Lin Liu 2
- Zicheng Zhang 2
- Wang Binbin 1
- Xiaolong Chen 1
- Xusong Chen 1
- Zuxu Chen 1
- Jianfeng Gao 1
- Saixing Gao 1
- Guoqiang Gong 1
- Yuxing Han 1
- Chengzhi Huang 1
- Weizhe Huang 1
- Zhirui Kuai 1
- Li Kuang 1
- Jing Li 1
- Mingming Li 1
- Yan Li 1
- Haopeng Liu 1
- Pengzhang Liu 1
- Tongxuan Liu 1
- Xiyu Liu 1
- Liuxinyu 1
- Dadong Miao 1
- Qingyuan Sang 1
- Leqi Shen 1
- Junhua Shu 1
- Guoyu Tang 1
- Dongyue Wang 1
- Hao Wang 1
- Huimu Wang 1
- Songlin Wang 1
- Wenzhe Wang 1
- Xingyu Wang 1
- Deping Xiang 1
- Yuanhang Xiao 1
- Jin Xu 1
- Wenjiang Xu 1
- Yin Xu 1
- Hailong Yang (杨海龙) 1
- Pei Yang 1
- Baopeng Yuan 1
- Yuting Zeng 1
- Jian Zhang 1
- Yifeng Zhang 1
- Lingfeng Zhou 1
- Ming Zhou 1
- Jingwei Zhuo 1