← Search

Wei Bi

37 accepted papers

2026

DreamOn: Diffusion Language Models For Code Infilling Beyond Fixed-size Canvas

ICLR 2026poster

Diffusion Language Models (DLMs) present a compelling alternative to autoregressive models, offering flexible, any-order infilling without specialized prompting design. However, their practical utility is blocked by a critical limitation: the requirement of a fixed-length masked sequence for generat…

Cited by 0SourcecodeScholar
2026

Weights to Code: Extracting Interpretable Algorithms from the Discrete Transformer

ICML 2026poster

Algorithm extraction aims to synthesize executable programs directly from models trained on algorithmic tasks, enabling *de novo* algorithm discovery without relying on human-written code. However, applying this paradigm to Transformer is hindered by representation entanglement (e.g., superposition)…

Cited by 0SourceScholar
2025

Exploring the Reliability of Large Language Models as Customized Evaluators for Diverse NLP Tasks

COLING 2025main

Previous work adopts large language models (LLMs) as evaluators to evaluate natural language process (NLP) tasks. However, certain shortcomings, e.g., fairness, scope, and accuracy, persist for current LLM evaluators. To analyze whether LLMs can serve as reliable alternatives to humans, we examine t…

2025

Scaling Diffusion Language Models via Adaptation from Autoregressive Models

ICLR 2025poster

Diffusion Language Models (DLMs) have emerged as a promising new paradigm for text generative modeling, potentially addressing limitations of autoregressive (AR) models. However, current DLMs have been studied at a smaller scale compared to their AR counterparts and lack fair comparison on language…

2025

What Makes a Good Reasoning Chain? Uncovering Structural Patterns in Long Chain-of-Thought Reasoning

EMNLP 2025

Recent advances in reasoning with large language models (LLMs) have popularized Long Chain-of-Thought (LCoT), a strategy that encourages deliberate and step-by-step reasoning before producing a final answer. While LCoTs have enabled expert-level performance in complex tasks, how the internal structu

Cited by 0SourcePDFScholar
2024

A Frustratingly Simple Decoding Method for Neural Text Generation

COLING 2024main

We introduce a frustratingly simple, highly efficient, and surprisingly effective decoding method, termed Frustratingly Simple Decoding (FSD), for neural text generation. The idea behind FSD is straightforward: We construct an anti-language model (anti-LM) based on previously generated text, which i…

2024

BBA: Bi-Modal Behavioral Alignment for Reasoning with Large Vision-Language Models

ACL 2024findings

Multimodal reasoning stands as a pivotal capability for large vision-language models (LVLMs). The integration with Domain-Specific Languages (DSL), offering precise visual representations, equips these models with the opportunity to execute more accurate reasoning in complex and professional domains…

Cited by 6SourcePDFScholar
2024

Diffusion of Thought: Chain-of-Thought Reasoning in Diffusion Language Models

NeurIPS 2024poster

Recently, diffusion models have garnered significant interest in the field of text processing due to their many potential advantages compared to conventional autoregressive models. In this work, we propose Diffusion-of-Thought (DoT), a novel approach that integrates diffusion models with Chain-of-T…

Cited by 9SourcePDFScholar
2024

GSM-Plus: A Comprehensive Benchmark for Evaluating the Robustness of LLMs as Mathematical Problem Solvers

ACL 2024long

Large language models (LLMs) have achieved impressive performance across various mathematical reasoning benchmarks. However, there are increasing debates regarding whether these models truly understand and apply mathematical knowledge or merely rely on shortcuts for mathematical reasoning. One essen…

2024

Gated Slot Attention for Efficient Linear-Time Sequence Modeling

NeurIPS 2024poster

Linear attention Transformers and their gated variants, celebrated for enabling parallel training and efficient recurrent inference, still fall short in recall-intensive tasks compared to traditional Transformers and demand significant resources for training from scratch. This paper introduces Gated…

2024

Knowledge Fusion of Large Language Models

