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
83
Safetensors
Model size
247M params
Tensor type
I64
ยท
F32
ยท
Inference API
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

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

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