← Search

Nan Duan

103 accepted papers

2026

Flash-GRPO: Efficient Alignment for Video Diffusion via One-Step Policy Optimization

ICML 2026poster

Group Relative Policy Optimization has emerged as essential for aligning video diffusion models with human preferences, but faces a critical computational bottleneck: training a 14B parametered model typically demands hundreds of GPU days per experiment. Existing efficiency methods reduce costs thro…

Cited by 0SourceScholar
2026

SetPO: Set-Level Policy Optimization for Diversity-Preserving LLM Reasoning

ICML 2026poster

Reinforcement learning with verifiable rewards has shown notable effectiveness in enhancing large language models (LLMs) reasoning performance, especially in mathematics tasks. However, such improvements often come with reduced outcome diversity, where the model concentrates probability mass on a na…

Cited by 0SourceScholar
2025

Alchemy: Amplifying Theorem-Proving Capability Through Symbolic Mutation

ICLR 2025poster

Formal proofs are challenging to write even for experienced experts. Recent progress in Neural Theorem Proving (NTP) shows promise in expediting this process. However, the formal corpora available on the Internet are limited compared to the general text, posing a significant data scarcity challenge…

2025

Automated Proof Generation for Rust Code via Self-Evolution

ICLR 2025poster

Ensuring correctness is crucial for code generation. Formal verification offers a definitive assurance of correctness, but demands substantial human effort in proof construction and hence raises a pressing need for automation. The primary obsta- cle lies in the severe lack of data—there is much fewe…

Cited by 4SourcePDFScholar
2025

Generative Pre-trained Autoregressive Diffusion Transformer

NeurIPS 2025poster

In this work, we present GPDiT, a Generative Pre-trained Autoregressive Diffusion Transformer that unifies the strengths of diffusion and autoregressive modeling for long-range video synthesis, within a continuous latent space. Instead of predicting discrete tokens, GPDiT autoregressively predicts f…

Cited by 0SourceScholar
2025

Key-Point-Driven Data Synthesis with Its Enhancement on Mathematical Reasoning

AAAI 2025technical

Large language models have shown great potential in complex reasoning tasks, yet their performance is often hampered by the scarcity of high-quality and reasoning-focused training datasets. Addressing this challenge, we propose Key-PointDriven Data Synthesis (KPDDS), a novel data synthesis framework…

2025

Taming Teacher Forcing for Masked Autoregressive Video Generation

CVPR 2025poster

