← Search

Shuaiqiang Wang

41 accepted papers

2026

AdaFuse: Accelerating Dynamic Adapter Inference via Token-Level Pre-Gating and Fused Kernel Optimization

AAAI 2026technical

The integration of dynamic, sparse structures like Mixture-of-Experts (MoE) with parameter-efficient adapters (e.g., LoRA) is a powerful technique for enhancing Large Language Models (LLMs). However, this architectural enhancement comes at a steep cost: despite minimal increases in computational loa

Cited by 0SourcePDFScholar
2026

Beyond Step Pruning: Information Theory Based Step-level Optimization for Self-Refining Large Language Models

AAAI 2026technical

Large language models (LLMs) have shown impressive capabilities in natural language tasks, yet they continue to struggle with multi-step mathematical reasoning, where correctness depends on a precise chain of intermediate steps. Preference optimization methods such as Direct Preference Optimization

Cited by 0SourcePDFScholar
2026

CurES: From Gradient Analysis to Efficient Curriculum Learning for Reasoning LLMs

ICLR 2026poster

Curriculum learning plays a crucial role in enhancing the training efficiency of large language models (LLMs) on reasoning tasks. However, existing methods often fail to adequately account for variations in prompt difficulty or rely on simplistic filtering mechanisms to select prompt datasets within…

Cited by 0SourcecodeScholar
2026

Efficient Thought Space Exploration Through Strategic Intervention

AAAI 2026technical

While large language models (LLMs) demonstrate emerging reasoning capabilities, current inference-time expansion methods incur prohibitive computational costs through exhaustive sampling. Through analyzing decoding trajectories, we observe that most next-token predictions align well with the golden

Cited by 0SourcePDFScholar
2026

Solving the Granularity Mismatch: Hierarchical Preference Learning for Long-Horizon LLM Agents

ICLR 2026poster

Large Language Models (LLMs) as autonomous agents are increasingly tasked with solving complex, long-horizon problems. Aligning these agents via preference-based methods like Direct Preference Optimization (DPO) is a promising direction, yet it faces a critical granularity mismatch. Trajectory-lev…

Cited by 0SourceScholar
2026

Thinking Forward and Backward: Multi-Objective Reinforcement Learning for Retrieval-Augmented Reasoning

AAAI 2026technical

Retrieval-augmented generation (RAG) has proven to be effective in mitigating hallucinations in large language models, yet its effectiveness remains limited in complex, multi-step reasoning scenarios. Recent efforts have incorporated search-based interactions into RAG, enabling iterative reasoning w

Cited by 0SourcePDFScholar
2026

ZeroGR: A Generalizable and Scalable Framework for Zero-Shot Generative Retrieval

ICLR 2026poster

Generative retrieval (GR) reformulates information retrieval (IR) by framing it as the generation of document identifiers (docids), thereby enabling an end-to-end optimization and seamless integration with generative language models (LMs). Despite notable progress under supervised training, GR still…

Cited by 0SourcecodeScholar
2025

CoRanking: Collaborative Ranking with Small and Large Ranking Agents

EMNLP 2025

Listwise ranking based on Large Language Models (LLMs) has achieved state-of-the-art performance in Information Retrieval (IR).However, their effectiveness often depends on LLMs with massive parameter scales and computationally expensive sliding window processing, leading to substantial efficiency b

2025

Divide-Then-Aggregate: An Efficient Tool Learning Method via Parallel Tool Invocation

ACL 2025long

While Large Language Models (LLMs) demonstrate remarkable capabilities, their ability to autonomously execute complex real-world tasks remains limited. Accordingly, tool learning has emerged to enable LLMs to effectively leverage external tools to extend their capabilities. Current tool-learning par…

2025

Enhancing Retrieval-Augmented Generation via Evidence Tree Search

ACL 2025long

Retrieval-Augmented Generation (RAG) is widely used to enhance Large Language Models (LLMs) by grounding responses in external knowledge. However, in real-world applications, retrievers often return lengthy documents with redundant or irrelevant content, confusing downstream readers. While evidence…

Cited by 0SourcePDFScholar
2025

From Exploration to Mastery: Enabling LLMs to Master Tools via Self-Driven Interactions

ICLR 2025oral

Tool learning enables Large Language Models (LLMs) to interact with external environments by invoking tools, serving as an effective strategy to mitigate the limitations inherent in their pre-training data. In this process, tool documentation plays a crucial role by providing usage instructions for…

2025

Improving Retrieval-Augmented Generation through Multi-Agent Reinforcement Learning

NeurIPS 2025poster

Retrieval-augmented generation (RAG) is widely utilized to incorporate external knowledge into large language models, thereby enhancing factuality and reducing hallucinations in question-answering (QA) tasks. A standard RAG pipeline consists of several components, such as query rewriting, document r…

Cited by 0SourcecodeScholar
2025

Knowledge Graph Retrieval-Augmented Generation for LLM-based Recommendation

ACL 2025long

