← Search

Dian Yu

39 accepted papers

2026

CDE: Curiosity-Driven Exploration for Efficient Reinforcement Learning in Large Language Models

ICLR 2026poster

Reinforcement Learning with Verifiable Rewards (RLVR) is a powerful paradigm for enhancing the reasoning ability of Large Language Models (LLMs). Yet current RLVR methods often explore poorly, leading to premature convergence and entropy collapse. Moreover, they tend to produce poorly calibrated pol…

Cited by 0SourcecodeScholar
2026

DeepMath-103K: A Large-Scale, Challenging, Decontaminated, and Verifiable Mathematical Dataset for Advancing Reasoning

ICLR 2026poster

Reinforcement learning (RL) with large language models shows promise in complex reasoning. However, its progress is hindered by the lack of large-scale training data that is sufficiently challenging, contamination-free and verifiable. To this end, we introduce DeepMath-103K, a large-scale mathematic…

Cited by 0SourcecodeScholar
2026

Self-Rewarding Vision-Language Model via Reasoning Decomposition and Multi-Reward Policy Optimization

ICLR 2026poster

Vision-Language Models (VLMs) often suffer from visual hallucinations – generating things that are not consistent with visual inputs – and language shortcuts, where they skip the visual part and just rely on text priors. These issues arise because most post-training methods for VLMs rely on simple v…

Cited by 0SourceScholar
2026

Stable and Efficient Single-Rollout RL for Multimodal Reasoning

CVPR 2026

Reinforcement Learning with Verifiable Rewards (RLVR) has become a key paradigm to improve the reasoning capabilities of Multimodal Large Language Models (MLLMs). However, prevalent group-based algorithms such as GRPO require multi-rollout sampling for each prompt. While more efficient single-rollou

Cited by 0SourceScholar
2025

DOTS: Learning to Reason Dynamically in LLMs via Optimal Reasoning Trajectories Search

ICLR 2025poster

Enhancing the capability of large language models (LLMs) in reasoning has gained significant attention in recent years. Previous studies have demonstrated the effectiveness of various prompting strategies in aiding LLMs in reasoning (called "reasoning actions"), such as step-by-step thinking, reflec…

2025

Do NOT Think That Much for 2+3=? On the Overthinking of Long Reasoning Models

ICML 2025poster

The remarkable performance of long reasoning models can be attributed to their ability to emulate human-like long-time thinking during inference. These models employ extended chain-of-thought (CoT) processes, exploring multiple strategies to enhance problem-solving capabilities. However, a critical…

2025

Don’t Get Lost in the Trees: Streamlining LLM Reasoning by Overcoming Tree Search Exploration Pitfalls

ACL 2025long

Recent advancements in tree search algorithms guided by verifiers have significantly enhanced the reasoning capabilities of large language models (LLMs), but at the cost of increased computational resources. In this work, we identify two key challenges contributing to this inefficiency: over-explora…

2025

Improving LLM General Preference Alignment via Optimistic Online Mirror Descent

NeurIPS 2025spotlight

Reinforcement learning from human feedback (RLHF) has demonstrated remarkable effectiveness in aligning large language models (LLMs) with human preferences. Many existing alignment approaches rely on the Bradley-Terry (BT) model assumption, which assumes the existence of a ground-truth reward for ea…

Cited by 0SourceScholar
2025

Iterative Nash Policy Optimization: Aligning LLMs with General Preferences via No-Regret Learning

ICLR 2025oral

Reinforcement Learning with Human Feedback (RLHF) has achieved great success in aligning large language models (LLMs) with human preferences. Prevalent RLHF approaches are reward-based, following the Bradley-Terry (BT) model assumption, which may not fully capture the complexity of human preferences…

Cited by 4SourcePDFScholar
2025

LiteSearch: Efficient Tree Search with Dynamic Exploration Budget for Math Reasoning

AAAI 2025technical

Recent research suggests that tree search algorithms (e.g. Monte Carlo Tree Search) can dramatically boost LLM performance on complex mathematical reasoning tasks. However, they often require more than 10 times the computational resources of greedy decoding due to wasteful search strategies, making…

Cited by 0SourcePDFScholar
2025

Thoughts Are All Over the Place: On the Underthinking of Long Reasoning Models

NeurIPS 2025spotlight

Long reasoning models (LRMs) such as OpenAI's o1 and DeepSeek's R1 have demonstrated remarkable abilities in complex reasoning tasks by scaling test-time compute and exhibiting human-like deep thinking. However, we identify a phenomenon we term underthinking, where LRMs frequently switch between dif…

Cited by 0SourcecodeScholar
2024

Alternate Diverse Teaching for Semi-supervised Medical Image Segmentation

ECCV 2024poster

"Semi-supervised medical image segmentation has shown promise in training models with limited labeled data. However, current dominant teacher-student based approaches can suffer from the confirmation bias. To address this challenge, we propose AD-MT, an alternate diverse teaching approach in a teach…

2024

Learn Beyond The Answer: Training Language Models with Reflection for Mathematical Reasoning

EMNLP 2024main

Supervised fine-tuning enhances the problem-solving abilities of language models across various mathematical reasoning tasks. To maximize such benefits, existing research focuses on *broadening* the training set with various data augmentation techniques, which is effective for standard single-round…

2024

MinT: Boosting Generalization in Mathematical Reasoning via Multi-view Fine-tuning

COLING 2024main

Reasoning in mathematical domains remains a significant challenge for relatively small language models (LMs). Many current methods focus on specializing LMs in mathematical reasoning and rely heavily on distilling knowledge from powerful yet inefficient large LMs (LLMs). In this work, we explore a n…

Cited by 20SourcePDFScholar
2024

Retrieval Augmented End-to-End Spoken Dialog Models

ICASSP 2024accepted

We recently developed a joint speech and language model (SLM [1]) which fuses a pretrained foundational speech model and a large language model (LLM), while preserving the in-context learning capability intrinsic to the pretrained LLM. In this paper, we apply SLM to dialog applications where the dia…

Cited by 0SourceScholar
2024

Skills-in-Context: Unlocking Compositionality in Large Language Models

EMNLP 2024finding

We investigate how to elicit compositional generalization capabilities in large language models (LLMs). Compositional generalization empowers LLMs to solve complex problems by combining foundational skills, a critical reasoning ability akin to human intelligence. However, even the most advanced LLMs…

Cited by 1SourcePDFScholar
2024

Sub-Sentence Encoder: Contrastive Learning of Propositional Semantic Representations

NAACL 2024long

We introduce sub-sentence encoder, a contrastively-learned contextual embedding model for fine-grained semantic representation of text. In contrast to the standard practice with sentence embeddings, where the meaning of an entire sequence of text is encoded into a fixed-length vector, the sub-senten…

2024

Toward Self-Improvement of LLMs via Imagination, Searching, and Criticizing

NeurIPS 2024poster

Despite the impressive capabilities of Large Language Models (LLMs) on various tasks, they still struggle with scenarios that involves complex reasoning and planning. Self-correction and self-learning emerge as viable solutions, employing strategies that allow LLMs to refine their outputs and learn…

2023

Document-Level Machine Translation with Large Language Models

EMNLP 2023long main

Large language models (LLMs) such as ChatGPT can produce coherent, cohesive, relevant, and fluent answers for various natural language processing (NLP) tasks. Taking document-level machine translation (MT) as a testbed, this paper provides an in-depth evaluation of LLMs' ability on discourse modelin…

Cited by 0SourcecodeScholar
2023

Knowledge-in-Context: Towards Knowledgeable Semi-Parametric Language Models

ICLR 2023top-25%

Fully-parametric language models generally require a huge number of model parameters to store the necessary knowledge for solving multiple natural language tasks in zero/few-shot settings. In addition, it is hard to adapt to the evolving world knowledge without the costly model re-training. In this…

Cited by 20SourcePDFScholar
2023

ReAct: Synergizing Reasoning and Acting in Language Models

ICLR 2023top-5%

While large language models (LLMs) have demonstrated impressive capabilities across tasks in language understanding and interactive decision making, their abilities for reasoning (e.g. chain-of-thought prompting) and acting (e.g. action plan generation) have primarily been studied as separate topics…

2023

Tree of Thoughts: Deliberate Problem Solving with Large Language Models

NeurIPS 2023oral

Language models are increasingly being deployed for general problem solving across a wide range of tasks, but are still confined to token-level, left-to-right decision-making processes during inference. This means they can fall short in tasks that require exploration, strategic lookahead, or where i…

2023

Zemi: Learning Zero-Shot Semi-Parametric Language Models from Multiple Tasks

ACL 2023findings

Although large language models have exhibited impressive zero-shot ability, the huge model size generally incurs high cost. Recently, semi-parametric language models, which augment a smaller language model with retrieved related background knowledge, alleviate the need for storing everything into th…

2022

C-MORE: Pretraining to Answer Open-Domain Questions by Consulting Millions of References

