← Search

Dejiao Zhang

16 accepted papers

2024

CODE REPRESENTATION LEARNING AT SCALE

ICLR 2024poster

Recent studies have shown that code language model at scale demonstrate significant performance gains on downstream tasks, i.e., code generation. However, most of the existing works on code representation learning train models at a hundred million parameter scale using very limited pretraining corpo…

Cited by 18SourcePDFScholar
2024

Reasoning in Token Economies: Budget-Aware Evaluation of LLM Reasoning Strategies

EMNLP 2024main

A diverse array of reasoning strategies has been proposed to elicit the capabilities of large language models. However, in this paper, we point out that traditional evaluations which focus solely on performance metrics miss a key factor: the increased effectiveness due to additional compute. By over…

2024

Repoformer: Selective Retrieval for Repository-Level Code Completion

ICML 2024oral

Recent advances in retrieval-augmented generation (RAG) have initiated a new era in repository-level code completion. However, the invariable use of retrieval in existing methods exposes issues in both efficiency and robustness, with a large proportion of the retrieved contexts proving unhelpful or…

Cited by 30SourcePDFScholar
2023

ContraCLM: Contrastive Learning For Causal Language Model

ACL 2023long

Despite exciting progress in causal language models, the expressiveness of their representations is largely limited due to poor discrimination ability. To remedy this issue, we present CONTRACLM, a novel contrastive learning framework at both the token-level and the sequence-level. We assess CONTRAC…

2023

Exploring Continual Learning for Code Generation Models

ACL 2023short

Large-scale code generation models such as Copilot and CodeT5 have achieved impressive performance. However, libraries are upgraded or deprecated very frequently and re-training large-scale language models is computationally expensive. Therefore, Continual Learning (CL) is an important aspect that r…

2023

Multitask Pretraining with Structured Knowledge for Text-to-SQL Generation

ACL 2023long

Many machine learning-based low-code or no-code applications involve generating code that interacts with structured knowledge. For example, one of the most studied tasks in this area is generating SQL code from a natural language statement. Prior work shows that incorporating context information fro…

2022

Learning Dialogue Representations from Consecutive Utterances

NAACL 2022long

Learning high-quality dialogue representations is essential for solving a variety of dialogue-oriented tasks, especially considering that dialogue systems often suffer from data scarcity. In this paper, we introduce Dialogue Sentence Embedding (DSE), a self-supervised contrastive learning method tha…

2022

Lifelong Pretraining: Continually Adapting Language Models to Emerging Corpora

NAACL 2022long

Pretrained language models (PTLMs) are typically learned over a large, static corpus and further fine-tuned for various downstream tasks. However, when deployed in the real world, a PTLM-based model must deal with data distributions that deviates from what the PTLM was initially trained on. In this…

2022

Virtual Augmentation Supported Contrastive Learning of Sentence Representations

ACL 2022findings

Despite profound successes, contrastive representation learning relies on carefully designed data augmentations using domain-specific knowledge. This challenge is magnified in natural language processing, where no general rules exist for data augmentation due to the discrete nature of natural langua…

2021

Answering Ambiguous Questions through Generative Evidence Fusion and Round-Trip Prediction

ACL 2021long

In open-domain question answering, questions are highly likely to be ambiguous because users may not know the scope of relevant topics when formulating them. Therefore, a system needs to find possible interpretations of the question, and predict one or multiple plausible answers. When multiple plaus…

2021

Improving Factual Consistency of Abstractive Summarization via Question Answering

ACL 2021long

A commonly observed problem with the state-of-the art abstractive summarization models is that the generated summaries can be factually inconsistent with the input documents. The fact that automatic summarization may produce plausible-sounding yet inaccurate summaries is a major concern that limits…

2021

Pairwise Supervised Contrastive Learning of Sentence Representations

EMNLP 2021main

Many recent successes in sentence representation learning have been achieved by simply fine-tuning on the Natural Language Inference (NLI) datasets with triplet loss or siamese loss. Nevertheless, they share a common weakness: sentences in a contradiction pair are not necessarily from different sema…

2021

Supporting Clustering with Contrastive Learning

NAACL 2021long

Unsupervised clustering aims at discovering the semantic categories of data according to some distance measured in the representation space. However, different categories often overlap with each other in the representation space at the beginning of the learning process, which poses a significant cha…

2018

LEARNING TO SHARE: SIMULTANEOUS PARAMETER TYING AND SPARSIFICATION IN DEEP LEARNING

ICLR 2018poster

Deep neural networks (DNNs) usually contain millions, maybe billions, of parameters/weights, making both storage and computation very expensive. This has motivated a large body of work to reduce the complexity of the neural network by using sparsity-inducing regularizers. Another well-known approac…

2016

Global Convergence of a Grassmannian Gradient Descent Algorithm for Subspace Estimation

AISTATS 2016poster

It has been observed in a variety of contexts that gradient descent methods have great success in solving low-rank matrix factorization problems, despite the relevant problem formulation being non-convex. We tackle a particular instance of this scenario, where we seek the d-dimensional subspace span…

Cited by 65SourcePDFScholar