← Search

Shuohang Wang

46 accepted papers

2026

ThetaEvolve: Test-time Learning on Open Problems

ICML 2026poster

Recent advances in large language models (LLMs) have enabled breakthroughs in mathematical discovery, exemplified by AlphaEvolve, a closed-source system that evolves programs to improve bounds on open problems. However, it relies on ensembles of frontier LLMs to achieve new bounds and is a pure infe…

Cited by 0SourceScholar
2025

Decoder-Hybrid-Decoder Architecture for Efficient Reasoning with Long Generation

NeurIPS 2025poster

Recent advances in language modeling have demonstrated the effectiveness of State Space Models (SSMs) for efficient sequence modeling. While hybrid architectures such as Samba and the decoder-decoder architecture, YOCO, have shown promising performance gains over Transformers, prior works have not i…

Cited by 0SourcecodeScholar
2025

Is Your World Simulator a Good Story Presenter? A Consecutive Events-Based Benchmark for Future Long Video Generation

CVPR 2025poster

The current state-of-the-art video generative models can produce commercial-grade videos with highly realistic details. However, they still struggle to coherently present multiple sequential events in specific short stories, which is foreseeable an essential capability for future long video generati…

2025

Reinforcement Learning for Reasoning in Large Language Models with One Training Example

NeurIPS 2025poster

We show that reinforcement learning with verifiable reward using one training example (1-shot RLVR) is effective in incentivizing the math reasoning capabilities of large language models (LLMs). Applying RLVR to the base model Qwen2.5-Math-1.5B, we identify a single example that elevates model perfo…

Cited by 0SourcecodeScholar
2025

Routing Mamba: Scaling State Space Models with Mixture-of-Experts Projection

NeurIPS 2025poster

State Space Models (SSMs) offer remarkable performance gains in efficient sequence modeling, with constant per-step inference-time computation and memory complexity. Recent advances, such as Mamba, further enhance SSMs with input-dependent gating and hardware-aware implementations, positioning them…

Cited by 0SourcecodeScholar
2024

SciAgent: Tool-augmented Language Models for Scientific Reasoning

EMNLP 2024main

Scientific reasoning poses an excessive challenge for even the most advanced Large Language Models (LLMs). To make this task more practical and solvable for LLMs, we introduce a new task setting named tool-augmented scientific reasoning. This setting supplements LLMs with scalable toolsets, and shif…

Cited by 28SourcePDFScholar
2024

Small Models are Valuable Plug-ins for Large Language Models

ACL 2024findings

Large language models (LLMs) such as GPT-3 and GPT-4 are powerful but their weights are often publicly unavailable and their immense sizes make the models difficult to be tuned with common hardware. As a result, effectively tuning these models with large-scale supervised data can be challenging. As…

2024

Temperature-Centric Investigation of Speculative Decoding with Knowledge Distillation

EMNLP 2024finding

Speculative decoding stands as a pivotal technique to expedite inference in autoregressive (large) language models. This method employs a smaller *draft* model to speculate a block of tokens, which the *target* model then evaluates for acceptance. Despite a wealth of studies aimed at increasing the…

2023

APOLLO: A Simple Approach for Adaptive Pretraining of Language Models for Logical Reasoning

ACL 2023long

