← Search

Yi Mao

11 accepted papers

2024

Enhancing Language Model Alignment: A Confidence-Based Approach to Label Smoothing

EMNLP 2024main

In recent years, Large Language Models (LLMs) have demonstrated remarkable capabilities across various domains. Within the training pipeline of LLMs, the Reinforcement Learning with Human Feedback (RLHF) phase is crucial for aligning LLMs with human preferences and values. Label smoothing, a techniq…

Cited by 0SourcePDFScholar
2023

HyperTuning: Toward Adapting Large Language Models without Back-propagation

ICML 2023poster

Fine-tuning large language models for different tasks can be costly and inefficient, and even methods that reduce the number of tuned parameters still require full gradient-based optimization. We propose HyperTuning, a novel approach to model adaptation that uses a hypermodel to generate task-specif…

Cited by 34SourcePDFScholar
2023

RepoCoder: Repository-Level Code Completion Through Iterative Retrieval and Generation

EMNLP 2023long main

The task of repository-level code completion is to continue writing the unfinished code based on a broader context of the repository. While for automated code completion tools, it is difficult to utilize the useful information scattered in different files. We propose RepoCoder, a simple, generic, an…

Cited by 0SourcecodeScholar
2022

A Token-level Reference-free Hallucination Detection Benchmark for Free-form Text Generation

ACL 2022long

Large pretrained generative models like GPT-3 often suffer from hallucinating non-existent or incorrect content, which undermines their potential merits in real applications. Existing work usually attempts to detect these hallucinations based on a corresponding oracle reference at a sentence or docu…

2022

An End-to-End Dialogue Summarization System for Sales Calls

NAACL 2022industry

Summarizing sales calls is a routine task performed manually by salespeople. We present a production system which combines generative models fine-tuned for customer-agent setting, with a human-in-the-loop user experience for an interactive summary curation process. We address challenging aspects of…

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

Knowledge-Grounded Dialogue Generation with a Unified Knowledge Representation

NAACL 2022long

Knowledge-grounded dialogue systems are challenging to build due to the lack of training data and heterogeneous knowledge sources. Existing systems perform poorly on unseen topics due to limited topics covered in the training data. In addition, it is challenging to generalize to the domains that req…

2022

OmniTab: Pretraining with Natural and Synthetic Data for Few-shot Table-based Question Answering

NAACL 2022long

The information in tables can be an important complement to text, making table-based question answering (QA) systems of great value. The intrinsic complexity of handling tables often adds an extra burden to both model design and data annotation. In this paper, we aim to develop a simple table-based…

2021

Finetuning Pretrained Transformers into RNNs

EMNLP 2021main

Transformers have outperformed recurrent neural networks (RNNs) in natural language generation. But this comes with a signifi- cant computational cost, as the attention mechanism’s complexity scales quadratically with sequence length. Efficient transformer variants have received increasing interest…

2018

Action-dependent Control Variates for Policy Optimization via Stein Identity

ICLR 2018poster

Policy gradient methods have achieved remarkable successes in solving challenging reinforcement learning problems. However, it still often suffers from the large variance issue on policy gradient estimation, which leads to poor sample efficiency during training. In this work, we propose a control va…

Cited by 100SourcePDFScholar