We introduce MAGI, a hybrid video generation framework that combines masked modeling for intra-frame generation with causal modeling for next-frame generation. Our key innovation, Complete Teacher Forcing (CTF), conditions masked frames on complete observation frames rather than masked ones (namely…

Cited by 3SourcePDFScholar
2024

AGIEval: A Human-Centric Benchmark for Evaluating Foundation Models

NAACL 2024findings

Assessing foundation models’ abilities for human-level tasks is crucial for Artificial General Intelligence (AGI) development.Traditional benchmarks, which rely on artificial datasets, may not accurately represent these capabilities. In this paper, we introduce AGIEval, a novel bilingual benchmark d…

2024

AnnoLLM: Making Large Language Models to Be Better Crowdsourced Annotators

NAACL 2024industry

Many natural language processing (NLP) tasks rely on labeled data to train machine learning models with high performance. However, data annotation is time-consuming and expensive, especially when the task involves a large amount of data or requires specialized domains. Recently, GPT-3.5 series model…

2024

CMMLU: Measuring massive multitask language understanding in Chinese

ACL 2024findings

As the capabilities of large language models (LLMs) continue to advance, evaluating their performance is becoming more important and more challenging. This paper aims to address this issue for Mandarin Chinese in the form of CMMLU, a comprehensive Chinese benchmark that covers various subjects, incl…

2024

CRITIC: Large Language Models Can Self-Correct with Tool-Interactive Critiquing

ICLR 2024poster

Recent developments in large language models (LLMs) have been impressive. However, these models sometimes show inconsistencies and problematic behavior, such as hallucinating facts, generating flawed code, or creating offensive and toxic content. Unlike these models, humans typically utilize externa…

2024

Competition-Level Problems are Effective LLM Evaluators

ACL 2024findings

Large language models (LLMs) have demonstrated impressive reasoning capabilities, yet there is ongoing debate about these abilities and the potential data contamination problem recently. This paper aims to evaluate the reasoning capacities of LLMs, specifically in solving recent competition-level pr…

Cited by 23SourcePDFScholar
2024

Enhancing Chain-of-Thoughts Prompting with Iterative Bootstrapping in Large Language Models

NAACL 2024findings

Large language models (LLMs) can achieve impressive performance on various reasoning tasks by incorporating chain-of-thought (CoT) prompting, where step-by-step reasoning is provided to guide LLMs to generate answers to questions, and the question-rationale-answer triplets are utilized as demonstrat…

2024

Enhancing Large Language Models in Coding Through Multi-Perspective Self-Consistency

ACL 2024long

Large language models (LLMs) have exhibited remarkable ability in code generation. However, generating the correct solution in a single attempt still remains a challenge. Prior works utilize verification properties in software engineering to verify and re-rank solutions in a majority voting manner.…

2024

HORIZON: High-Resolution Semantically Controlled Panorama Synthesis

AAAI 2024technical

Panorama synthesis endeavors to craft captivating 360-degree visual landscapes, immersing users in the heart of virtual worlds. Nevertheless, contemporary panoramic synthesis techniques grapple with the challenge of semantically guiding the content generation process. Although recent breakthroughs i…

Cited by 0SourcePDFScholar
2024

Large Language Models Can Learn Representation in Natural Language

ACL 2024findings

One major challenge for Large Language Models (LLMs) is completing complex tasks involving multiple entities, such as tool APIs. To tackle this, one approach is to retrieve relevant entities to enhance LLMs in task completion. A crucial issue here is obtaining accurate natural language representatio…

2024

LayoutNUWA: Revealing the Hidden Layout Expertise of Large Language Models

ICLR 2024poster

Graphic layout generation, a growing research field, plays a significant role in user engagement and information perception. Existing methods primarily treat layout generation as a numerical optimization task, focusing on quantitative aspects while overlooking the semantic information of layout, su…

2024

Learning to Plan by Updating Natural Language

EMNLP 2024finding

Large Language Models (LLMs) have shown remarkable performance in various basic natural language tasks. For completing the complex task, we still need a plan for the task to guide LLMs to generate the specific solutions step by step. LLMs can directly generate task plans, but these plans may still c…

2024

Low-code LLM: Graphical User Interface over Large Language Models

NAACL 2024system demonstrations

Utilizing Large Language Models (LLMs) for complex tasks is challenging, often involving a time-consuming and uncontrollable prompt engineering process. This paper introduces a novel human-LLM interaction framework, Low-code LLM. It incorporates six types of simple low-code visual programming intera…

2024

Machine-Created Universal Language for Cross-Lingual Transfer

AAAI 2024technical

There are two primary approaches to addressing cross-lingual transfer: multilingual pre-training, which implicitly aligns the hidden representations of various languages, and translate-test, which explicitly translates different languages into an intermediate language, such as English. Translate-tes…

2024

Not All Tokens Are What You Need for Pretraining

NeurIPS 2024oral

Previous language model pre-training methods have uniformly applied a next-token prediction loss to all training tokens. Challenging this norm, we posit that ''Not all tokens in a corpus are equally important for language model training''. Our initial analysis examines token-level training dynamics…

Cited by 9SourcePDFScholar
2024

ORES: Open-Vocabulary Responsible Visual Synthesis

AAAI 2024technical

Avoiding synthesizing specific visual concepts is an essential challenge in responsible visual synthesis. However, the visual concept that needs to be avoided for responsible visual synthesis tends to be diverse, depending on the region, context, and usage scenarios. In this work, we formalize a new…

2024

PPTC Benchmark: Evaluating Large Language Models for PowerPoint Task Completion

ACL 2024findings

Recent evaluations of Large Language Models (LLMs) have centered around testing their zero-shot/few-shot capabilities for basic natural language tasks and their ability to translate instructions into tool APIs. However, the evaluation of LLMs utilizing complex tools to finish multi-turn, multi-modal…

2024

PPTC-R benchmark: Towards Evaluating the Robustness of Large Language Models for PowerPoint Task Completion

EMNLP 2024finding

The growing dependence on Large Language Models (LLMs) for finishing user instructions necessitates a comprehensive understanding of their robustness to complex task completion in real-world situations. To address this critical need, we propose the PowerPoint Task Completion-Robustness (PPTC-R) benc…

2024

PROM: A Phrase-level Copying Mechanism with Pre-training for Abstractive Summarization

COLING 2024main

Based on the remarkable achievements of pre-trained language models in abstractive summarization, the copying mechanism has proved helpful by improving the factuality, stability, and overall performance. This work proposes PROM, a new PhRase-level cOpying Mechanism that enhances attention on n-grams…

2024

StrokeNUWA—Tokenizing Strokes for Vector Graphic Synthesis

ICML 2024poster

To leverage LLMs for visual synthesis, traditional methods convert raster image information into discrete grid tokens through specialized visual modules, while disrupting the model’s ability to capture the true semantic representation of visual scenes. This paper posits that an alternative represent…

Cited by 12SourcePDFScholar
2024

ToRA: A Tool-Integrated Reasoning Agent for Mathematical Problem Solving

ICLR 2024poster

Large language models have made significant progress in various language tasks, yet they still struggle with complex mathematics. In this paper, we propose ToRA a series of Tool-integrated Reasoning Agents designed to solve challenging mathematical problems by seamlessly integrating natural language…

2024

Using Left and Right Brains Together: Towards Vision and Language Planning

ICML 2024poster

Large Language Models (LLMs) and Large Multi-modality Models (LMMs) have demonstrated remarkable decision masking capabilities on a variety of tasks. However, they inherently operate planning within the language space, lacking the vision and spatial imagination ability. In contrast, humans utilize b…

Cited by 5SourcePDFScholar
2024

Voila-A: Aligning Vision-Language Models with User's Gaze Attention

NeurIPS 2024spotlight

In recent years, the integration of vision and language understanding has led to significant advancements in artificial intelligence, particularly through Vision-Language Models (VLMs). However, existing VLMs face challenges in handling real-world applications with complex scenes and multiple object…

Cited by 10SourcePDFScholar
2023

AR-Diffusion: Auto-Regressive Diffusion Model for Text Generation

NeurIPS 2023poster

Diffusion models have gained significant attention in the realm of image generation due to their exceptional performance. Their success has been recently expanded to text generation via generating all tokens within a sequence concurrently. However, natural language exhibits a far more pronounced se…

2023

Allies: Prompting Large Language Model with Beam Search

EMNLP 2023long findings

With the advance of large language models (LLMs), the research field of LLM applications becomes more and more popular and the idea of constructing pipelines to accomplish complex tasks by stacking LLM API calls come true. However, this kind of methods face two limitations: narrow information covera…

Cited by 0SourcecodeScholar
2023

Analyzing and Reducing the Performance Gap in Cross-Lingual Transfer with Fine-tuning Slow and Fast

ACL 2023long

Existing research has shown that a multilingual pre-trained language model fine-tuned with one (source) language also performs well on downstream tasks for non-source languages, even though no fine-tuning is done on these languages. However, there is a clear gap between the performance of the source…

Cited by 2SourcePDFScholar
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

BridgeTower: Building Bridges between Encoders in Vision-Language Representation Learning

AAAI 2023technical

Vision-Language (VL) models with the Two-Tower architecture have dominated visual-language representation learning in recent years. Current VL models either use lightweight uni-modal encoders and learn to extract, align and fuse both modalities simultaneously in a deep cross-modal encoder, or feed t…

2023

CAPSTONE: Curriculum Sampling for Dense Retrieval with Document Expansion

EMNLP 2023long main

The dual-encoder has become the de facto architecture for dense retrieval. Typically, it computes the latent representations of the query and document independently, thus failing to fully capture the interactions between the query and document. To alleviate this, recent research has focused on obta…

Cited by 0SourcecodeScholar
2023

CONE: An Efficient COarse-to-fiNE Alignment Framework for Long Video Temporal Grounding

ACL 2023long

This paper tackles an emerging and challenging problem of long video temporal grounding (VTG) that localizes video moments related to a natural language (NL) query. Compared with short videos, long videos are also highly demanded but less explored, which brings new challenges in higher inference com…

2023

Code Execution with Pre-trained Language Models

ACL 2023findings

Code execution is a fundamental aspect of programming language semantics that reflects the exact behavior of the code. However, most pre-trained models for code intelligence ignore the execution trace and only rely on source code and syntactic structures. In this paper, we investigate how well pre-t…

2023

Disentangling Reasoning Capabilities from Language Models with Compositional Reasoning Transformers

ACL 2023findings

This paper presents ReasonFormer, a unified reasoning framework for mirroring the modular and compositional reasoning process of humans in complex decision-making. Inspired by dual-process theory in cognitive science, the representation module (automatic thinking) and reasoning modules (controlled t…

2023

Enhancing Retrieval-Augmented Large Language Models with Iterative Retrieval-Generation Synergy

EMNLP 2023long findings

Retrieval-augmented generation has raise extensive attention as it is promising to address the limitations of large language models including outdated knowledge and hallucinations. However, retrievers struggle to capture relevance, especially for queries with complex information needs. Recent work h…

Cited by 0SourceScholar
2023

Intervention-Based Alignment of Code Search with Execution Feedback

EMNLP 2023long findings

One of the fundamental goals in code search is to retrieve a functionally correct code for a given natural language query. As annotating for correctness requires executing test cases (i.e. obtaining execution feedback), existing code search training datasets approximate text-code co-occurrences as…

Cited by 0SourceScholar
2023

Joint Generator-Ranker Learning for Natural Language Generation

ACL 2023findings

Generate-then-rank is a widely used mechanism for text generation, where a generator produces multiple text candidates and a ranker chooses the best one among the text candidates. However, existing methods usually train the generator and the ranker individually, neglecting the mutual feedback that c…

2023

Learning 3D Photography Videos via Self-supervised Diffusion on Single Images

IJCAI 2023poster

3D photography renders a static image into a video with appealing 3D visual effects. Existing approaches typically first conduct monocular depth estimation, then render the input frame to subsequent frames with various viewpoints, and finally use an inpainting model to fill those missing/occluded re…

Cited by 4SourcePDFScholar
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 -…

2023

ManagerTower: Aggregating the Insights of Uni-Modal Experts for Vision-Language Representation Learning

ACL 2023long

Two-Tower Vision-Language (VL) models have shown promising improvements on various downstream VL tasks. Although the most advanced work improves performance by building bridges between encoders, it suffers from ineffective layer-by-layer utilization of uni-modal representations and cannot flexibly e…

2023

Modeling Sequential Sentence Relation to Improve Cross-lingual Dense Retrieval

ICLR 2023poster

Recently multi-lingual pre-trained language models (PLM) such as mBERT and XLM-R have achieved impressive strides in cross-lingual dense retrieval. Despite its successes, they are general-purpose PLM while the multilingual PLM tailored for cross-lingual retrieval is still unexplored. Motivated by an…

2023

NUWA-XL: Diffusion over Diffusion for eXtremely Long Video Generation

ACL 2023long

In this paper, we propose NUWA-XL, a novel Diffusion over Diffusion architecture for eXtremely Long video generation. Most current work generates long videos segment by segment sequentially, which normally leads to the gap between training on short videos and inferring long videos, and the sequentia…

Cited by 118SourcePDFScholar
2023

ReCo: Region-Controlled Text-to-Image Generation

CVPR 2023poster

Recently, large-scale text-to-image (T2I) models have shown impressive performance in generating high-fidelity images, but with limited controllability, e.g., precisely specifying the content in a specific region with a free-form text description. In this paper, we propose an effective technique for…

2023

Synthetic Prompting: Generating Chain-of-Thought Demonstrations for Large Language Models

ICML 2023poster

Large language models can perform various reasoning tasks by using chain-of-thought prompting, which guides them to find answers through step-by-step demonstrations. However, the quality of the prompts depends on the demonstrations given to the models, and creating many of them by hand is costly. We…

Cited by 105SourcePDFScholar
2023

Text Generation with Diffusion Language Models: A Pre-training Approach with Continuous Paragraph Denoise

ICML 2023poster

In this paper, we introduce a novel dIffusion language modEl pre-training framework for text generation, which we call GENIE. GENIE is a large-scale pre-trained diffusion language model that consists of an encoder and a diffusion-based decoder, which can generate text by gradually transforming a ran…

2022

Adversarial Retriever-Ranker for Dense Text Retrieval

ICLR 2022poster

Current dense text retrieval models face two typical challenges. First, it adopts a siamese dual-encoder architecture to encode query and document independently for fast indexing and searching, whereas neglecting the finer-grained term-wise interactions. This results in a sub-optimal recall performa…

2022

Analytical Reasoning of Text

NAACL 2022findings

Analytical reasoning is an essential and challenging task that requires a system to analyze a scenario involving a set of particular circumstances and perform reasoning over it to make conclusions. However, current neural models with implicit reasoning ability struggle to solve this task. In this pa…

2022

CULG: Commercial Universal Language Generation

NAACL 2022industry

Pre-trained language models (PLMs) have dramatically improved performance for many natural language processing (NLP) tasks in domains such as finance and healthcare. However, the application of PLMs in the domain of commerce, especially marketing and advertising, remains less studied. In this work,…

Cited by 1SourcePDFScholar
2022

CodeExp: Explanatory Code Document Generation

EMNLP 2022finding

Developing models that can automatically generate detailed code explanation can greatly benefit software maintenance and programming education. However, existing code-to-text generation models often produce only high-level summaries of code that do not capture implementation-level choices essential…

2022

CodeRetriever: A Large Scale Contrastive Pre-Training Method for Code Search

EMNLP 2022main

In this paper, we propose the CodeRetriever model, which learns the function-level code semantic representations through large-scale code-text contrastive pre-training. We adopt two contrastive learning schemes in CodeRetriever: unimodal contrastive learning and bimodal contrastive learning. For uni…

2022

Contextual Fine-to-Coarse Distillation for Coarse-grained Response Selection in Open-Domain Conversations

ACL 2022long

We study the problem of coarse-grained response selection in retrieval-based dialogue systems. The problem is equally important with fine-grained response selection, but is less explored in existing literature. In this paper, we propose a Contextual Fine-to-Coarse (CFC) distilled model for coarse-gr…

2022

Cross-Lingual Ability of Multilingual Masked Language Models: A Study of Language Structure

ACL 2022long

Multilingual pre-trained language models, such as mBERT and XLM-R, have shown impressive cross-lingual ability. Surprisingly, both of them use multilingual masked language model (MLM) without any cross-lingual supervision or aligned data. Despite the encouraging results, we still lack a clear unders…

Cited by 13SourcePDFScholar
2022

DialogVED: A Pre-trained Latent Variable Encoder-Decoder Model for Dialog Response Generation

ACL 2022long

Dialog response generation in open domain is an important research topic where the main challenge is to generate relevant and diverse responses. In this paper, we propose a new dialog pre-training framework called DialogVED, which introduces continuous latent variables into the enhanced encoder-deco…

2022

KD-VLP: Improving End-to-End Vision-and-Language Pretraining with Object Knowledge Distillation

NAACL 2022findings

Self-supervised vision-and-language pretraining (VLP) aims to learn transferable multi-modal representations from large-scale image-text data and to achieve strong performances on a broad scope of vision-language tasks after finetuning. Previous mainstream VLP approaches typically adopt a two-step s…

Cited by 32SourcePDFScholar
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

Learning to Complete Code with Sketches

ICLR 2022poster

Code completion is usually cast as a language modelling problem, i.e., continuing an input in a left-to-right fashion. However, in practice, some parts of the completion (e.g., string literals) may be very hard to predict, whereas subsequent parts directly follow from the context. To handle this, we…

Cited by 55SourcePDFScholar
2022

LogiGAN: Learning Logical Reasoning via Adversarial Pre-training

NeurIPS 2022accept

We present LogiGAN, an unsupervised adversarial pre-training framework for improving logical reasoning abilities of language models. Upon automatic identification of logical reasoning phenomena in massive text corpus via detection heuristics, we train language models to predict the masked-out logica…

2022

Logic-Driven Context Extension and Data Augmentation for Logical Reasoning of Text

ACL 2022findings

Logical reasoning of text requires identifying critical logical structures in the text and performing inference over them. Existing methods for logical reasoning mainly focus on contextual semantics of text while struggling to explicitly model the logical inference process. In this paper, we not onl…

2022

Metric-guided Distillation: Distilling Knowledge from the Metric to Ranker and Retriever for Generative Commonsense Reasoning

EMNLP 2022main

Commonsense generation aims to generate a realistic sentence describing a daily scene under the given concepts, which is very challenging, since it requires models to have relational reasoning and compositional generalization capabilities. Previous work focuses on retrieving prototype sentences for…

2022

Mixed-modality Representation Learning and Pre-training for Joint Table-and-Text Retrieval in OpenQA

EMNLP 2022finding

Retrieving evidences from tabular and textual resources is essential for open-domain question answering (OpenQA), which provides more comprehensive information. However, training an effective dense table-text retriever is difficult due to the challenges of table-text discrepancy and data sparsity pr…

2022

Multi-View Document Representation Learning for Open-Domain Dense Retrieval

ACL 2022long

Dense retrieval has achieved impressive advances in first-stage retrieval from a large-scale document collection, which is built on bi-encoder architecture to produce single vector representation of query and document. However, a document can usually answer multiple potential queries from different…

Cited by 65SourcePDFScholar
2022

NUWA-Infinity: Autoregressive over Autoregressive Generation for Infinite Visual Synthesis

NeurIPS 2022accept

Infinite visual synthesis aims to generate high-resolution images, long-duration videos, and even visual generation of infinite size. Some recent work tried to solve this task by first dividing data into processable patches and then training the models on them without considering the dependencies be…

2022

NÜWA: Visual Synthesis Pre-training for Neural visUal World creAtion

ECCV 2022poster

"This paper presents a unified multimodal pre-trained model called NÜWA that can generate new or manipulate existing visual data (i.e., image and video) for various visual synthesis tasks. To cover language, image, and video at the same time for different scenarios, a 3D transformer encoder-decoder…

Cited by 350SourcePDFScholar
2022

ProQA: Structural Prompt-based Pre-training for Unified Question Answering

NAACL 2022long

Question Answering (QA) is a longstanding challenge in natural language processing. Existing QA works mostly focus on specific question types, knowledge domains, or reasoning skills. The specialty in QA research hinders systems from modeling commonalities between tasks and generalization for wider a…

2022

ReACC: A Retrieval-Augmented Code Completion Framework

ACL 2022long

Code completion, which aims to predict the following code token(s) according to the code context, can improve the productivity of software development. Recent work has proved that statistical language modeling with transformers can greatly improve the performance in the code completion task via lear…

2022

Reasoning over Hybrid Chain for Table-and-Text Open Domain Question Answering

IJCAI 2022poster

Tabular and textual question answering requires systems to perform reasoning over heterogeneous information, considering table structure, and the connections among table and text. In this paper, we propose a ChAin-centric Reasoning and Pre-training framework (CARP). CARP utilizes hybrid chain to mod…

2022

Sentiment-Aware Word and Sentence Level Pre-training for Sentiment Analysis

EMNLP 2022main

Most existing pre-trained language representation models (PLMs) are sub-optimal in sentiment analysis tasks, as they capture the sentiment information from word-level while under-considering sentence-level information. In this paper, we propose SentiWSP, a novel Sentiment-aware pre-trained language…

2022

SimANS: Simple Ambiguous Negatives Sampling for Dense Text Retrieval

EMNLP 2022industry

Sampling proper negatives from a large document pool is vital to effectively train a dense retrieval model. However, existing negative sampling strategies suffer from the uninformative or false negative problem. In this work, we empirically show that according to the measured relevance scores, the n…

2022

Soft-Labeled Contrastive Pre-Training for Function-Level Code Representation

EMNLP 2022finding

Code contrastive pre-training has recently achieved significant progress on code-related tasks. In this paper, we present SCodeR, a Soft-labeled contrastive pre-training framework with two positive sample construction methods to learn functional-level Code Representation. Considering the relevance b…

2022

Towards Compositional Generalization in Code Search

EMNLP 2022main

We study compositional generalization, which aims to generalize on unseen combinations of seen structural elements, for code search. Unlike existing approaches of partially pursuing this goal, we study how to extract structural elements, which we name a template that directly targets compositional g…

2022

UniXcoder: Unified Cross-Modal Pre-training for Code Representation

ACL 2022long

Pre-trained models for programming languages have recently demonstrated great success on code intelligence. To support both code-related understanding and generation tasks, recent works attempt to pre-train unified encoder-decoder models. However, such encoder-decoder framework is sub-optimal for au…

2022

Unsupervised Context Aware Sentence Representation Pretraining for Multi-lingual Dense Retrieval

IJCAI 2022poster

Recent research demonstrates the effectiveness of using pretrained language models (PLM) to improve dense retrieval and multilingual dense retrieval. In this work, we present a simple but effective monolingual pretraining task called contrastive context prediction (CCP) to learn sentence representat…

2022

XLM-K: Improving Cross-Lingual Language Model Pre-training with Multilingual Knowledge

AAAI 2022technical

Cross-lingual pre-training has achieved great successes using monolingual and bilingual plain text corpora. However, most pre-trained models neglect multilingual knowledge, which is language agnostic but comprises abundant cross-lingual structure alignment. In this paper, we propose XLM-K, a cross-l…

2021

BANG: Bridging Autoregressive and Non-autoregressive Generation with Large Scale Pretraining

ICML 2021spotlight

In this paper, we propose BANG, a new pretraining model to Bridge the gap between Autoregressive (AR) and Non-autoregressive (NAR) Generation. AR and NAR generation can be uniformly regarded as to what extent previous tokens can be attended, and BANG bridges AR and NAR generation through designing a…

2021

CoSQA: 20,000+ Web Queries for Code Search and Question Answering

ACL 2021long

Finding codes given natural language query is beneficial to the productivity of software developers. Future progress towards better semantic matching between query and code requires richer supervised training resources. To remedy this, we introduce CoSQA dataset. It includes 20,604 labels for pairs…

2021

CodeXGLUE: A Machine Learning Benchmark Dataset for Code Understanding and Generation

NeurIPS 2021poster

Benchmark datasets have a significant impact on accelerating research in programming language tasks. In this paper, we introduce CodeXGLUE, a benchmark dataset to foster machine learning research for program understanding and generation. CodeXGLUE includes a collection of 10 tasks across 14 datasets…

Cited by 981SourcecodeScholar
2021

Compare to The Knowledge: Graph Neural Fake News Detection with External Knowledge

ACL 2021long

Nowadays, fake news detection, which aims to verify whether a news document is trusted or fake, has become urgent and important. Most existing methods rely heavily on linguistic and semantic features from the news content, and fail to effectively exploit external knowledge which could help determine…

2021

Control Image Captioning Spatially and Temporally

ACL 2021long

Generating image captions with user intention is an emerging need. The recently published Localized Narratives dataset takes mouse traces as another input to the image captioning task, which is an intuitive and efficient way for a user to control what to describe in the image. However, how to effect…

2021

Discovering Representation Sprachbund For Multilingual Pre-Training

EMNLP 2021finding

Multilingual pre-trained models have demonstrated their effectiveness in many multilingual NLP tasks and enabled zero-shot or few-shot transfer from high-resource languages to low-resource ones. However, due to significant typological differences and contradictions between some languages, such model…

2021

EL-Attention: Memory Efficient Lossless Attention for Generation

ICML 2021spotlight

Transformer model with multi-head attention requires caching intermediate results for efficient inference in generation tasks. However, cache brings new memory-related costs and prevents leveraging larger batch size for faster speed. We propose memory-efficient lossless attention (called EL-attentio…

2021

GraphCodeBERT: Pre-training Code Representations with Data Flow

ICLR 2021poster

Pre-trained models for programming language have achieved dramatic empirical improvements on a variety of code-related tasks such as code search, code completion, code summarization, etc. However, existing pre-trained models regard a code snippet as a sequence of tokens, while ignoring the inherent…

2021

KFCNet: Knowledge Filtering and Contrastive Learning for Generative Commonsense Reasoning

EMNLP 2021finding

Pre-trained language models have led to substantial gains over a broad range of natural language processing (NLP) tasks, but have been shown to have limitations for natural language generation tasks with high-quality requirements on the output, such as commonsense generation and ad keyword generatio…

Cited by 28SourcePDFScholar
2021

Learning from Inside: Self-driven Siamese Sampling and Reasoning for Video Question Answering

NeurIPS 2021poster

Recent advances in the video question answering (i.e., VideoQA) task have achieved strong success by following the paradigm of fine-tuning each clip-text pair independently on the pretrained transformer-based model via supervised learning. Intuitively, multiple samples (i.e., clips) should be interd…

Cited by 48SourcePDFScholar
2021

Long-Range Modeling of Source Code Files with eWASH: Extended Window Access by Syntax Hierarchy

EMNLP 2021main

Statistical language modeling and translation with transformers have found many successful applications in program understanding and generation tasks, setting high benchmarks for tools in modern software development environments. The finite context window of these neural models means, however, that…

2021

M3P: Learning Universal Representations via Multitask Multilingual Multimodal Pre-Training

CVPR 2021poster

We present M3P, a Multitask Multilingual Multimodal Pre-trained model that combines multilingual pre-training and multimodal pre-training into a unified framework via multitask pre-training. Our goal is to learn universal representations that can map objects occurred in different modalities or texts…

Cited by 128PDFScholar
2021

Mask Attention Networks: Rethinking and Strengthen Transformer

NAACL 2021long

Transformer is an attention-based neural network, which consists of two sublayers, namely, Self-Attention Network (SAN) and Feed-Forward Network (FFN). Existing research explores to enhance the two sublayers separately to improve the capability of Transformer for text representation. In this paper,…

2021

Poolingformer: Long Document Modeling with Pooling Attention

ICML 2021spotlight

In this paper, we introduce a two-level attention schema, Poolingformer, for long document modeling. Its first level uses a smaller sliding window pattern to aggregate information from neighbors. Its second level employs a larger window to increase receptive fields with pooling attention to reduce b…

2021

WhiteningBERT: An Easy Unsupervised Sentence Embedding Approach

EMNLP 2021finding

Producing the embedding of a sentence in anunsupervised way is valuable to natural language matching and retrieval problems in practice. In this work, we conduct a thorough examination of pretrained model based unsupervised sentence embeddings. We study on fourpretrained models and conduct massive e…

2020

An Enhanced Knowledge Injection Model for Commonsense Generation

COLING 2020main

Commonsense generation aims at generating plausible everyday scenario description based on a set of provided concepts. Digging the relationship of concepts from scratch is non-trivial, therefore, we retrieve prototypes from external knowledge to assist the understanding of the scenario for better de…

Cited by 36SourcePDFScholar
2020

Multi-level Alignment Pretraining for Multi-lingual Semantic Parsing

COLING 2020main

In this paper, we present a multi-level alignment pretraining method in a unified architecture formulti-lingual semantic parsing. In this architecture, we use an adversarial training method toalign the space of different languages and use sentence level and word level parallel corpus assupervision i…

2019

A Tensorized Transformer for Language Modeling

NeurIPS 2019poster

Latest development of neural models has connected the encoder and decoder through a self-attention mechanism. In particular, Transformer, which is solely based on self-attention, has led to breakthroughs in Natural Language Processing (NLP) tasks. However, the multi-head attention mechanism, as a ke…

2019

PasteGAN: A Semi-Parametric Method to Generate Image from Scene Graph

NeurIPS 2019poster

Despite some exciting progress on high-quality image generation from structured (scene graphs) or free-form (sentences) descriptions, most of them only guarantee the image-level semantical consistency, i.e. the generated image matching the semantic meaning of the description. They still lack the inv…

2018

Dialog-to-Action: Conversational Question Answering Over a Large-Scale Knowledge Base

NeurIPS 2018poster

We present an approach to map utterances in conversation to logical forms, which will be executed on a large-scale knowledge base. To handle enormous ellipsis phenomena in conversation, we introduce dialog memory management to manipulate historical entities, predicates, and logical forms when inferr…

Cited by 152SourcePDFScholar
2018

Visual Question Generation as Dual Task of Visual Question Answering

CVPR 2018poster

Visual question answering (VQA) and visual question generation (VQG) are two trending topics in the computer vision, but they are usually explored separately despite their intrinsic complementary relationship. In this paper, we propose an end-to-end unified model, the Invertible Question Answering N…

Cited by 198SourcePDFScholar