Logical reasoning over text is an important ability that requires understanding the semantics of the text and reasoning through them to arrive at correct inferences. Prior works on pretraining language models to improve the logical reasoning ability require complex processing of training data (e.g.,…

2023

Auto-Instruct: Automatic Instruction Generation and Ranking for Black-Box Language Models

EMNLP 2023long findings

Large language models (LLMs) can perform a wide range of tasks by following natural language instructions, without the necessity of task-specific fine-tuning. Unfortunately, the performance of LLMs is greatly influenced by the quality of these instructions, and manually writing effective instruction…

Cited by 0SourceScholar
2023

G-Eval: NLG Evaluation using Gpt-4 with Better Human Alignment

EMNLP 2023long main

The quality of texts generated by natural language generation (NLG) systems is hard to measure automatically. Conventional reference-based metrics, such as BLEU and ROUGE, have been shown to have relatively low correlation with human judgments, especially for tasks that require creativity and diver…

Cited by 0SourcecodeScholar
2023

Generate rather than Retrieve: Large Language Models are Strong Context Generators

ICLR 2023poster

Knowledge-intensive tasks, such as open-domain question answering (QA), require access to a large amount of world or domain knowledge. A common approach for knowledge-intensive tasks is to employ a retrieve-then-read pipeline that first retrieves a handful of relevant contextual documents from an ex…

2023

In-Context Demonstration Selection with Cross Entropy Difference

EMNLP 2023long findings

Large language models (LLMs) can use in-context demonstrations to improve performance on zero-shot tasks. However, selecting the best in-context examples is challenging because model performance can vary widely depending on the selected examples. We present a cross-entropy difference (CED) method fo…

Cited by 0SourcecodeScholar
2023

InheritSumm: A General, Versatile and Compact Summarizer by Distilling from GPT

EMNLP 2023long findings

While large models such as GPT-3 demonstrate exceptional performance in zeroshot and fewshot summarization tasks, their extensive serving and fine-tuning costs hinder their utilization in various applications. Conversely, previous studies have found that although automatic metrics tend to favor smal…

Cited by 0SourceScholar
2023

LMGQS: A Large-scale Dataset for Query-focused Summarization

EMNLP 2023long findings

Query-focused summarization (QFS) aims to extract or generate a summary of an input document that directly answers or is relevant to a given query. The lack of large-scale datasets in the form of documents, queries, and summaries has hindered model development in this area. In contrast, multiple lar…

Cited by 0SourceScholar
2023

Prompting GPT-3 To Be Reliable

ICLR 2023poster

Large language models (LLMs) show impressive abilities via few-shot prompting. Commercialized APIs such as OpenAI GPT-3 further increase their use in real-world language applications. However, the crucial problem of how to improve the reliability of GPT-3 is still under-explored. While reliability i…

2023

Sparse Modular Activation for Efficient Sequence Modeling

NeurIPS 2023poster

Recent hybrid models combining Linear State Space Models (SSMs) with self-attention mechanisms have demonstrated impressive results across a range of sequence modeling tasks. However, current approaches apply attention modules statically and uniformly to all elements in the input sequences, leading…

2023

The Shifted and The Overlooked: A Task-oriented Investigation of User-GPT Interactions

EMNLP 2023long main

Recent progress in Large Language Models (LLMs) has produced models that exhibit remarkable performance across a variety of NLP tasks. However, it remains unclear whether the existing focus of NLP research accurately captures the genuine requirements of human users. This paper provides a comprehensi…

Cited by 0SourcecodeScholar
2022

AdaPrompt: Adaptive Model Training for Prompt-based NLP

EMNLP 2022finding

Prompt-based learning, with its capability to tackle zero-shot and few-shot NLP tasks, has gained much attention in the community.The main idea is to bridge the gap between NLP downstream tasks and language modeling (LM), by mapping these tasks into natural language prompts, which are then filled by…

2022

An Empirical Study of Training End-to-End Vision-and-Language Transformers

CVPR 2022poster

Vision-and-language (VL) pre-training has proven to be highly effective on various VL downstream tasks. While recent work has shown that fully transformer-based VL models can be more efficient than previous region-feature-based methods, their performance on downstream tasks often degrades significan…

Cited by 430PDFcodeScholar
2022

CLIP-Event: Connecting Text and Images With Event Structures

CVPR 2022oral

Vision-language (V+L) pretraining models have achieved great success in supporting multimedia applications by understanding the alignments between images and text. While existing vision-language pretraining models primarily focus on understanding objects in images or entities in text, they often ign…

Cited by 151PDFcodeScholar
2022

Dict-BERT: Enhancing Language Model Pre-training with Dictionary

ACL 2022findings

Pre-trained language models (PLMs) aim to learn universal language representations by conducting self-supervised training tasks on large-scale corpora. Since PLMs capture word semantics in different contexts, the quality of word representations highly depends on word frequency, which usually follows…

Cited by 78SourcePDFScholar
2022

Empowering Language Models with Knowledge Graph Reasoning for Open-Domain Question Answering

EMNLP 2022main

Answering open-domain questions requires world knowledge about in-context entities. As pre-trained Language Models (LMs) lack the power to store all required knowledge, external knowledge sources, such as knowledge graphs, are often used to augment LMs. In this work, we propose knOwledge REasOning e…

Cited by 61SourcePDFScholar
2022

Human Parity on CommonsenseQA: Augmenting Self-Attention with External Attention

IJCAI 2022poster

Most of today's AI systems focus on using self-attention mechanisms and transformer architectures on large amounts of diverse data to achieve impressive performance gains. In this paper, we propose to augment the transformer architecture with an external attention mechanism to bring external knowled…

2022

KG-FiD: Infusing Knowledge Graph in Fusion-in-Decoder for Open-Domain Question Answering

ACL 2022long

Current Open-Domain Question Answering (ODQA) models typically include a retrieving module and a reading module, where the retriever selects potentially relevant passages from open-source documents for a given question, and the reader produces an answer based on the retrieved passages. The recently…

Cited by 128SourcePDFScholar
2022

Language Models with Image Descriptors are Strong Few-Shot Video-Language Learners

NeurIPS 2022accept

The goal of this work is to build flexible video-language models that can generalize to various video-to-text tasks from few examples. Existing few-shot video-language learners focus exclusively on the encoder, resulting in the absence of a video-to-text decoder to handle generative tasks. Video cap…

2022

Leveraging Knowledge in Multilingual Commonsense Reasoning

ACL 2022findings

Commonsense reasoning (CSR) requires models to be equipped with general world knowledge. While CSR is a language-agnostic process, most comprehensive knowledge sources are restricted to a small number of languages, especially English. Thus, it remains unclear how to effectively conduct multilingual…

2022

ParaTag: A Dataset of Paraphrase Tagging for Fine-Grained Labels, NLG Evaluation, and Data Augmentation

EMNLP 2022main

Paraphrase identification has been formulated as a binary classification task to decide whether two sentences hold a paraphrase relationship. Existing paraphrase datasets only annotate a binary label for each sentence pair. However, after a systematical analysis of existing paraphrase datasets, we f…

2022

Playing Lottery Tickets with Vision and Language

AAAI 2022technical

Large-scale pre-training has recently revolutionized vision-and-language (VL) research. Models such as LXMERT and UNITER have significantly lifted the state of the art over a wide range of VL tasks. However, the large number of parameters in such models hinders their application in practice. In para…

Cited by 57SourcePDFScholar
2022

Retrieval Augmentation for Commonsense Reasoning: A Unified Approach

EMNLP 2022main

A common thread of retrieval-augmented methods in the existing literature focuses on retrieving encyclopedic knowledge, such as Wikipedia, which facilitates well-defined entity and relation spaces that can be modeled. However, applying such methods to commonsense reasoning tasks faces two unique cha…

2022

Task Compass: Scaling Multi-task Pre-training with Task Prefix

EMNLP 2022finding

Leveraging task-aware annotated data as supervised signals to assist with self-supervised learning on large-scale unlabeled data has become a new trend in pre-training language models. Existing studies show that multi-task learning with large-scale supervised tasks suffers from negative effects acro…

2022

Training Data is More Valuable than You Think: A Simple and Effective Method by Retrieving from Training Data

ACL 2022long

Retrieval-based methods have been shown to be effective in NLP tasks via introducing external knowledge. However, the indexing and retrieving of large-scale corpora bring considerable computational cost. Surprisingly, we found that REtrieving from the traINing datA (REINA) only can lead to significa…

2021

Adversarial GLUE: A Multi-Task Benchmark for Robustness Evaluation of Language Models

NeurIPS 2021poster

Large-scale pre-trained language models have achieved tremendous success across a wide range of natural language understanding (NLU) tasks, even surpassing human performance. However, recent studies reveal that the robustness of these models can be challenged by carefully crafted textual adversarial…

Cited by 245SourcecodeScholar
2021

EarlyBERT: Efficient BERT Training via Early-bird Lottery Tickets

ACL 2021long

Heavily overparameterized language models such as BERT, XLNet and T5 have achieved impressive success in many NLP tasks. However, their high model complexity requires enormous computation resources and extremely long training time for both pre-training and fine-tuning. Many works have studied model…

2021

FILTER: An Enhanced Fusion Method for Cross-lingual Language Understanding

AAAI 2021technical

Large-scale cross-lingual language models (LM), such as mBERT, Unicoder and XLM, have achieved great success in cross-lingual representation learning. However, when applied to zero-shot cross-lingual transfer tasks, most existing methods use only single-language input for LM finetuning, without leve…

2021

InfoBERT: Improving Robustness of Language Models from An Information Theoretic Perspective

ICLR 2021poster

Large-scale language models such as BERT have achieved state-of-the-art performance across a wide range of NLP tasks. Recent studies, however, show that such BERT-based models are vulnerable facing the threats of textual adversarial attacks. We aim to address this problem from an information-theoret…

2021

LightningDOT: Pre-training Visual-Semantic Embeddings for Real-Time Image-Text Retrieval

NAACL 2021long

Multimodal pre-training has propelled great advancement in vision-and-language research. These large-scale pre-trained models, although successful, fatefully suffer from slow inference speed due to enormous computational cost mainly from cross-modal attention in Transformer architecture. When applie…

2021

NOAHQA: Numerical Reasoning with Interpretable Graph Question Answering Dataset

EMNLP 2021finding

While diverse question answering (QA) datasets have been proposed and contributed significantly to the development of deep learning models for QA tasks, the existing datasets fall short in two aspects. First, we lack QA datasets covering complex questions that involve answers as well as the reasonin…

2021

On Orthogonality Constraints for Transformers

ACL 2021short

Orthogonality constraints encourage matrices to be orthogonal for numerical stability. These plug-and-play constraints, which can be conveniently incorporated into model training, have been studied for popular architectures in natural language processing, such as convolutional neural networks and re…

Cited by 24SourcePDFScholar
2021

The Elastic Lottery Ticket Hypothesis

NeurIPS 2021poster

Lottery Ticket Hypothesis (LTH) raises keen attention to identifying sparse trainable subnetworks, or winning tickets, which can be trained in isolation to achieve similar or even better performance compared to the full models. Despite many efforts being made, the most effective method to identify s…

2021

UC2: Universal Cross-Lingual Cross-Modal Vision-and-Language Pre-Training

CVPR 2021poster

Vision-and-language pre-training has achieved impressive success in learning multimodal representations between vision and language. To generalize this success to non-English languages, we introduce UC^2, the first machine translation-augmented framework for cross-lingual cross-modal representation…

Cited by 101PDFScholar
2018

Evidence Aggregation for Answer Re-Ranking in Open-Domain Question Answering

ICLR 2018poster

Very recently, it comes to be a popular approach for answering open-domain questions by first searching question-related passages, then applying reading comprehension models to extract answers. Existing works usually extract answers from single passages independently, thus not fully make use of the…