← Search

Canwen Xu

21 accepted papers

2026

Agent World Model: Infinity Synthetic Environments for Agentic Reinforcement Learning

ICML 2026poster

Recent advances in large language model (LLM) have empowered autonomous agents to perform complex tasks that require multi-turn interactions with external tools and environments. However, scaling such agent training is limited by the lack of diverse and reliable environments. In this paper, we propo…

Cited by 0SourceScholar
2024

Automatic Pair Construction for Contrastive Post-training

NAACL 2024findings

Alignment serves as an important step to steer large language models (LLMs) towards human preferences. In this paper, we propose an automatic way to construct contrastive data for LLM, using preference pairs from multiple models of varying strengths (e.g., InstructGPT, ChatGPT and GPT-4). We compare…

2024

RepoBench: Benchmarking Repository-Level Code Auto-Completion Systems

ICLR 2024poster

Large Language Models (LLMs) have greatly advanced code auto-completion systems, with a potential for substantial productivity enhancements for developers. However, current benchmarks mainly focus on single-file tasks, leaving an assessment gap for more complex, real-world, multi-file programming sc…

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…

2023

Baize: An Open-Source Chat Model with Parameter-Efficient Tuning on Self-Chat Data

EMNLP 2023long main

Chat models, such as ChatGPT, have shown impressive capabilities and have been rapidly adopted across numerous domains. However, these models are only accessible through a restricted API, creating barriers for new research and progress in the field. We propose a pipeline that can automatically gener…

Cited by 0SourceScholar
2023

LongCoder: A Long-Range Pre-trained Language Model for Code Completion

ICML 2023poster

In this paper, we introduce a new task for code completion that focuses on handling long code input and propose a sparse Transformer model, called LongCoder, to address this task. LongCoder employs a sliding window mechanism for self-attention and introduces two types of globally accessible tokens -…

2022

Automatic Multi-Label Prompting: Simple and Interpretable Few-Shot Classification

NAACL 2022long

Prompt-based learning (i.e., prompting) is an emerging paradigm for exploiting knowledge learned by a pretrained language model. In this paper, we propose Automatic Multi-Label Prompting (AMuLaP), a simple yet effective method to automatically select label mappings for few-shot text classification w…

2022

InforMask: Unsupervised Informative Masking for Language Model Pretraining

EMNLP 2022main

Masked language modeling is widely used for pretraining large language models for natural language understanding (NLU). However, random masking is suboptimal, allocating an equal masking rate for all tokens. In this paper, we propose InforMask, a new unsupervised masking strategy for training masked…

2022

LaPraDoR: Unsupervised Pretrained Dense Retriever for Zero-Shot Text Retrieval

ACL 2022findings

In this paper, we propose LaPraDoR, a pretrained dual-tower dense retriever that does not require any supervised data for training. Specifically, we first present Iterative Contrastive Learning (ICoL) that iteratively trains the query and document encoders with a cache mechanism. ICoL not only enlar…

2022

Leashing the Inner Demons: Self-Detoxification for Language Models

AAAI 2022technical

Language models (LMs) can reproduce (or amplify) toxic language seen during training, which poses a risk to their practical application. In this paper, we conduct extensive experiments to study this phenomenon. We analyze the impact of prompts, decoding strategies and training corpora on the output…

Cited by 26SourcePDFScholar
2022

Multitask Prompted Training Enables Zero-Shot Task Generalization

ICLR 2022spotlight

Large language models have recently been shown to attain reasonable zero-shot generalization on a diverse set of tasks (Brown et al., 2020). It has been hypothesized that this is a consequence of implicit multitask learning in language models’ pretraining (Radford et al., 2019). Can zero-shot genera…

2021

Beyond Preserved Accuracy: Evaluating Loyalty and Robustness of BERT Compression

EMNLP 2021main

Recent studies on compression of pretrained language models (e.g., BERT) usually use preserved accuracy as the metric for evaluation. In this paper, we propose two new metrics, label loyalty and probability loyalty that measure how closely a compressed model (i.e., student) mimics the original model…

2021

Blow the Dog Whistle: A Chinese Dataset for Cant Understanding with Common Sense and World Knowledge

NAACL 2021long

Cant is important for understanding advertising, comedies and dog-whistle politics. However, computational research on cant is hindered by a lack of available datasets. In this paper, we propose a large and diverse Chinese dataset for creating and understanding cant from a computational linguistics…

2021

Datasets: A Community Library for Natural Language Processing

EMNLP 2021system demonstrations

The scale, variety, and quantity of publicly-available NLP datasets has grown rapidly as researchers propose new tasks, larger models, and novel benchmarks. Datasets is a community library for contemporary NLP designed to support this ecosystem. Datasets aims to standardize end-user interfaces, vers…

2021

Improving Sequence-to-Sequence Pre-training via Sequence Span Rewriting

EMNLP 2021main

In this paper, we propose Sequence Span Rewriting (SSR), a self-supervised task for sequence-to-sequence (Seq2Seq) pre-training. SSR learns to refine the machine-generated imperfect text spans into ground truth text. SSR provides more fine-grained and informative supervision in addition to the origi…

2020

BERT Loses Patience: Fast and Robust Inference with Early Exit

NeurIPS 2020poster

In this paper, we propose Patience-based Early Exit, a straightforward yet effective inference method that can be used as a plug-and-play technique to simultaneously improve the efficiency and robustness of a pretrained language model (PLM). To achieve this, our approach couples an internal-classifi…