Demonstration venue · read-only. Every page can be browsed; the buttons that would change it are switched off. Create an account to run TaxoReview on your own data. A simulation from public publication records: the people listed were not invited and did not agree to review.
? Tutorial for this page

Junyi Li 0001

dblp:28/6612-1 · DBLP profile ↗
← Back
32ranked-venue papers
12as first author
29since 2021 · last 2026
0009-0007-0480-5593ORCID · conflict

Domains — the database's venue-derived domains; a paper can count in several

Artificial intelligence and machine learning · 25 · 10 first-author · 23 since 2021Databases, data management, data science and information retrieval · 6 · 2 first-author · 4 since 2021Applied, interdisciplinary, general and emerging computing · 4 · 1 first-author · 3 since 2021Graphics, computer vision, multimedia, augmented reality and games · 2 · 1 first-author · 2 since 2021

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Artificial intelligence
20 papers
Language models and text generation · 57% Generative modeling · 9% Trustworthy machine learning · 8%
Databases, data mining, and information retrieval
6 papers
Information retrieval · 72% Recommender systems · 23% Knowledge graphs · 5%

Topics — the 30 heaviest of 62, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Natural language and speech › Language models and text generation
large language model
2.642025
Reasoning Models Hallucinate More: Factuality-Aware Reinforcement Learning for Large Reasoning Models · NeurIPS 2025
Exploring Context Window of Large Language Models via Decomposed Positional Vectors · NeurIPS 2024
The Dawn After the Dark: An Empirical Study on Factuality Hallucination in Large Language Models · ACL (1) 2024
Machine learning › Trustworthy machine learning
hallucination
2.332025
Reasoning Models Hallucinate More: Factuality-Aware Reinforcement Learning for Large Reasoning Models · NeurIPS 2025
The Dawn After the Dark: An Empirical Study on Factuality Hallucination in Large Language Models · ACL (1) 2024
HaluEval: A Large-Scale Hallucination Evaluation Benchmark for Large Language Models · EMNLP 2023
Machine learning › Generative modeling
diffusion model
1.222026
Beyond Fully Random Masking: Attention-Guided Denoising and Optimization for Diffusion Language Models · ACL (1) 2026
Learning to Imagine: Visually-Augmented Natural Language Generation · ACL (1) 2023
Information retrieval
retrieval-augmented generation
1.122026
REAR: A Relevance-Aware Retrieval-Augmented Framework for Open-Domain Question Answering · EMNLP 2024
MemSearch-o1: Empowering Large Language Models with Reasoning-Aligned Memory Growth in Agentic Search · ACL (1) 2026
Natural language and speech › Language models and text generation › LLM agents
agentic search
1.012026
MemSearch-o1: Empowering Large Language Models with Reasoning-Aligned Memory Growth in Agentic Search · ACL (1) 2026
Machine learning › Generative modeling › diffusion model
denoising
1.012026
Beyond Fully Random Masking: Attention-Guided Denoising and Optimization for Diffusion Language Models · ACL (1) 2026
Machine learning › Generative modeling › diffusion model › discrete diffusion model
diffusion language model
1.012026
Beyond Fully Random Masking: Attention-Guided Denoising and Optimization for Diffusion Language Models · ACL (1) 2026
Machine learning › Reinforcement learning › partially observable reinforcement learning › memory-based reinforcement learning
memory-augmented agent
1.012026
MemSearch-o1: Empowering Large Language Models with Reasoning-Aligned Memory Growth in Agentic Search · ACL (1) 2026
Natural language and speech › Language models and text generation › large language model training
post-training
1.012026
Beyond Fully Random Masking: Attention-Guided Denoising and Optimization for Diffusion Language Models · ACL (1) 2026
Information retrieval
agentic search
1.012026
To Search or Not to Search: Aligning the Decision Boundary of Deep Search Agents via Causal Intervention · WWW 2026
Information retrieval › information filtering › technology-assisted review
stopping criteria
1.012026
To Search or Not to Search: Aligning the Decision Boundary of Deep Search Agents via Causal Intervention · WWW 2026
Natural language and speech › Language models and text generation › text generation › domain-specific text generation
review generation
0.922021
Knowledge-based Review Generation by Coherence Enhanced Text Planning · SIGIR 2021
Generating Long and Informative Reviews with Aspect-Aware Coarse-to-Fine Decoding · ACL (1) 2019
Natural language and speech › Language models and text generation › text generation › knowledge-grounded generation
attributed text generation
0.912025
Think&Cite: Improving Attributed Text Generation with Self-Guided Tree Search and Progress Reward Modeling · ACL (1) 2025
Natural language and speech › Language models and text generation › large language model training
continual pre-training
0.912025
Mix-CPT: A Domain Adaptation Framework via Decoupling Knowledge Learning and Format Alignment · ICLR 2025
Machine learning › Efficient and distributed learning › distributed training
data parallel training
0.912025
Just Go Parallel: Improving the Multilingual Capabilities of Large Language Models · ACL (1) 2025
Machine learning › Transfer learning and domain adaptation
domain adaptation
0.912025
Mix-CPT: A Domain Adaptation Framework via Decoupling Knowledge Learning and Format Alignment · ICLR 2025
Natural language and speech › Language models and text generation › large language model reasoning
efficient reasoning
0.912025
Incentivizing Dual Process Thinking for Efficient Large Language Model Reasoning · NeurIPS 2025
Natural language and speech › Language models and text generation
hallucination mitigation
0.912025
Think&Cite: Improving Attributed Text Generation with Self-Guided Tree Search and Progress Reward Modeling · ACL (1) 2025
Natural language and speech › Language models and text generation
in-context learning
0.912025
Investigating the Pre-Training Dynamics of In-Context Learning: Task Recognition vs. Task Learning · ICLR 2025
Natural language and speech › Language models and text generation
large language model reasoning
0.912025
Incentivizing Dual Process Thinking for Efficient Large Language Model Reasoning · NeurIPS 2025
Natural language and speech › Language models and text generation
multilingual language models
0.912025
Just Go Parallel: Improving the Multilingual Capabilities of Large Language Models · ACL (1) 2025
Machine learning › Reinforcement learning
policy optimization
0.912025
Incentivizing Dual Process Thinking for Efficient Large Language Model Reasoning · NeurIPS 2025
Natural language and speech › Language models and text generation › large language model training
pretraining dynamics
0.912025
Investigating the Pre-Training Dynamics of In-Context Learning: Task Recognition vs. Task Learning · ICLR 2025
Natural language and speech › Language models and text generation › large language model
reasoning model
0.912025
Reasoning Models Hallucinate More: Factuality-Aware Reinforcement Learning for Large Reasoning Models · NeurIPS 2025
Machine learning › Reinforcement learning › reinforcement learning for NLP
reinforcement fine-tuning
0.912025
Reasoning Models Hallucinate More: Factuality-Aware Reinforcement Learning for Large Reasoning Models · NeurIPS 2025
Information retrieval › interactive information retrieval
conversational information seeking
0.912025
LLM-based Search Assistant with Holistically Guided MCTS for Intricate Information Seeking · SIGIR 2025
Recommender systems › interactive recommendation
conversational recommendation
0.912025
Search-Based Interaction For Conversation Recommendation via Generative Reward Model Based Simulated User · SIGIR 2025
Information retrieval › interactive information retrieval
search assistant
0.912025
LLM-based Search Assistant with Holistically Guided MCTS for Intricate Information Seeking · SIGIR 2025
Information retrieval › evaluation › user simulation
user simulator
0.912025
Search-Based Interaction For Conversation Recommendation via Generative Reward Model Based Simulated User · SIGIR 2025
Natural language and speech › Language models and text generation › language modeling › long-context language modeling
context window extension
0.812024
Exploring Context Window of Large Language Models via Decomposed Positional Vectors · NeurIPS 2024

