← Search

Hengyi Cai

16 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
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

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

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

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

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

Answering Ambiguous Questions via Iterative Prompting

ACL 2023long

In open-domain question answering, due to the ambiguity of questions, multiple plausible answers may exist. To provide feasible answers to an ambiguous question,one approach is to directly predict all valid answers, but this can struggle with balancing relevance and diversity. An alternative is to g…

2023

Contrastive Learning with Dialogue Attributes for Neural Dialogue Generation

ICASSP 2023accepted

Designing an effective learning method remains a challenge in neural dialogue generation systems as it requires the training objective to well approximate the intrinsic human-preferred dialogue properties. Conventional training approaches such as maximum likelihood estimation focus on modeling gener…

Cited by 0SourceScholar