Edit model card

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 2022 and achieved a 26.3% Rouge2 F1-Score on held-out validation data.

Live Demo: https://paper-titles.ey.r.appspot.com/

Downloads last month
377
Safetensors
Model size
247M params
Tensor type
I64
ยท
F32
ยท
Inference API
Examples
This model can be loaded on Inference API (serverless).

Dataset used to train Callidior/bert2bert-base-arxiv-titlegen

Spaces using Callidior/bert2bert-base-arxiv-titlegen 4