← Search

Caixia Yuan

20 accepted papers

2026

An Information-Theoretic Parameter-Free Bayesian Framework for Probing Labeled Dependency Trees from Attention Score

ICLR 2026poster

Figuring out how neural language models comprehend syntax acts as a key to revealing how they understand languages. We systematically analyzed methods of extracting syntax from models, namely _probing_, and found limitations yet widely exist in previous probing practice. We proposed a method capab…

Cited by 0SourcecodeScholar
2026

EntroKV: Entropy-Guided Dynamic Budget Allocation for KV-Cache Compression

ICML 2026spotlight

The prohibitive memory footprint of the Key-Value (KV) cache imposes a critical bottleneck for efficient long-context LLM serving. Current compression techniques typically rely on static or uniform budget allocation, overlooking the significant heterogeneity in information density across attention h…

Cited by 0SourceScholar
2026

Reading Images Like Texts: Sequential Image Understanding in Vision-Language Models

ICLR 2026poster

Vision-Language Models (VLMs) have demonstrated remarkable performance across a variety of real-world tasks. However, existing VLMs typically process visual information by serializing images, a method that diverges significantly from the parallel nature of human vision. Moreover, their opaque intern…

Cited by 0SourcecodeScholar
2026

Semantic Impact–Driven Visual Scheduling in Vision-Language Models

ICML 2026poster

Vision-Language Models (VLMs) suffer from high inference latency due to long visual sequences. To enable efficient, on-demand utilization of visual information, we argue that visual necessity should be assessed by its semantic impact on the output distribution, rather than inferred from intermediate…

Cited by 0SourceScholar
2025

A Systematic Exploration of Knowledge Graph Alignment with Large Language Models in Retrieval Augmented Generation

AAAI 2025technical

Retrieval Augmented Generation (RAG) with Knowledge Graphs (KGs) is an effective way to enhance Large Language Models (LLMs). Due to the natural discrepancy between structured KGs and sequential LLMs, KGs must be linearized to text before being inputted into LLMs, leading to the problem of KG Alignm…

2025

CoTD-PO: Chain-of-Thought Distillation with Preference Optimization

EMNLP 2025

Chain-of-Thought (CoT) distillation has emerged as a promising paradigm to enhance the reasoning ability of small language models by imitating the reasoning and outputs of larger teacher models. However, existing approaches suffer from a critical limitation: a distribution mismatch between teacher-g

2025

Collab-Overcooked: Benchmarking and Evaluating Large Language Models as Collaborative Agents

EMNLP 2025

Large Language Models (LLMs) based agent systems have made great strides in real-world applications beyond traditional NLP tasks. This paper proposes a new LLM-based Multi-Agent System (LLM-MAS) benchmark, Collab-Overcooked, built on the popular Overcooked-AI game with more applicable and challengin

2025

Controlled Low-Rank Adaptation with Subspace Regularization for Continued Training on Large Language Models

ACL 2025long

Large language models (LLMs) exhibit remarkable capabilities in natural language processing but face catastrophic forgetting when learning new tasks, where adaptation to a new domain leads to a substantial decline in performance on previous tasks. In this paper, we propose Controlled LoRA (CLoRA), a…

2025

Data with High and Consistent Preference Difference Are Better for Reward Model

AAAI 2025technical

Reinforcement Learning from Human Feedback (RLHF) is a commonly used alignment method for Large Language Models (LLMs). This method relies on a reward model trained on a preference dataset to provide scalar rewards. However, the human-annotated preference data is often sparse, noisy, and costly to o…

2024

KG-Adapter: Enabling Knowledge Graph Integration in Large Language Models through Parameter-Efficient Fine-Tuning

ACL 2024findings

Although large language models (LLMs) show remarkable capabilities and generalizability across various tasks, they are criticized for lack of expertise. One promising solution is to combine knowledge graphs (KGs) with LLMs, and recent studies focus on integrating KGs into LLMs through prompt-based m…

2023

An Asynchronous Updating Reinforcement Learning Framework for Task-Oriented Dialog System

ICASSP 2023accepted

Reinforcement learning has been applied to train the dialog systems in many works. Previous approaches divide the dialog system into multiple modules including DST (dialog state tracking) and DP (dialog policy), and train these modules simultaneously. However, different modules influence each other…

Cited by 0SourceScholar
2023

Explicit Alignment and Many-to-many Entailment Based Reasoning for Conversational Machine Reading

EMNLP 2023long findings

Conversational Machine Reading (CMR) requires answering a user's initial question through multi-turn dialogue interactions based on a given document. Although there exist many effective methods, they largely neglected the alignment between the $\textit{document}$ and the $\textit{user-provided infor…

Cited by 0SourceScholar
2023

Multimodal Recommendation Dialog with Subjective Preference: A New Challenge and Benchmark

ACL 2023findings

Existing multimodal task-oriented dialog data fails to demonstrate the diverse expressions of user subjective preferences and recommendation acts in the real-life shopping scenario. This paper introduces a new dataset SURE (Multimodal Recommendation Dialog with Subjective Preference), which contains…

2023

SPRING: Situated Conversation Agent Pretrained with Multimodal Questions from Incremental Layout Graph

AAAI 2023technical

Existing multimodal conversation agents have shown impressive abilities to locate absolute positions or retrieve attributes in simple scenarios, but they fail to perform well when complex relative positions and information alignments are involved, which poses a bottleneck in response quality. In thi…

2022

A Slot Is Not Built in One Utterance: Spoken Language Dialogs with Sub-Slots

ACL 2022findings

A slot value might be provided segment by segment over multiple-turn interactions in a dialog, especially for some important information such as phone numbers and names. It is a common phenomenon in daily life, but little attention has been paid to it in previous work. To fill the gap, this paper de…

2022

Learn to Adapt for Generalized Zero-Shot Text Classification

ACL 2022long

Generalized zero-shot text classification aims to classify textual instances from both previously seen classes and incrementally emerging unseen classes. Most existing methods generalize poorly since the learned parameters are only optimal for seen classes rather than for both classes, and the param…

2021

Converse, Focus and Guess – Towards Multi-Document Driven Dialogue

AAAI 2021technical

We propose a novel task, Multi-Document Driven Dialogue (MD3), in which an agent can guess the target document that the user is interested in by leading a dialogue. To benchmark progress, we introduce a new dataset of GuessMovie, which contains 16,881 documents, each describing a movie, and associat…

2021

Task-Oriented Clustering for Dialogues

EMNLP 2021finding

A reliable clustering algorithm for task-oriented dialogues can help developer analysis and define dialogue tasks efficiently. It is challenging to directly apply prior normal text clustering algorithms for task-oriented dialogues, due to the inherent differences between them, such as coreference, o…

2021

Topic-Aware Contrastive Learning for Abstractive Dialogue Summarization

EMNLP 2021finding

Unlike well-structured text, such as news reports and encyclopedia articles, dialogue content often comes from two or more interlocutors, exchanging information with each other. In such a scenario, the topic of a conversation can vary upon progression and the key information for a certain topic is o…