← Search

Chongyang Tao

59 accepted papers

2026

AIR: Post-training Data Selection for Reasoning via Attention Head Influence

ICML 2026poster

LLMs achieve remarkable multi-step reasoning capabilities, yet effectively transferring these skills via post-training distillation remains challenging. Existing data selection methods, ranging from manual curation to heuristics based on length, entropy, or overall loss, fail to capture the causal i…

Cited by 0SourceScholar
2026

Dynamic Stratified Contrastive Learning with Upstream Augmentation for MILP Branching

ICML 2026spotlight

Mixed Integer Linear Programming (MILP) is a fundamental NP-hard problem that has garnered significant attention from both academia and industry. The Branch-and-Bound (B&B) algorithm is the dominant approach for solving MILPs, where branching decisions play a critical role and have recently been enh…

Cited by 0SourceScholar
2026

D²Evo: Dual Difficulty-Aware Self-Evolution for Data-Efficient Reinforcement Learning

ICML 2026poster

Reinforcement learning (RL) has demonstrated potential for enhancing reasoning in large language models (LLMs). However, effective RL training, which requires medium-difficulty training samples, faces two fundamental challenges: Effective Data Scarcity and Dynamic Difficulty Shifts, where medium-dif…

Cited by 0SourceScholar
2026

NL2Repo-Bench: Towards Long-Horizon Repository Generation Evaluation of Coding Agents

ICML 2026poster

Recent advances in coding agents suggest rapid progress toward autonomous software development, yet existing benchmarks primarily evaluate short-horizon behaviors such as localized code generation, scaffolded completion, or repository repair, leaving it unclear whether agents can sustain coherent re…

Cited by 0SourceScholar
2026

Unsat Core Prediction through Polarity-Aware Representation Learning over Clause-Literal Hypergraphs

ICML 2026poster

Graph neural networks have been widely used in Boolean satisfiability (SAT) tasks to learn structural information from SAT formulas. The goal of these studies is to solve SAT instances or to enhance SAT solvers, including tasks such as unsat-core prediction. However, most existing approaches model a…

Cited by 0SourceScholar
2025

A Comprehensive Evaluation on Event Reasoning of Large Language Models

AAAI 2025technical

Event reasoning is a fundamental ability that underlies many applications. It requires event schema knowledge to perform global reasoning and needs to deal with the diversity of the inter-event relations and the reasoning paradigms. The extent to which LLMs excel in event reasoning across various re…

2025

Benchmarking Long-Context Language Models on Long Code Understanding

ACL 2025long

Current advanced long-context language models offer great potential for real-world software engineering applications. However, progress in this critical domain remains hampered by a fundamental limitation: the absence of a rigorous evaluation framework for long code understanding. To gap this obstac…

Cited by 0SourcePDFScholar
2025

WizardMath: Empowering Mathematical Reasoning for Large Language Models via Reinforced Evol-Instruct

ICLR 2025oral

Large language models (LLMs), such as GPT-4, have shown remarkable performance in natural language processing (NLP) tasks, including challenging mathematical reasoning. However, most existing open-source models are only pre-trained on large-scale internet data and without math-related optimization.…

Cited by 414SourcePDFScholar
2024

ADAM: Dense Retrieval Distillation with Adaptive Dark Examples

ACL 2024findings

To improve the performance of the dual-encoder retriever, one effective approach is knowledge distillation from the cross-encoder ranker. Existing works prepare training instances by pairing each query with one positive and a batch of negatives. However, most hard negatives mined by advanced dense r…

Cited by 5SourcePDFScholar
2024

Fine-Grained Distillation for Long Document Retrieval

AAAI 2024technical

Long document retrieval aims to fetch query-relevant documents from a large-scale collection, where knowledge distillation has become de facto to improve a retriever by mimicking a heterogeneous yet powerful cross-encoder. However, in contrast to passages or sentences, retrieval on long documents su…

Cited by 52SourcePDFScholar
2024

Leveraging Large Language Models for NLG Evaluation: Advances and Challenges

EMNLP 2024main

In the rapidly evolving domain of Natural Language Generation (NLG) evaluation, introducing Large Language Models (LLMs) has opened new avenues for assessing generated content quality, e.g., coherence, creativity, and context relevance. This paper aims to provide a thorough overview of leveraging LL…