Recommender systems have become increasingly vital in our daily lives, helping to alleviate the problem of information overload across various user-oriented online services. The emergence of Large Language Models (LLMs) has yielded remarkable achievements, demonstrating their potential for the devel…

Cited by 0SourcePDFScholar
2025

LLMs + Persona-Plug = Personalized LLMs

ACL 2025long

Personalization plays a critical role in numerous language tasks and applications, since users with the same requirements may prefer diverse outputs based on their interests. This has led to the development of various personalized approaches aimed at adapting large language models (LLMs) to generate…

2025

Mitigating Hallucinations in Large Vision-Language Models via Entity-Centric Multimodal Preference Optimization

EMNLP 2025

Large Visual Language Models (LVLMs) have demonstrated impressive capabilities across multiple tasks. However, their trustworthiness is often challenged by hallucinations, which can be attributed to the modality misalignment and the inherent hallucinations of their underlying Large Language Models (

2025

PA-RAG: RAG Alignment via Multi-Perspective Preference Optimization

NAACL 2025long

The emergence of Retrieval-augmented generation (RAG) has alleviated the issues of outdated and hallucinatory content in the generation of large language models (LLMs), yet it still reveals numerous limitations. When a general-purpose LLM serves as the RAG generator, it often suffers from inadequate…

2025

Reasoning-to-Defend: Safety-Aware Reasoning Can Defend Large Language Models from Jailbreaking

EMNLP 2025

Large Reasoning Models (LRMs) have recently demonstrated impressive performances across diverse domains. However, how the safety of Large Language Models (LLMs) benefits from enhanced reasoning capabilities against jailbreak queries remains unexplored. To bridge this gap, in this paper, we propose R

2025

Retrieval Models Aren’t Tool-Savvy: Benchmarking Tool Retrieval for Large Language Models

ACL 2025finding

Tool learning aims to augment large language models (LLMs) with diverse tools, enabling them to act as agents for solving practical tasks. Due to the limited context length of tool-using LLMs, adopting information retrieval (IR) models to select useful tools from large toolsets is a critical initial…

Cited by 0SourcePDFScholar
2025

Sliding Windows Are Not the End: Exploring Full Ranking with Long-Context Large Language Models

ACL 2025long

Large Language Models (LLMs) have shown exciting performance in listwise passage ranking. Due to the limited input length, existing methods often adopt the sliding window strategy. Such a strategy, though effective, is inefficient as it involves repetitive and serialized processing, which usually re…

2025

TP-RAG: Benchmarking Retrieval-Augmented Large Language Model Agents for Spatiotemporal-Aware Travel Planning

EMNLP 2025

Large language models (LLMs) have shown promise in automating travel planning, yet they often fall short in addressing nuanced spatiotemporal rationality. While existing benchmarks focus on basic plan validity, they neglect critical aspects such as route efficiency, POI appeal, and real-time adaptab

2025

Task Knowledge Injection via Interpolations and Reinstatement for Large Language Model Generalization

ACL 2025finding

Large language models have shown tremendous potential across various NLP tasks, and instruction tuning has been widely adopted to elicit their superior performance. However, instruction tuning may overly tailor the models to task-specific formats, potentially compromising their generalization on uns…

2025

Uplift-RAG: Uplift-Driven Knowledge Preference Alignment for Retrieval-Augmented Generation

EMNLP 2025

Retrieval-augmented generation (RAG) has proven effective in enhancing the knowledge coverage of large language models (LLMs) and mitigating hallucinations by incorporating external retrieved documents. However, documents deemed relevant by the retriever are not necessarily helpful for answer genera

2024

A Robust Semantics-based Watermark for Large Language Model against Paraphrasing

NAACL 2024findings

Large language models (LLMs) have show their remarkable ability in various natural language tasks. However, there are concerns that LLMs are possible to be used improperly or even illegally. To prevent the malicious usage of LLMs, detecting LLM-generated text becomes crucial in the deployment of LLM…

2024

AdaSwitch: Adaptive Switching between Small and Large Agents for Effective Cloud-Local Collaborative Learning

EMNLP 2024main

Recent advancements in large language models (LLMs) have been remarkable. Users face a choice between using cloud-based LLMs for generation quality and deploying local-based LLMs for lower computational cost. The former option is typically costly and inefficient, while the latter usually fails to de…

Cited by 2SourcePDFScholar
2024

Cross-model Control: Improving Multiple Large Language Models in One-time Training

NeurIPS 2024poster

The number of large language models (LLMs) with varying parameter scales and vocabularies is increasing. While they deliver powerful performance, they also face a set of common optimization needs to meet specific requirements or standards, such as instruction following or avoiding the output of sens…

2024

Exploring Memorization in Fine-tuned Language Models

ACL 2024long

Large language models (LLMs) have shown great capabilities in various tasks but also exhibited memorization of training data, raising tremendous privacy and copyright concerns. While prior works have studied memorization during pre-training, the exploration of memorization during fine-tuning is rath…

Cited by 26SourcePDFScholar
2024

G3: An Effective and Adaptive Framework for Worldwide Geolocalization Using Large Multi-Modality Models

NeurIPS 2024poster

Worldwide geolocalization aims to locate the precise location at the coordinate level of photos taken anywhere on the Earth. It is very challenging due to 1) the difficulty of capturing subtle location-aware visual semantics, and 2) the heterogeneous geographical distribution of image data. As a res…