Methods — techniques the papers use, named apart from their topics

large language model · 3.7reinforcement learning · 2.7reasoning-aligned memory · 2.0preference optimization · 2.0counterfactual trajectory analysis · 2.0causal intervention · 2.0supervised fine-tuning · 1.9fine-tuning · 1.3transformer · 1.2attention-guided denoising · 1.0reward modeling · 0.9prompt learning · 0.9monte carlo tree search · 0.9knowledge graph · 0.9instruction tuning · 0.9
YearPublicationVenuePosition
2026 Beyond Fully Random Masking: Attention-Guided Denoising and Optimization for Diffusion Language Models
abstract
Diffusion large language models (dLLMs) offer an efficient alternative to autoregressive models through parallel decoding, yet existing post-training methods largely rely on random masking strategies that overlook intrinsic token dependencies.In this work, we present an empirical analysis of attention in dLLMs and show that tokens attending more strongly to unmasked context exhibit greater generation stability and play a critical role in reasoning.Motivated by these findings, we propose AGDO, an attention-guided denoising and optimization framework that aligns both training and optimization with attention-derived dependencies.AGDO determines the denoising order based on attention structure and emphasizes attentioncritical tokens during supervised fine-tuning and reinforcement learning.Experiments on mathematical and coding benchmarks demonstrate that AGDO consistently improves reasoning performance, outperforming state-of-theart post-training methods for dLLMs.
Junyi Li 0001, Wayne Xin Zhao, Jinpeng Wang 0001, Ji-Rong Wen
ACL (1)2
2026 MemSearch-o1: Empowering Large Language Models with Reasoning-Aligned Memory Growth in Agentic Search
abstract
Sheng Zhang, Junyi Li, Yingyi Zhang, Pengyue Jia, Yichao Wang, Xiaowei Qian, Wenlin Zhang, Maolin Wang, Yong Liu, Xiangyu Zhao. Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2026.
Sheng Zhang 0028, Junyi Li 0001, Yingyi Zhang 0001, Pengyue Jia, Yichao Wang 0002, Wenlin Zhang 0001, Maolin Wang 0001, Yong Liu 0020, Xiangyu Zhao 0001
ACL (1)2
2026 To Search or Not to Search: Aligning the Decision Boundary of Deep Search Agents via Causal Intervention
abstract
Deep search agents, which autonomously iterate through multi-turn web-based reasoning, represent a promising paradigm for complex information-seeking tasks. However, current agents suffer from critical inefficiency: they conduct excessive searches as they cannot accurately judge when to stop searching and start answering. This stems from outcome-centric training that prioritize final results over the search process itself. We identify the root cause as misaligned decision boundaries, the threshold determining when accumulated information suffices to answer. This causes over-search (redundant searching despite sufficient knowledge) and under-search (premature termination yielding incorrect answers). To address these errors, we propose a comprehensive framework comprising two key components. First, we introduce causal intervention-based diagnosis that identifies boundary errors by comparing factual and counterfactual trajectories at each decision point. Second, we develop Decision Boundary Alignment for Deep Search agents (DAS), which constructs preference datasets from causal feedback and aligns policies via preference optimization. Experiments on public datasets demonstrate that decision boundary errors are pervasive across state-of-the-art agents. Our DAS method effectively calibrates these boundaries, mitigating both over-search and under-search to achieve substantial gains in accuracy and efficiency. Our code and data are publicly available at: https://github.com/Applied-Machine-Learning-Lab/WWW2026-DAS. © 2026 Owner/Author.
Wenlin Zhang 0001, Kuicai Dong, Junyi Li 0001, Yingyi Zhang 0001, Xiaopeng Li 0014, Pengyue Jia, Yi Wen 0001, Derong Xu, Maolin Wang 0001, Yichao Wang 0002, Yong Liu 0020, Xiangyu Zhao 0001
WWW3
2026 APL-LLM: adaptive pseudo-labeling with large language models for few-shot node classification
Junyi Li 0001, Chenweinan Jiang, Daixin Wang, Guo Ye, Libang Zhang, Huimei He, Binbin Hu, Zhiqiang Zhang 0012, Fuzhen Zhuang
Frontiers Comput. Sci.1
2026 A Survey of Large Language Models
abstract
Abstract The rapid evolution of large language models (LLMs) has driven a transformative shift in artificial intelligence (AI), reshaping both research paradigms and practical applications. Distinguished from their predecessors by unprecedented scale and advanced capabilities, LLMs necessitate new frameworks for understanding their development, behavior, and societal impact. This survey systematically reviews recent advancements in LLM techniques across four key dimensions: (1) pre-training methodologies, which establish core model capabilities through large-scale self-supervised training, architectural innovations, and data curation strategies; (2) post-training techniques, including supervised fine-tuning and reinforcement learning, which adapt foundational models to downstream tasks and enhance their alignment and safety; (3) utilization strategies, such as in-context learning, prompt engineering, and agentic reasoning, that optimize real-world deployment and enable effective interaction with external environments; and (4) evaluation methods, encompassing benchmarks for key ability dimensions such as core language capabilities, reasoning, and safety, which support comprehensive and reliable assessment of model performance. Additionally, we identify critical research issues, including those concerning theoretical foundations, efficient scaling, alignment, and agentic capability, and highlight the open challenges they present. By synthesizing state-of-the-art insights and emerging trends, this survey aims to provide a systematic and comprehensive framework for understanding the trajectory, current limitations, and future directions of LLM progress.
Wayne Xin Zhao, Kun Zhou 0002, Junyi Li 0001, Zican Dong, Yupeng Hou, Beichen Zhang 0003, Yingqian Min, Junjie Zhang 0009, Peiyu Liu 0002, Xiaolei Wang 0005, Yifan Du 0002, Chen Yang 0032, Zhipeng Chen 0001, Jinhao Jiang, Ruiyang Ren, Yifan Li 0009, Xinyu Tang 0004, Zikang Liu 0001, Jian-Yun Nie, Ji-Rong Wen
Frontiers Comput. Sci.3
2025 Think&Cite: Improving Attributed Text Generation with Self-Guided Tree Search and Progress Reward Modeling
abstract
Despite their outstanding capabilities, large language models (LLMs) are prone to hallucination and producing factually incorrect information. This challenge has spurred efforts in attributed text generation, which prompts LLMs to generate content with supporting evidence. In this paper, we propose a novel framework, called Think&Cite, and formulate attributed text generation as a multi-step reasoning problem integrated with search. Specifically, we propose Self-Guided Monte Carlo Tree Search (SG-MCTS), which capitalizes on the self-reflection capability of LLMs to reason about the intermediate states of MCTS for guiding the tree expansion process. To provide reliable and comprehensive feedback, we introduce Progress Reward Modeling to measure the progress of tree search from the root to the current state from two aspects, i.e., generation and attribution progress. We conduct extensive experiments on three datasets and the results show that our approach significantly outperforms baseline approaches.
Junyi Li 0001, Hwee Tou Ng
ACL (1)1
2025 Just Go Parallel: Improving the Multilingual Capabilities of Large Language Models
abstract
Large language models (LLMs) have demonstrated impressive translation capabilities even without being explicitly trained on parallel data.This remarkable property has led some to believe that parallel data is no longer necessary for building multilingual language models.While some attribute this to the emergent abilities of LLMs due to scale, recent work suggests that it is actually caused by incidental bilingual signals present in the training data.Various methods have been proposed to maximize the utility of parallel data to enhance the multilingual capabilities of multilingual encoder-based and encoder-decoder language models.However, some decoder-based LLMs opt to ignore parallel data instead.In this work, we conduct a systematic study on the impact of adding parallel data on LLMs' multilingual capabilities, focusing specifically on translation and multilingual common-sense reasoning.Through controlled experiments, we demonstrate that parallel data can significantly improve LLMs' multilingual capabilities.1
Muhammad Reza Qorib, Junyi Li 0001, Hwee Tou Ng
ACL (1)2
2025 Investigating the Pre-Training Dynamics of In-Context Learning: Task Recognition vs. Task Learning
abstract
The emergence of in-context learning (ICL) is potentially attributed to two major abilities: task recognition (TR) for recognizing the task from demonstrations and utilizing pre-trained priors, and task learning (TL) for learning from demonstrations. However, relationships between the two abilities and how such relationships affect the emergence of ICL is unclear. In this paper, we take the first step by examining the pre-training dynamics of the emergence of ICL. With carefully designed metrics, we find that these two abilities are, in fact, competitive during pre-training. Moreover, we observe a negative correlation between the competition and the performance of ICL. Further analysis of common pre-training factors (i.e., model size, dataset size, and data curriculum) demonstrates possible ways to regulate the competition. Based on these insights, we propose a simple yet effective method to better integrate these two abilities for ICL at inference time. Through adaptive ensemble learning, the performance of ICL can be significantly boosted, enabling two small models to outperform a larger one with more than twice the parameters.
Xiaolei Wang 0005, Xinyu Tang 0004, Junyi Li 0001, Wayne Xin Zhao, Ji-Rong Wen
ICLR3
2025 Mix-CPT: A Domain Adaptation Framework via Decoupling Knowledge Learning and Format Alignment
abstract
Adapting large language models (LLMs) to specialized domains typically requires domain-specific corpora for continual pre-training to facilitate knowledge memorization and related instructions for fine-tuning to apply this knowledge. However, this method may lead to inefficient knowledge memorization due to a lack of awareness of knowledge utilization during the continual pre-training and demands LLMs to simultaneously learn knowledge utilization and format alignment with divergent training objectives during the fine-tuning. To enhance the domain adaptation of LLMs, we revise this process and propose a new domain adaptation framework including domain knowledge learning and general format alignment, called \emph{Mix-CPT}. Specifically, we first conduct a knowledge mixture continual pre-training that concurrently focuses on knowledge memorization and utilization. To avoid catastrophic forgetting, we further propose a logit swap self-distillation constraint. By leveraging the knowledge and capabilities acquired during continual pre-training, we then efficiently perform instruction tuning and alignment with a few general training samples to achieve format alignment. Extensive experiments show that our proposed \emph{Mix-CPT} framework can simultaneously improve the task-solving capabilities of LLMs on the target and general domains.
Jinhao Jiang, Junyi Li 0001, Wayne Xin Zhao, Yang Song 0021, Tao Zhang 0070, Ji-Rong Wen
ICLR2
2025 RAG-Star: Enhancing Deliberative Reasoning with Retrieval Augmented Verification and Refinement
abstract
Jinhao Jiang, Jiayi Chen, Junyi Li, Ruiyang Ren, Shijie Wang, Xin Zhao, Yang Song, Tao Zhang. 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). 2025.
Jinhao Jiang, Jiayi Chen 0005, Junyi Li 0001, Ruiyang Ren, Wayne Xin Zhao, Yang Song 0021, Tao Zhang 0070
NAACL (Long Papers)3
2025 Incentivizing Dual Process Thinking for Efficient Large Language Model Reasoning
abstract
Large reasoning models (LRMs) have demonstrated strong performance on complex reasoning tasks, but often suffer from overthinking, generating redundant content regardless of task difficulty. Inspired by the dual process theory in cognitive science, we propose Adaptive Cognition Policy Optimization (ACPO), a reinforcement learning framework that enables LRMs to achieve efficient reasoning through adaptive cognitive allocation and dynamic system switch. ACPO incorporates two key components: (1) introducing system-aware reasoning tokens to explicitly represent the thinking modes thereby making the model's cognitive process transparent, and (2) integrating online difficulty estimation and token length budget to guide adaptive system switch and reasoning during reinforcement learning. To this end, we propose a two-stage training strategy. The first stage begins with supervised fine-tuning to cold start the model, enabling it to generate reasoning paths with explicit thinking modes. In the second stage, we apply ACPO to further enhance adaptive system switch for difficulty-aware reasoning. Experimental results demonstrate that ACPO effectively reduces redundant reasoning while adaptively adjusting cognitive allocation based on task complexity, achieving efficient hybrid reasoning.
Xiaoxue Cheng, Junyi Li 0001, Zhenduo Zhang, Xinyu Tang 0004, Wayne Xin Zhao, Xinyu Kong, Zhiqiang Zhang 0012
NeurIPS2
2025 Reasoning Models Hallucinate More: Factuality-Aware Reinforcement Learning for Large Reasoning Models
abstract
Large language models (LLMs) have significantly advanced in reasoning tasks through reinforcement learning (RL) optimization, achieving impressive capabilities across various challenging benchmarks. However, our empirical analysis reveals a critical drawback: reasoning-oriented RL fine-tuning significantly increases the prevalence of hallucinations. We theoretically analyze the RL training dynamics, identifying high-variance gradient, entropy-induced randomness, and susceptibility to spurious local optima as key factors leading to hallucinations. To address this drawback, we propose Factuality-aware Step-wise Policy Optimization (FSPO), an innovative RL fine-tuning algorithm incorporating explicit factuality verification at each reasoning step. FSPO leverages automated verification against given evidence to dynamically adjust token-level advantage values, incentivizing factual correctness throughout the reasoning process. Experiments across mathematical reasoning and hallucination benchmarks using Qwen2.5 and Llama models demonstrate that FSPO effectively reduces hallucinations while enhancing reasoning accuracy, substantially improving both reliability and performance.
Junyi Li 0001, Hwee Tou Ng
NeurIPS1
2025 Search-Based Interaction For Conversation Recommendation via Generative Reward Model Based Simulated User
abstract
Conversational recommendation systems (CRSs) use multi-turn interaction to capture user preferences and provide personalized recommendations. A fundamental challenge in CRSs lies in effectively understanding user preferences from conversations. Previous research primarily focuses on the issue of insufficient contextual information in conversations. They address this by introducing external knowledge sources, such as knowledge graphs, large language models (LLMs), and conversational recommendation corpora. Based on this, they design specific alignment strategies (e.g., prompt learning and instruction tuning) to integrate such knowledge for user preference understanding and item recommendation. However, user preferences can be multifaceted and complex, posing significant challenges for accurate recommendations even with access to abundant external knowledge. While interaction with users can clarify their true preferences, frequent user involvement may lead to a degraded user experience.
Xiaolei Wang 0005, Chunxuan Xia, Junyi Li 0001, Fanzhe Meng, Jinpeng Wang 0001, Wayne Xin Zhao, Ji-Rong Wen
SIGIR3
2025 LLM-based Search Assistant with Holistically Guided MCTS for Intricate Information Seeking
abstract
In the era of vast digital information, the sheer volume and heterogeneity of available information present significant challenges for intricate information seeking. Users frequently face multistep web search tasks that involve navigating vast and varied data sources. This complexity demands every step remains comprehensive, accurate, and relevant. However, traditional search methods often struggle to balance the need for localized precision with the broader context required for holistic understanding, leaving critical facets of intricate queries underexplored. In this paper, we introduce an LLM-based search assistant that adopts a new information seeking paradigm with holistically guided Monte Carlo tree search (HG-MCTS). We reformulate the task as a progressive information collection process with a knowledge memory and unite an adaptive checklist with multi-perspective reward modeling in MCTS. The adaptive checklist provides explicit sub-goals to guide the MCTS process toward comprehensive coverage of complex user queries. Simultaneously, our multi-perspective reward modeling offers both exploration and retrieval rewards, along with progress feedback that tracks completed and remaining sub-goals, refining the checklist as the tree search progresses. By striking a balance between localized tree expansion and global guidance, HG-MCTS reduces redundancy in search paths and ensures that all crucial aspects of an intricate query are properly addressed. Extensive experiments on real-world intricate information seeking tasks demonstrate that HG-MCTS acquires thorough knowledge collections and delivers more accurate final responses compared with existing baselines.
Ruiyang Ren, Yuhao Wang 0007, Junyi Li 0001, Jinhao Jiang, Wayne Xin Zhao, Wenjie Wang 0007, Tat-Seng Chua
SIGIR3
2024 The Dawn After the Dark: An Empirical Study on Factuality Hallucination in Large Language Models
abstract
Junyi Li, Jie Chen, Ruiyang Ren, Xiaoxue Cheng, Xin Zhao, Jian-Yun Nie, Ji-Rong Wen. Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2024.
Junyi Li 0001, Jie Chen 0007, Ruiyang Ren, Xiaoxue Cheng, Wayne Xin Zhao, Jian-Yun Nie, Ji-Rong Wen
ACL (1)1
2024 ChainLM: Empowering Large Language Models with Improved Chain-of-Thought Prompting
abstract
Chain-of-Thought (CoT) prompting can enhance the reasoning capabilities of large language models (LLMs), establishing itself as a primary approach to solving complex reasoning tasks. Existing CoT synthesis approaches usually focus on simpler reasoning tasks and thus result in low-quality and inconsistent CoT prompts. In response to this challenge, we present an empirical investigation of CoT prompting and introduce CoTGenius, a novel framework designed for the automatic generation of superior CoT prompts. CoTGenius is developed based on three major evolution strategies, i.e., complicate, diversify, and specify—alongside two filtering mechanisms: evolutionary success judgement and correctness verification. We further employ CoTGenius to create an extensive CoT dataset, and subsequently fine-tune the Llama 2-Chat 7B and 13B models on this dataset. We call the resulting model ChainLM. To deal with the cumulative error issue in reasoning steps, we propose a step-level debating method, wherein multiple debaters discuss each reasoning step to arrive at the correct answer. Extensive experiments demonstrate that our ChainLM models exhibit enhanced proficiency in addressing a spectrum of complex reasoning problems compared to existing models. In addition, we conduct an in-depth analysis of the impact of data categories within CoTGenius on the model performance. We release our dataset and code at https://github.com/RUCAIBox/ChainLM.
Xiaoxue Cheng, Junyi Li 0001, Wayne Xin Zhao, Ji-Rong Wen
LREC/COLING2
2024 BAMBOO: A Comprehensive Benchmark for Evaluating Long Text Modeling Capacities of Large Language Models
abstract
Large language models (LLMs) have achieved dramatic proficiency over NLP tasks with normal length. Recently, multiple studies have committed to extending the context length and enhancing the long text modeling capabilities of LLMs. To comprehensively evaluate the long context ability of LLMs, we propose BAMBOO, a multi-task long context benchmark. BAMBOO has been designed with four principles: comprehensive capacity evaluation, avoidance of data contamination, accurate automatic evaluation, and different length levels. It consists of 10 datasets from 5 different long text understanding tasks, i.e., question answering, hallucination detection, text sorting, language modeling, and code completion, to cover various domains and core capacities of LLMs. We conduct experiments with five widely-used long-context models and further discuss five key questions for long text research. In the end, we discuss problems of current long-context models and point out future directions for enhancing long text modeling capacities. We release our data, prompts, and code at https://anonymous.4open.science/r/BAMBOO/.
Zican Dong, Junyi Li 0001, Wayne Xin Zhao, Ji-Rong Wen
LREC/COLING3
2024 Small Agent Can Also Rock! Empowering Small Language Models as Hallucination Detector
abstract
Hallucination detection is a challenging task for large language models (LLMs), and existing studies heavily rely on powerful closed-source LLMs such as GPT-4.In this paper, we propose an autonomous LLM-based agent framework, called HaluAgent, which enables relatively smaller LLMs 1 (e.g., Baichuan2-Chat 7B) to actively select suitable tools for detecting multiple hallucination types such as text, code, and mathematical expression.In HaluAgent, we integrate the LLM, multi-functional toolbox, and design a fine-grained three-stage detection framework along with memory mechanism.To facilitate the effectiveness of HaluAgent, we leverage existing Chinese and English datasets to synthesize detection trajectories for fine-tuning, which endows HaluAgent with the capability for bilingual hallucination detection.Extensive experiments demonstrate that only using 2K samples for tuning LLMs, HaluAgent can perform hallucination detection on various types of tasks and datasets, achieving performance comparable to or even higher than GPT-4 without tool enhancements on both indomain and out-of-domain datasets.We release our dataset and code at https://github.com/ RUCAIBox/HaluAgent.
Xiaoxue Cheng, Junyi Li 0001, Wayne Xin Zhao, Di Zhang 0026, Kun Gai, Ji-Rong Wen
EMNLP2
2024 REAR: A Relevance-Aware Retrieval-Augmented Framework for Open-Domain Question Answering
abstract
Considering the limited internal parametric knowledge, retrieval-augmented generation (RAG) has been widely used to extend the knowledge scope of large language models (LLMs).Despite the extensive efforts on RAG research, in existing methods, LLMs cannot precisely assess the relevance of retrieved documents, thus likely leading to misleading or even incorrect utilization of external knowledge (i.e., retrieved documents).To address this issue, in this paper, we propose REAR, a RElevance-Aware Retrievalaugmented approach for open-domain question answering (QA).As the key motivation, we aim to enhance the self-awareness regarding the reliability of external knowledge for LLMs, so as to adaptively utilize external knowledge in RAG systems.Specially, we develop a novel architecture for LLM-based RAG systems, by incorporating a specially designed assessment module that precisely assesses the relevance of retrieved documents.Furthermore, we propose an improved training method based on bigranularity relevance fusion and noise-resistant training.By combining the improvements in both architecture and training, our proposed REAR can better utilize external knowledge by effectively perceiving the relevance of retrieved documents.Experiments on four opendomain QA tasks show that REAR significantly outperforms previous a number of competitive RAG approaches.
Yuhao Wang 0007, Ruiyang Ren, Junyi Li 0001, Wayne Xin Zhao, Jing Liu 0022, Ji-Rong Wen
EMNLP3
2024 Exploring Context Window of Large Language Models via Decomposed Positional Vectors
abstract
Transformer-based large language models (LLMs) typically have a limited context window, resulting in significant performance degradation when processing text beyond the length of the context window. Extensive studies have been proposed to extend the context window and achieve length extrapolation of LLMs, but there is still a lack of in-depth interpretation of these approaches. In this study, we explore the positional information within and beyond the context window for deciphering the underlying mechanism of LLMs. By using a mean-based decomposition method, we disentangle positional vectors from hidden states of LLMs and analyze their formation and effect on attention. Furthermore, when texts exceed the context window, we analyze the change of positional vectors in two settings, i.e., direct extrapolation and context window extension. Based on our findings, we design two training-free context window extension methods, positional vector replacement and attention window extension. Experimental results show that our methods can effectively extend the context window length.
Zican Dong, Junyi Li 0001, Xin Men, Wayne Xin Zhao, Bingning Wang, Zhen Tian 0001, Weipeng Chen, Ji-Rong Wen
NeurIPS2
2023 Learning to Imagine: Visually-Augmented Natural Language Generation
abstract
People often imagine relevant scenes to aid in the writing process.In this work, we aim to utilize visual information for composition in the same manner as humans.We propose a method, LIVE, that makes pre-trained language models (PLMs) Learn to Imagine for Visuallyaugmented natural language gEneration.First, we imagine the scene based on the text: we use a diffusion model to synthesize high-quality images conditioned on the input texts.Second, we use CLIP to determine whether the text can evoke the imagination in a posterior way.Finally, our imagination is dynamic, and we conduct synthesis for each sentence rather than generate only one image for an entire paragraph.Technically, we propose a novel plug-and-play fusion layer to obtain visually-augmented representations for each text.Our vision-text fusion layer is compatible with Transformerbased architecture.We have conducted extensive experiments on four generation tasks using BART and T5, and the automatic results and human evaluation demonstrate the effectiveness of our proposed method.We will release the code, model, and data at the link: https://github.com/RUCAIBox/LIVE.
Yifan Du 0002, Junyi Li 0001, Wayne Xin Zhao, Ji-Rong Wen
ACL (1)4
2023 HaluEval: A Large-Scale Hallucination Evaluation Benchmark for Large Language Models
abstract
Large language models (LLMs), such as Chat-GPT, are prone to generate hallucinations, i.e., content that conflicts with the source or cannot be verified by the factual knowledge.To understand what types of content and to which extent LLMs are apt to hallucinate, we introduce the Hallucination Evaluation benchmark for Large Language Models (HaluEval), a large collection of generated and human-annotated hallucinated samples for evaluating the performance of LLMs in recognizing hallucination.To generate these samples automatically, we propose a two-stage framework, i.e., samplingthen-filtering.Besides, we hire some human labelers to annotate the hallucinations in Chat-GPT responses.The empirical results suggest that ChatGPT is likely to generate hallucinated content related to specific topics by fabricating unverifiable information (i.e., about 19.5% responses).Moreover, existing LLMs face great challenges in recognizing the hallucinations in texts.However, our experiments also prove that providing external knowledge or adding reasoning steps can help LLMs recognize hallucinations.Our benchmark can be accessed at https://github.com/RUCAIBox/HaluEval.
Junyi Li 0001, Xiaoxue Cheng, Wayne Xin Zhao, Jian-Yun Nie, Ji-Rong Wen
EMNLP1
2022 Context-Tuning: Learning Contextualized Prompts for Natural Language Generation
abstract
Recently, pretrained language models (PLMs) have had exceptional success in language generation. To leverage the rich knowledge encoded by PLMs, a simple yet powerful paradigm is to use prompts in the form of either discrete tokens or continuous embeddings. In existing studies, these prompting methods are typically independent of the inputs, lacking sufficient consideration of input semantics. To address this issue, we propose a novel continuous prompting approach, called context-tuning, to fine-tuning PLMs for natural language generation. Firstly, the prompts are derived based on the input text to elicit useful knowledge from PLMs for generation. We refer to such prompts as contextualized prompts. Secondly, we use continuous inverse prompting to improve the process of natural language generation by modeling an inverse generation process from output to input, making the generated text more relevant to the inputs. Furthermore, we utilize a lightweight context-tuning method that fine-tunes only 0.12% of the parameters while maintaining good performance. Our code is publicly available at https://github.com/RUCAIBox/Context-Tuning.
Junyi Li 0001, Wayne Xin Zhao, Ji-Rong Wen
COLING2
2022 ELMER: A Non-Autoregressive Pre-trained Language Model for Efficient and Effective Text Generation
abstract
We study the text generation task under the approach of pre-trained language models (PLMs).Typically, an auto-regressive (AR) method is adopted for generating texts in a token-bytoken manner.Despite many advantages of AR generation, it usually suffers from inefficient inference.Therefore, non-autoregressive (NAR) models are proposed to generate all target tokens simultaneously.However, NAR models usually generate texts of lower quality due to the absence of token dependency in the output text.In this paper, we propose ELMER: an Efficient and effective PLM for NAR tExt geneRation to explicitly model the token dependency during NAR generation.By leveraging the early exit technique, ELMER enables the token generations at different layers, according to their prediction confidence (a more confident token will exit at a lower layer).Besides, we propose a novel pre-training objective, Layer Permutation Language Modeling, to pre-train ELMER by permuting the exit layer for each token in sequences.Experiments on three text generation tasks show that ELMER significantly outperforms NAR models and further narrows the performance gap with AR PLMs (e.g., ELMER (29.92) vs BART (30.61)ROUGE-L in XSUM) while achieving over 10 times inference speedup.
Junyi Li 0001, Wayne Xin Zhao, Jian-Yun Nie, Ji-Rong Wen
EMNLP1
2022 A Survey of Vision-Language Pre-Trained Models
abstract
As transformer evolves, pre-trained models have advanced at a breakneck pace in recent years. They have dominated the mainstream techniques in natural language processing (NLP) and computer vision (CV). How to adapt pre-training to the field of Vision-and-Language (V-L) learning and improve downstream task performance becomes a focus of multimodal learning. In this paper, we review the recent progress in Vision-Language Pre-Trained Models (VL-PTMs). As the core content, we first briefly introduce several ways to encode raw images and texts to single-modal embeddings before pre-training. Then, we dive into the mainstream architectures of VL-PTMs in modeling the interaction between text and image representations. We further present widely-used pre-training tasks, and then we introduce some common downstream tasks. We finally conclude this paper and present some promising research directions. Our survey aims to provide researchers with synthesis and pointer to related research.
Yifan Du 0002, Zikang Liu 0001, Junyi Li 0001, Wayne Xin Zhao
IJCAI3
2022 ElitePLM: An Empirical Study on General Language Ability Evaluation of Pretrained Language Models
abstract
Junyi Li, Tianyi Tang, Zheng Gong, Lixin Yang, Zhuohao Yu, Zhipeng Chen, Jingyuan Wang, Xin Zhao, Ji-Rong Wen. Proceedings of the 2022 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies. 2022.
Junyi Li 0001, Zheng Gong 0001, Lixin Yang 0005, Zhuohao Yu 0001, Zhipeng Chen 0001, Jingyuan Wang 0001, Wayne Xin Zhao, Ji-Rong Wen
NAACL-HLT1
2022 Learning to Transfer Prompts for Text Generation
abstract
Junyi Li, Tianyi Tang, Jian-Yun Nie, Ji-Rong Wen, Xin Zhao. Proceedings of the 2022 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies. 2022.
Junyi Li 0001, Jian-Yun Nie, Ji-Rong Wen, Wayne Xin Zhao
NAACL-HLT1
2021 Pretrained Language Model for Text Generation: A Survey
abstract
Text generation has become one of the most important yet challenging tasks in natural language processing (NLP). The resurgence of deep learning has greatly advanced this field by neural generation models, especially the paradigm of pretrained language models (PLMs). In this paper, we present an overview of the major advances achieved in the topic of PLMs for text generation. As the preliminaries, we present the general task definition and briefly describe the mainstream architectures of PLMs for text generation. As the core content, we discuss how to adapt existing PLMs to model different input data and satisfy special properties in the generated text. We further summarize several important fine-tuning strategies for text generation. Finally, we present several future directions and conclude this paper. Our survey aims to provide text generation researchers a synthesis and pointer to related research.
Junyi Li 0001, Wayne Xin Zhao, Ji-Rong Wen
IJCAI1
2021 Knowledge-based Review Generation by Coherence Enhanced Text Planning
abstract
As a natural language generation task, it is challenging to generate informative and coherent review text. In order to enhance the informativeness of the generated text, existing solutions typically learn to copy entities or triples from knowledge graphs (KGs). However, they lack overall consideration to select and arrange the incorporated knowledge, which tends to cause text incoherence. To address the above issue, we focus on improving entity-centric coherence of the generated reviews by leveraging the semantic structure of KGs. In this paper, we propose a novel Coherence Enhanced Text Planning model (CETP) based on knowledge graphs (KGs) to improve both global and local coherence for review generation. The proposed model learns a two-level text plan for generating a document: (1) the document plan is modeled as a sequence of sentence plans in order, and (2) the sentence plan is modeled as an entity-based subgraph from KG. Local coherence can be naturally enforced by KG subgraphs through intra-sentence correlations between entities. For global coherence, we design a hierarchical self-attentive architecture with both subgraph- and node-level attention to enhance the correlations between subgraphs. To our knowledge, we are the first to utilize a KG-based text planning model to enhance text coherence for review generation. Extensive experiments on three datasets confirm the effectiveness of our model on improving the content coherence of generated texts.
Junyi Li 0001, Wayne Xin Zhao, Zhicheng Wei, Nicholas Jing Yuan, Ji-Rong Wen
SIGIR1
2020 Knowledge-Enhanced Personalized Review Generation with Capsule Graph Neural Network
abstract
Personalized review generation (PRG) aims to automatically produce review text reflecting user preference, which is a challenging natural language generation task. Most of previous studies do not explicitly model factual description of products, tending to generate uninformative content. Moreover, they mainly focus on word-level generation, but cannot accurately reflect more abstractive user preference in multiple aspects. To address the above issues, we propose a novel knowledgeenhanced PRG model based on capsule graph neural network (CapsGNN). We first construct a heterogeneous knowledge graph (HKG) for utilizing rich item attributes. We adopt Caps-GNN to learn graph capsules for encoding underlying characteristics from the HKG. Our generation process contains two major steps, namely aspect sequence generation and sentence generation. First, based on graph capsules, we adaptively learn aspect capsules for inferring the aspect sequence. Then, conditioned on the inferred aspect label, we design a graph-based copy mechanism to generate sentences by incorporating related entities or words from HKG. To our knowledge, we are the first to utilize knowledge graph for the PRG task. The incorporated KG information is able to enhance user preference at both aspect and word levels. Extensive experiments on three real-world datasets have demonstrated the effectiveness of our model on the PRG task.
Junyi Li 0001, Siqing Li, Wayne Xin Zhao, Gaole He, Zhicheng Wei, Nicholas Jing Yuan, Ji-Rong Wen
CIKM1
2020 Mining Implicit Entity Preference from User-Item Interaction Data for Knowledge Graph Completion via Adversarial Learning
abstract
The task of Knowledge Graph Completion (KGC) aims to automatically infer the missing fact information in Knowledge Graph (KG). In this paper, we take a new perspective that aims to leverage rich user-item interaction data (user interaction data for short) for improving the KGC task. Our work is inspired by the observation that many KG entities correspond to online items in application systems. However, the two kinds of data sources have very different intrinsic characteristics, and it is likely to hurt the original performance using simple fusion strategy.
Gaole He, Junyi Li 0001, Wayne Xin Zhao, Peiju Liu, Ji-Rong Wen
WWW2
2019 Generating Long and Informative Reviews with Aspect-Aware Coarse-to-Fine Decoding
abstract
Generating long and informative review text is a challenging natural language generation task.Previous work focuses on word-level generation, neglecting the importance of topical and syntactic characteristics from natural languages.In this paper, we propose a novel review generation model by characterizing an elaborately designed aspect-aware coarse-tofine generation process.First, we model the aspect transitions to capture the overall content flow.Then, to generate a sentence, an aspectaware sketch will be predicted using an aspectaware decoder.Finally, another decoder fills in the semantic slots by generating corresponding words.Our approach is able to jointly utilize aspect semantics, syntactic sketch, and context information.Extensive experiments results have demonstrated the effectiveness of the proposed model.
Junyi Li 0001, Wayne Xin Zhao, Ji-Rong Wen, Yang Song 0021
ACL (1)1