2024

MEEL: Multi-Modal Event Evolution Learning

ACL 2024findings

Multi-modal Event Reasoning (MMER) endeavors to endow machines with the ability to comprehend intricate event relations across diverse data modalities. MMER is fundamental and underlies a wide broad of applications. Despite extensive instruction fine-tuning, current multi-modal large language models…

2024

Meta-Task Prompting Elicits Embeddings from Large Language Models

ACL 2024long

We introduce a new unsupervised text embedding method, Meta-Task Prompting with Explicit One-Word Limitation (MetaEOL), for generating high-quality sentence embeddings from Large Language Models (LLMs) without the need for model fine-tuning. Leveraging meta-task prompting, MetaEOL guides LLMs to pro…

2024

Pre-training Cross-Modal Retrieval by Expansive Lexicon-Patch Alignment

COLING 2024main

Recent large-scale vision-language pre-training depends on image-text global alignment by contrastive learning and is further boosted by fine-grained alignment in a weakly contrastive manner for cross-modal retrieval. Nonetheless, besides semantic matching learned by contrastive learning, cross-moda…

2024

Re-Reading Improves Reasoning in Large Language Models

EMNLP 2024main

To enhance the reasoning capabilities of off-the-shelf Large Language Models (LLMs), we introduce a simple, yet general and effective prompting method, RE2, i.e., Re-Reading the question as input. Unlike most thought-eliciting prompting methods, such as Chain-of-Thought (CoT), which aim to elicit th…

2024

Retrieval-Augmented Retrieval: Large Language Models are Strong Zero-Shot Retriever

ACL 2024findings

We propose a simple method that applies a large language model (LLM) to large-scale retrieval in zero-shot scenarios. Our method, the Large language model as Retriever (LameR), is built upon no other neural models but an LLM in a retrieval-augmented retrieval fashion, while breaking brute-force comb…

2024

Synergistic Interplay between Search and Large Language Models for Information Retrieval

ACL 2024long

Information retrieval (IR) plays a crucial role in locating relevant resources from vast amounts of data, and its applications have evolved from traditional knowledge bases to modern retrieval models (RMs). The emergence of large language models (LLMs) has further revolutionized the IR field by enab…

2024

WizardCoder: Empowering Code Large Language Models with Evol-Instruct

ICLR 2024poster

Code Large Language Models (Code LLMs), such as StarCoder, have demonstrated remarkable performance in various code-related tasks. However, different from their counterparts in the general language modeling field, the technique of instruction fine-tuning remains relatively under-researched in this d…

2024

WizardLM: Empowering Large Pre-Trained Language Models to Follow Complex Instructions

ICLR 2024poster

Training large language models (LLMs) with open-domain instruction following data brings colossal success. However, manually creating such instruction data is very time-consuming and labor-intensive. Moreover, humans may struggle to produce high-complexity instructions. In this paper, we show an ave…

Cited by 175SourcePDFScholar
2023

Attend, Select and Eliminate: Accelerating Multi-turn Response Selection with Dual-attention-based Content Elimination

ACL 2023findings

Although the incorporation of pre-trained language models (PLMs) significantly pushes the research frontier of multi-turn response selection, it brings a new issue of heavy computation costs. To alleviate this problem and make the PLM-based response selection model both effective and efficient, we p…

Cited by 1SourcePDFScholar
2023

CORE: Cooperative Training of Retriever-Reranker for Effective Dialogue Response Selection

ACL 2023long

Establishing retrieval-based dialogue systems that can select appropriate responses from the pre-built index has gained increasing attention. Recent common practice is to construct a two-stage pipeline with a fast retriever (e.g., bi-encoder) for first-stage recall followed by a smart response reran…

Cited by 5SourcePDFScholar
2023

FAA: Fine-grained Attention Alignment for Cascade Document Ranking

ACL 2023long

Document ranking aims at sorting a collection of documents with their relevance to a query. Contemporary methods explore more efficient transformers or divide long documents into passages to handle the long input. However, intensive query-irrelevant content may lead to harmful distraction and high q…

Cited by 4SourcePDFScholar
2023

HypeR: Multitask Hyper-Prompted Training Enables Large-Scale Retrieval Generalization

ICLR 2023poster