ICLR 2024poster

While training large language models (LLMs) from scratch can generate models with distinct functionalities and strengths, it comes at significant costs and may result in redundant capabilities. Alternatively, a cost-effective and compelling approach is to merge existing pre-trained LLMs into a more…

2024

Knowledge Verification to Nip Hallucination in the Bud

EMNLP 2024main

While large language models (LLMs) have demonstrated exceptional performance across various tasks following human alignment, they may still generate responses that sound plausible but contradict factual knowledge, a phenomenon known as hallucination. In this paper, we demonstrate the feasibility of…

2024

MAGE: Machine-generated Text Detection in the Wild

ACL 2024long

Large language models (LLMs) have achieved human-level text generation, emphasizing the need for effective deepfake text detection to mitigate risks like the spread of fake news and plagiarism. Existing research has been constrained by evaluating detection methods o specific domains or particular la…

2024

SEGO: Sequential Subgoal Optimization for Mathematical Problem-Solving

ACL 2024long

Large Language Models (LLMs) have driven substantial progress in artificial intelligence in recent years, exhibiting impressive capabilities across a wide range of tasks, including mathematical problem-solving. Inspired by the success of subgoal-based methods, we propose a novel framework called SEq…

Cited by 4SourcePDFScholar
2024

Spotting AI’s Touch: Identifying LLM-Paraphrased Spans in Text

ACL 2024findings

AI-generated text detection has attracted increasing attention as powerful language models approach human-level generation. Limited work is devoted to detecting (partially) AI-paraphrased texts. However, AI paraphrasing is commonly employed in various application scenarios for text refinement and di…

2023

Explicit Syntactic Guidance for Neural Text Generation

ACL 2023long

Most existing text generation models follow the sequence-to-sequence paradigm. Generative Grammar suggests that humans generate natural language texts by learning language grammar. We propose a syntax-guided generation schema, which generates the sequence guided by a constituency parse tree in a top…

2023

Explore-Instruct: Enhancing Domain-Specific Instruction Coverage through Active Exploration

EMNLP 2023long main

Instruction-tuning can be substantially optimized through enhanced diversity, resulting in models capable of handling a broader spectrum of tasks. However, existing data employed for such tuning often exhibit an inadequate coverage of individual domains, limiting the scope for nuanced comprehension…

Cited by 0SourcecodeScholar
2023

Masked Jigsaw Puzzle: A Versatile Position Embedding for Vision Transformers

CVPR 2023poster

Position Embeddings (PEs), an arguably indispensable component in Vision Transformers (ViTs), have been shown to improve the performance of ViTs on many vision tasks. However, PEs have a potentially high risk of privacy leakage since the spatial information of the input patches is exposed. This cave…

2023

Multi-Grained Knowledge Retrieval for End-to-End Task-Oriented Dialog

ACL 2023long

Retrieving proper domain knowledge from an external database lies at the heart of end-to-end task-oriented dialog systems to generate informative responses. Most existing systems blend knowledge retrieval with response generation and optimize them with direct supervision from reference responses, le…

2023

Pre-training Multi-party Dialogue Models with Latent Discourse Inference

ACL 2023long

Multi-party dialogues are more difficult for models to understand than one-to-one two-party dialogues, since they involve multiple interlocutors, resulting in interweaving reply-to relations and information flows. To step over these obstacles, an effective way is to pre-train a model that understand…

2023

Retrieval-Generation Alignment for End-to-End Task-Oriented Dialogue System

EMNLP 2023long main

Developing an efficient retriever to retrieve knowledge from a large-scale knowledge base (KB) is critical for task-oriented dialogue systems to effectively handle localized and specialized tasks. However, widely used generative models such as T5 and ChatGPT often struggle to differentiate subtle di…

Cited by 0SourcecodeScholar
2023

RobustGEC: Robust Grammatical Error Correction Against Subtle Context Perturbation

EMNLP 2023long main

