README.md

metadata

language:
  - en
tags:
  - summarization
license: apache-2.0
datasets:
  - arxiv_dataset
metrics:
  - rouge

Paper Title Generator

Generates titles for computer science papers given an abstract.

The model is a BERT2BERT Encoder-Decoder using the official bert-base-uncased checkpoint as initialization for the encoder and decoder. It was fine-tuned on 318,500 computer science papers posted on arXiv.org between 2007 and 2020 and achieved a 26.3% Rouge2 F1-Score on held-out validation data.