Recently, large-scale text retrieval has made impressive progress, facilitating both information retrieval and downstream knowledge-intensive tasks (e.g., open-domain QA and dialogue). With a moderate amount of data, a neural text retriever can outperform traditional methods such as BM25 by a large…

Cited by 10SourcePDFScholar
2023

Improving the Robustness of Summarization Systems with Dual Augmentation

ACL 2023long

A robust summarization system should be able to capture the gist of the document, regardless of the specific word choices or noise in the input. In this work, we first explore the summarization models’ robustness against perturbations including word-level synonym substitution and noise. To create se…

2023

KnowDA: All-in-One Knowledge Mixture Model for Data Augmentation in Low-Resource NLP

ICLR 2023poster

This paper focuses on data augmentation for low-resource NLP tasks where the training set is limited. The existing solutions either leverage task-independent heuristic rules (e.g., Synonym Replacement) or fine-tune general-purpose pre-trained language models (e.g., GPT2) using the limited training i…

Cited by 3SourcePDFScholar
2023

Length-Adaptive Distillation: Customizing Small Language Model for Dynamic Token Pruning

EMNLP 2023long findings

Pre-trained language models greatly improve the performance of various tasks but at a cost of high computation overhead. To facilitate practical applications, there are mainly two lines of research to accelerate model inference: model compression and dynamic computation (e.g., dynamic token pruning)…

Cited by 0SourceScholar
2023

LexLIP: Lexicon-Bottlenecked Language-Image Pre-Training for Large-Scale Image-Text Sparse Retrieval

ICCV 2023poster

Image-text retrieval (ITR) aims to retrieve images or texts that match a query originating from the other modality. The conventional dense retrieval paradigm relies on encoding images and texts into dense representations with dual-stream encoders. However, this approach is limited by slow retrieval…

Cited by 20PDFcodeScholar
2023

LexMAE: Lexicon-Bottlenecked Pretraining for Large-Scale Retrieval

ICLR 2023poster

In large-scale retrieval, the lexicon-weighting paradigm, learning weighted sparse representations in vocabulary space, has shown promising results with high quality and low latency. Despite it deeply exploiting the lexicon-representing capability of pre-trained language models, a crucial gap remain…

2023

MADNet: Maximizing Addressee Deduction Expectation for Multi-Party Conversation Generation

EMNLP 2023long main

Modeling multi-party conversations (MPCs) with graph neural networks has been proven effective at capturing complicated and graphical information flows. However, existing methods rely heavily on the necessary addressee labels and can only be applied to an ideal setting where each utterance must be ta…

Cited by 0SourcecodeScholar
2023

MMDialog: A Large-scale Multi-turn Dialogue Dataset Towards Multi-modal Open-domain Conversation

ACL 2023long

Responding with multi-modal content has been recognized as an essential capability for an intelligent conversational agent. In this paper, we introduce the MMDialog dataset to facilitate multi-modal conversation better. MMDialog is composed of a curated set of 1.08 million real-world dialogues with…

2023

SEAG: Structure-Aware Event Causality Generation

ACL 2023findings

Extracting event causality underlies a broad spectrum of natural language processing applications. Cutting-edge methods break this task into Event Detection and Event Causality Identification. Although the pipelined solutions succeed in achieving acceptable results, the inherent nature of separating…

Cited by 7SourcePDFScholar
2023

Towards Robust Ranker for Text Retrieval

ACL 2023findings

A neural ranker plays an indispensable role in the de facto ‘retrieval & rerank’ pipeline, but its training still lags behind due to the weak negative mining during contrastive learning. Compared to retrievers boosted by self-adversarial (i.e., in-distribution) negative mining, the ranker’s heavy st…

Cited by 53SourcePDFScholar
2023

UMSE: Unified Multi-scenario Summarization Evaluation

ACL 2023findings

Summarization quality evaluation is a non-trivial task in text summarization. Contemporary methods can be mainly categorized into two scenarios: (1) reference-based: evaluating with human-labeled reference summary; (2) reference-free: evaluating the summary consistency of the document. Recent studie…

2023

UniEvent: Unified Generative Model with Multi-Dimensional Prefix for Zero-Shot Event-Relational Reasoning

ACL 2023long

Reasoning about events and their relations attracts surging research efforts since it is regarded as an indispensable ability to fulfill various event-centric or common-sense reasoning tasks. However, these tasks often suffer from limited data availability due to the labor-intensive nature of their…