2024

GS2P: A Generative Pre-trained Learning to Rank Model with Over-parameterization for Web-Scale Search (Extended Abstract)

IJCAI 2024poster

While Learning to Rank (LTR) is widely employed in web searches to prioritize pertinent webpages from the retrieved contents based on input queries, traditional LTR models stumble over two principal stumbling blocks leading to subpar performance: 1) the lack of well-annotated query-webpage pairs wit…

Cited by 7SourcePDFScholar
2024

Improving the Robustness of Large Language Models via Consistency Alignment

COLING 2024main

Large language models (LLMs) have shown tremendous success in following user instructions and generating helpful responses. Nevertheless, their robustness is still far from optimal, as they may generate significantly inconsistent responses due to minor changes in the verbalized instructions. Recent…

2024

KnowTuning: Knowledge-aware Fine-tuning for Large Language Models

EMNLP 2024main

Despite their success at many natural language processing (NLP) tasks, large language models still struggle to effectively leverage knowledge for knowledge-intensive tasks, manifesting limitations such as generating incomplete, non-factual, or illogical answers. These limitations stem from inadequat…

2024

Knowing What LLMs DO NOT Know: A Simple Yet Effective Self-Detection Method

NAACL 2024long

Large Language Models (LLMs) have shown great potential in Natural Language Processing (NLP) tasks.However, recent literature reveals that LLMs hallucinate intermittently, which impedes their reliability for further utilization. In this paper, we propose a novel self-detection method to detect which…

2024

MILL: Mutual Verification with Large Language Models for Zero-Shot Query Expansion

NAACL 2024long

Query expansion, pivotal in search engines, enhances the representation of user information needs with additional terms. While existing methods expand queries using retrieved or generated contextual documents, each approach has notable limitations. Retrieval-based methods often fail to accurately ca…

2024

MPGraf: a Modular and Pre-trained Graphformer for Learning to Rank at Web-scale (Extended Abstract)

IJCAI 2024poster

Both Transformer and Graph Neural Networks (GNNs) have been used in learning to rank (LTR), however, they adhere to two distinct yet complementary problem formulations, i.e., ranking score regression based on query-webpage pairs and link prediction within query-webpage bipartite graphs, respectively…

Cited by 0SourcePDFScholar
2024

The Good and The Bad: Exploring Privacy Issues in Retrieval-Augmented Generation (RAG)

ACL 2024findings

Retrieval-augmented generation (RAG) is a powerful technique to facilitate language model generation with proprietary and private data, where data privacy is a pivotal concern. Whereas extensive research has demonstrated the privacy risks of large language models (LLMs), the RAG technique could pote…

2024

Towards Verifiable Text Generation with Evolving Memory and Self-Reflection

EMNLP 2024main

Despite the remarkable ability of large language models (LLMs) in language comprehension and generation, they often suffer from producing factually incorrect information, also known as hallucination. A promising solution to this issue is verifiable text generation, which prompts LLMs to generate con…

Cited by 16SourcePDFScholar
2023

Boosting Event Extraction with Denoised Structure-to-Text Augmentation

ACL 2023findings

Event extraction aims to recognize pre-defined event triggers and arguments from texts, which suffer from the lack of high-quality annotations. In most NLP applications, involving a large scale of synthetic training data is a practical and effective approach to alleviate the problem of data scarcity…

2023

DiQAD: A Benchmark Dataset for Open-domain Dialogue Quality Assessment

EMNLP 2023long findings

Dialogue assessment plays a critical role in the development of open-domain dialogue systems. Existing work are uncapable of providing an end-to-end and human-epistemic assessment dataset, while they only provide sub-metrics like coherence or the dialogues are conversed between annotators far from r…

Cited by 0SourcecodeScholar
2023

Is ChatGPT Good at Search? Investigating Large Language Models as Re-Ranking Agents

EMNLP 2023long main

Large Language Models (LLMs) have demonstrated remarkable zero-shot generalization across various language-related tasks, including search engines. However, existing work utilizes the generative ability of LLMs for Information Retrieval (IR) rather than direct passage ranking. The discrepancy betwe…

Cited by 0SourcecodeScholar
2023

Learning to Tokenize for Generative Retrieval

NeurIPS 2023poster

As a new paradigm in information retrieval, generative retrieval directly generates a ranked list of document identifiers (docids) for a given query using generative language models (LMs). How to assign each document a unique docid (denoted as document tokenization) is a critical problem, because it…

Cited by 112SourcePDFScholar
2022

A Large Scale Search Dataset for Unbiased Learning to Rank

NeurIPS 2022accept

The unbiased learning to rank (ULTR) problem has been greatly advanced by recent deep learning techniques and well-designed debias algorithms. However, promising results on the existing benchmark datasets may not be extended to the practical scenario due to some limitations of existing datasets. Fir…