malteos/scincl-SciNCL SciNCL is a pre-tr...

malteos/scincl-SciNCL SciNCL is a pre-tr...并不只看表面做法，关键还要理解相关条件、限制和后续影响。

SciNCL

SciNCL is a pre-trained BERT language model to generate document-level embeddings of research papers.

It uses the citation graph neighborhood to generate samples for contrastive learning.

Prior to the contrastive training, the model is initialized with weights from scibert-scivocab-uncased.

The underlying citation embeddings are trained on the S2ORC citation graph.

Paper: Neighborhood Contrastive Learning for Scientific Document Representations with Citation Embeddings (EMNLP 2022 paper).

Code: https://github.com/malteos/scincl

PubMedNCL: Working with biomedical papers? Try PubMedNCL.

How to use the pretrained model

from transformers import AutoTokenizer, AutoModel

# load model and tokenizer

tokenizer = AutoTokenizer.from_pretrained('malteos/scincl')

model = AutoModel.from_pretrained('malteos/scincl')

papers = [{'title': 'BERT', 'abstract': 'We introduce a new language representation model called BERT'},

{'title': 'Attention is all you need', 'abstract': ' The dominant sequence transduction models are based on complex recurrent or convolutional neural networks'}]

# concatenate title and abstract with [SEP] token

title_abs = [d['title'] + tokenizer.sep_token + (d.get('abstract') or '') for d in papers]

# preprocess the input

inputs = tokenizer(title_abs, padding=True, truncation=True, return_tensors="pt", max_length=512)

# inference

result = model(inputs)

# take the first token ([CLS] token) in the batch as the embedding

embeddings = result.last_hidden_state[:, 0, :]

Triplet Mining Parameters

malteos/scincl官网入口：https://huggingface.co/malteos/scincl