2022

Collaborative Reasoning on Multi-Modal Semantic Graphs for Video-Grounded Dialogue Generation

EMNLP 2022finding

We study video-grounded dialogue generation, where a response is generated based on the dialogue context and the associated video. The primary challenges of this task lie in (1) the difficulty of integrating video data into pre-trained language models (PLMs) which presents obstacles to exploiting th…

Cited by 5SourcePDFScholar
2022

HeterMPC: A Heterogeneous Graph Neural Network for Response Generation in Multi-Party Conversations

ACL 2022long

Recently, various response generation models for two-party conversations have achieved impressive improvements, but less effort has been paid to multi-party conversations (MPCs) which are more practical and complicated. Compared with a two-party conversation where a dialogue context is a sequence of…

2022

How to Represent Context Better? An Empirical Study on Context Modeling for Multi-turn Response Selection

EMNLP 2022finding

Building retrieval-based dialogue models that can predict appropriate responses based on the understanding of multi-turn context messages is a challenging problem. Early models usually concatenate all utterances or independently encode each dialogue turn, which may lead to an inadequate understandin…

Cited by 4SourcePDFScholar
2022

Learning to Express in Knowledge-Grounded Conversation

NAACL 2022long

Grounding dialogue generation by extra knowledge has shown great potentials towards building a system capable of replying with knowledgeable and engaging responses. Existing studies focus on how to synthesize a response with proper knowledge, yet neglect that the same knowledge could be expressed di…

2022

Multi-Granularity Structural Knowledge Distillation for Language Model Compression

ACL 2022long

Transferring the knowledge to a small model through distillation has raised great interest in recent years. Prevailing methods transfer the knowledge derived from mono-granularity language units (e.g., token-level or sample-level), which is not enough to represent the rich semantics of a text and ma…

2022

PCL: Peer-Contrastive Learning with Diverse Augmentations for Unsupervised Sentence Embeddings

EMNLP 2022main

Learning sentence embeddings in an unsupervised manner is fundamental in natural language processing. Recent common practice is to couple pre-trained language models with unsupervised contrastive learning, whose success relies on augmenting a sentence with a semantically-close positive instance to c…

2022

PromDA: Prompt-based Data Augmentation for Low-Resource NLU Tasks

ACL 2022long

This paper focuses on the Data Augmentation for low-resource Natural Language Understanding (NLU) tasks. We propose Prompt-based Data Augmentation model (PromDA) which only trains small-scale Soft Prompt (i.e., a set of trainable vectors) in the frozen Pre-trained Language Models (PLMs). This avoids…

2022

ProphetChat: Enhancing Dialogue Generation with Simulation of Future Conversation

ACL 2022long

Typical generative dialogue models utilize the dialogue history to generate the response. However, since one dialogue utterance can often be appropriately answered by multiple distinct responses, generating a desired response solely based on the historical information is not easy. Intuitively, if th…

2022

Reciprocal Learning of Knowledge Retriever and Response Ranker for Knowledge-Grounded Conversations

COLING 2022main

Grounding dialogue agents with knowledge documents has sparked increased attention in both academia and industry. Recently, a growing body of work is trying to build retrieval-based knowledge-grounded dialogue systems. While promising, these approaches require collecting pairs of dialogue context an…

Cited by 4SourcePDFScholar
2022

Rethinking Task-Specific Knowledge Distillation: Contextualized Corpus as Better Textbook

EMNLP 2022main

Knowledge distillation has been proven effective when customizing small language models for specific tasks. Here, a corpus as ‘textbook’ plays an indispensable role, only through which the teacher can teach the student. Prevailing methods adopt a two-stage distillation paradigm: general distillation…

Cited by 9SourcePDFScholar
2022

TegTok: Augmenting Text Generation via Task-specific and Open-world Knowledge

ACL 2022findings

Generating natural and informative texts has been a long-standing problem in NLP. Much effort has been dedicated into incorporating pre-trained language models (PLMs) with various open-world knowledge, such as knowledge graphs or wiki pages. However, their ability to access and manipulate the task-s…

2022

There Are a Thousand Hamlets in a Thousand People’s Eyes: Enhancing Knowledge-grounded Dialogue with Personal Memory