ACL 2022short

We consider the problem of pretraining a two-stage open-domain question answering (QA) system (retriever + reader) with strong transfer capabilities. The key challenge is how to construct a large amount of high-quality question-answer-context triplets without task-specific annotations. Specifically,…

2022

Improving Machine Reading Comprehension with Contextualized Commonsense Knowledge

ACL 2022long

To perform well on a machine reading comprehension (MRC) task, machine readers usually require commonsense knowledge that is not explicitly mentioned in the given documents. This paper aims to extract a new kind of structured knowledge from scripts and use it to improve MRC. We focus on scripts as t…

2022

Knowledge-grounded Dialog State Tracking

EMNLP 2022finding

Knowledge (including structured knowledge such as schema and ontology and unstructured knowledge such as web corpus) is a critical part of dialog understanding, especially for unseen tasks and domains. Traditionally, such domain-specific knowledge is encoded implicitly into model parameters for the…

Cited by 5SourcePDFScholar
2022

Learning-by-Narrating: Narrative Pre-Training for Zero-Shot Dialogue Comprehension

ACL 2022short

Comprehending a dialogue requires a model to capture diverse kinds of key information in the utterances, which are either scattered around or implicitly implied in different turns of conversations. Therefore, dialogue comprehension requires diverse capabilities such as paraphrasing, summarizing, and…

2022

NarraSum: A Large-Scale Dataset for Abstractive Narrative Summarization

EMNLP 2022finding

Narrative summarization aims to produce a distilled version of a narrative to describe its most salient events and characters. Writing a summary for a narrative is challenging as it requires an understanding of event causality and character behaviors. To encourage research in this direction, we prop…

2022

Unsupervised Slot Schema Induction for Task-oriented Dialog

NAACL 2022long

Carefully-designed schemas describing how to collect and annotate dialog corpora are a prerequisite towards building task-oriented dialog systems. In practical applications, manually designing schemas can be error-prone, laborious, iterative, and slow, especially when the schema is complicated. To a…

Cited by 15SourcePDFScholar
2021

Attribute Alignment: Controlling Text Generation from Pre-trained Language Models

EMNLP 2021finding

Large language models benefit from training with a large amount of unlabeled text, which gives them increasingly fluent and diverse generation capabilities. However, using these models for text generation that takes into account target attributes, such as sentiment polarity or specific topics, remai…

2021

Connect-the-Dots: Bridging Semantics between Words and Definitions via Aligning Word Sense Inventories

EMNLP 2021main

Word Sense Disambiguation (WSD) aims to automatically identify the exact meaning of one word according to its context. Existing supervised models struggle to make correct predictions on rare word senses due to limited training data and can only select the best definition sentence from one predefined…

Cited by 10SourcePDFScholar
2021

Few-shot Intent Classification and Slot Filling with Retrieved Examples

NAACL 2021long

Few-shot learning arises in important practical scenarios, such as when a natural language understanding system needs to learn new semantic labels for an emerging, resource-scarce domain. In this paper, we explore retrieval-based methods for intent classification and slot filling tasks in few-shot s…

Cited by 60SourcePDFScholar
2021

QA-Driven Zero-shot Slot Filling with Weak Supervision Pretraining

ACL 2021short

Slot-filling is an essential component for building task-oriented dialog systems. In this work, we focus on the zero-shot slot-filling problem, where the model needs to predict slots and their values, given utterances from new domains without training on the target domain. Prior methods directly enc…

2021

Self-Teaching Machines to Read and Comprehend with Large-Scale Multi-Subject Question-Answering Data

EMNLP 2021finding

Despite considerable progress, most machine reading comprehension (MRC) tasks still lack sufficient training data to fully exploit powerful deep neural network models with millions of parameters, and it is laborious, expensive, and time-consuming to create large-scale, high-quality MRC data through…

Cited by 6SourcePDFScholar
2020

CLUE: A Chinese Language Understanding Evaluation Benchmark

COLING 2020main

The advent of natural language understanding (NLU) benchmarks for English, such as GLUE and SuperGLUE allows new NLU models to be evaluated across a diverse set of tasks. These comprehensive benchmarks have facilitated a broad range of research and applications in natural language processing (NLP).…

2020

Exploration Based Language Learning for Text-Based Games

IJCAI 2020poster

This work presents an exploration and imitation-learning-based agent capable of state-of-the-art performance in playing text-based computer games. These games are of interest as they can be seen as a testbed for language understanding, problem-solving, and language generation by artificial agents.…

Cited by 0SourcePDFScholar