Grammatical Error Correction (GEC) systems play a vital role in assisting people with their daily writing tasks. However, users may sometimes come across a GEC system that initially performs well but fails to correct errors when the inputs are slightly modified. To ensure an ideal user experience, a…

Cited by 0SourcecodeScholar
2023

TRAMS: Training-free Memory Selection for Long-range Language Modeling

EMNLP 2023short findings

The Transformer architecture is crucial for numerous AI models, but it still faces challenges in long-range language modeling. Though several specific transformer architectures have been designed to tackle issues of long-range dependencies, existing methods like Transformer-XL are plagued by a high…

Cited by 0SourcecodeScholar
2022

A Model-agnostic Data Manipulation Method for Persona-based Dialogue Generation

ACL 2022long

Towards building intelligent dialogue agents, there has been a growing interest in introducing explicit personas in generation models. However, with limited persona-based dialogue data at hand, it may be difficult to train a dialogue generation model well. We point out that the data challenges of th…

2022

Event Transition Planning for Open-ended Text Generation

ACL 2022findings

Open-ended text generation tasks, such as dialogue generation and story completion, require models to generate a coherent continuation given limited preceding context. The open-ended nature of these tasks brings new challenges to the neural auto-regressive text generators nowadays. Despite these neu…

2022

Lexical Knowledge Internalization for Neural Dialog Generation

ACL 2022long

We propose knowledge internalization (KI), which aims to complement the lexical knowledge into neural dialog models. Instead of further conditioning the knowledge-grounded dialog (KGD) models on externally retrieved knowledge, we seek to integrate knowledge about each input token internally into the…

2022

On Synthetic Data for Back Translation

NAACL 2022long

Back translation (BT) is one of the most significant technologies in NMT research fields. Existing attempts on BT share a common characteristic: they employ either beam search or random sampling to generate synthetic data with a backward model but seldom work studies the role of synthetic data in th…

2021

Efficient Training of Visual Transformers with Small Datasets

NeurIPS 2021poster

Visual Transformers (VTs) are emerging as an architectural paradigm alternative to Convolutional networks (CNNs). Differently from CNNs, VTs can capture global relations between image elements and they potentially have a larger representation capacity. However, the lack of the typical convolutional…

2021

Good for Misconceived Reasons: An Empirical Revisiting on the Need for Visual Context in Multimodal Machine Translation

ACL 2021long

A neural multimodal machine translation (MMT) system is one that aims to perform better translation by extending conventional text-only translation models with multimodal information. Many recent studies report improvements when equipping their models with the multimodal module, despite the controve…

2021

Learning from My Friends: Few-Shot Personalized Conversation Systems via Social Networks

AAAI 2021technical

Personalized conversation models (PCMs) generate responses according to speaker preferences. Existing personalized conversation tasks typically require models to extract speaker preferences from user descriptions or their conversation histories, which are scarce for newcomers and inactive users. In…

2021

Set Generation Networks for End-to-End Knowledge Base Population

EMNLP 2021main

The task of knowledge base population (KBP) aims to discover facts about entities from texts and expand a knowledge base with these facts. Previous studies shape end-to-end KBP as a machine translation task, which is required to convert unordered fact into a sequence according to a pre-specified ord…

Cited by 16SourcePDFScholar
2020

Dual Dynamic Memory Network for End-to-End Multi-turn Task-oriented Dialog Systems

COLING 2020main

Existing end-to-end task-oriented dialog systems struggle to dynamically model long dialog context for interactions and effectively incorporate knowledge base (KB) information into dialog generation. To conquer these limitations, we propose a Dual Dynamic Memory Network (DDMN) for multi-turn dialog…

2020

TableGPT: Few-shot Table-to-Text Generation with Table Structure Reconstruction and Content Matching

COLING 2020main

Although neural table-to-text models have achieved remarkable progress with the help of large-scale datasets, they suffer insufficient learning problem with limited training data. Recently, pre-trained language models show potential in few-shot learning with linguistic knowledge learnt from pretrain…