ACL 2022long

Knowledge-grounded conversation (KGC) shows great potential in building an engaging and knowledgeable chatbot, and knowledge selection is a key ingredient in it. However, previous methods for knowledge selection only concentrate on the relevance between knowledge and dialogue context, ignoring the f…

2022

There Is No Standard Answer: Knowledge-Grounded Dialogue Generation with Adversarial Activated Multi-Reference Learning

EMNLP 2022main

Knowledge-grounded dialogue (KGC) shows excellent potential to deliver an engaging and informative response. However, existing approaches emphasize selecting one golden knowledge given a particular dialogue context, overlooking the one-to-many phenomenon in dialogue. As a result, existing paradigm l…

2021

A Pre-training Strategy for Zero-Resource Response Selection in Knowledge-Grounded Conversations

ACL 2021long

Recently, many studies are emerging towards building a retrieval-based dialogue system that is able to effectively leverage background knowledge (e.g., documents) when conversing with humans. However, it is non-trivial to collect large-scale dialogues that are naturally grounded on the background do…

Cited by 9SourcePDFScholar
2021

A Survey on Response Selection for Retrieval-based Dialogues

IJCAI 2021poster

Building an intelligent dialogue system capable of naturally and coherently conversing with humans has been a long-standing goal of artificial intelligence. In the past decade, with the development of machine/deep learning technology and the explosive growth of available conversation data in social…

Cited by 36SourcePDFScholar
2021

Learning an Effective Context-Response Matching Model with Self-Supervised Tasks for Retrieval-based Dialogues

AAAI 2021technical

Building an intelligent dialogue system with the ability to select a proper response according to a multi-turn context is a great challenging task. Existing studies focus on building a context-response matching model with various neural architectures or pretrained language models (PLMs) and typicall…

2021

Learning to Organize a Bag of Words into Sentences with Neural Networks: An Empirical Study

NAACL 2021long

Sequential information, a.k.a., orders, is assumed to be essential for processing a sequence with recurrent neural network or convolutional neural network based encoders. However, is it possible to encode natural languages without orders? Given a bag of words from a disordered sentence, humans may s…

Cited by 9SourcePDFScholar
2021

MPC-BERT: A Pre-Trained Language Model for Multi-Party Conversation Understanding

ACL 2021long

Recently, various neural models for multi-party conversation (MPC) have achieved impressive improvements on a variety of tasks such as addressee recognition, speaker identification and response prediction. However, these existing methods on MPC usually represent interlocutors and utterances individu…

2021

Maria: A Visual Experience Powered Conversational Agent

ACL 2021long

Arguably, the visual perception of conversational agents to the physical world is a key way for them to exhibit the human-like intelligence. Image-grounded conversation is thus proposed to address this challenge. Existing works focus on exploring the multimodal dialog models that ground the conversa…

2021

Neural Rule-Execution Tracking Machine For Transformer-Based Text Generation

NeurIPS 2021poster

Sequence-to-Sequence (Seq2Seq) neural text generation models, especially the pre-trained ones (e.g., BART and T5), have exhibited compelling performance on various natural language generation tasks. However, the black-box nature of these models limits their application in tasks where specific rules…

Cited by 12SourcePDFScholar
2020

Low-Resource Knowledge-Grounded Dialogue Generation

ICLR 2020poster

Responding with knowledge has been recognized as an important capability for an intelligent conversational agent. Yet knowledge-grounded dialogues, as training data for learning such a response generation model, are difficult to obtain. Motivated by the challenge in practice, we consider knowledge-g…

Cited by 117SourceScholar
2020

Zero-Resource Knowledge-Grounded Dialogue Generation

NeurIPS 2020poster

While neural conversation models have shown great potentials towards generating informative and engaging responses via introducing external knowledge, learning such a model often requires knowledge-grounded dialogues that are difficult to obtain. To overcome the data challenge and reduce the cost of…

2019

Overcoming Catastrophic Forgetting for Continual Learning via Model Adaptation

ICLR 2019poster

Learning multiple tasks sequentially is important for the development of AI and lifelong learning systems. However, standard neural network architectures suffer from catastrophic forgetting which makes it difficult for them to learn a sequence of tasks. Several continual learning methods have been p…

Cited by 211SourcePDFScholar