HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

SimCLS: A Simple Framework for Contrastive Learning of Abstractive Summarization

Yixin Liu Pengfei Liu

SimCLS: A Simple Framework for Contrastive Learning of Abstractive Summarization

Abstract

In this paper, we present a conceptually simple while empirically powerful framework for abstractive summarization, SimCLS, which can bridge the gap between the learning objective and evaluation metrics resulting from the currently dominated sequence-to-sequence learning framework by formulating text generation as a reference-free evaluation problem (i.e., quality estimation) assisted by contrastive learning. Experimental results show that, with minor modification over existing top-scoring systems, SimCLS can improve the performance of existing top-performing models by a large margin. Particularly, 2.51 absolute improvement against BART and 2.50 over PEGASUS w.r.t ROUGE-1 on the CNN/DailyMail dataset, driving the state-of-the-art performance to a new level. We have open-sourced our codes and results: https://github.com/yixinL7/SimCLS. Results of our proposed models have been deployed into ExplainaBoard platform, which allows researchers to understand our systems in a more fine-grained way.

Code Repositories

andrejmiscic/simcls-pytorch
pytorch
Mentioned in GitHub
yixinL7/SimCLS
Official
pytorch

Benchmarks

BenchmarkMethodologyMetrics
abstractive-text-summarization-on-cnn-dailyBART + SimCLS
ROUGE-1: 46.67
ROUGE-2: 22.15
ROUGE-L: 43.54
text-summarization-on-x-sumPEGASUS + SimCLS
ROUGE-1: 47.61
ROUGE-2: 24.57
ROUGE-L: 39.44

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp