CoolFace
Modelpublic

SIRIS-Lab/social-sciences_binary_classifier

sourceHugging Faceupdated 3mo agoView on Hugging Face
0likes8downloads
Model Card

๐Ÿ“— SPECTER2 โ€“ Social Sciences Classifier (Binary Classification)

This model is a fine-tuned version of allenai/specter2_base for identifying whether a scientific publication belongs to the Social Sciences domain.

It achieves the following results on the evaluation set:

  • โ€”Loss: 0.1382
  • โ€”Accuracy: 0.9670
  • โ€”F1 Micro: 0.9670
  • โ€”F1 Macro: 0.9480
  • โ€”F1 Weighted: 0.9670

Model description

This model performs binary document classification and predicts whether a publication belongs to the Social Sciences domain.

The model was trained using title and abstract text from multiple openly available datasets with native disciplinary annotations, including:

Each dataset was converted into a common binary label indicating whether a document belongs to the Social Sciences according to mappings from the original classification systems. :contentReference[oaicite:0]{index=0}

Key characteristics

  • โ€”Base model: allenai/specter2_base
  • โ€”Task: binary document classification
  • โ€”Labels:
  • โ€”False โ†’ Non-Social Sciences
  • โ€”True โ†’ Social Sciences
  • โ€”Activation: softmax
  • โ€”Loss: CrossEntropyLoss

Intended uses

This model is suitable for:

  • โ€”Identifying Social Sciences publications
  • โ€”Research information systems
  • โ€”Funding portfolio analysis
  • โ€”Metadata enrichment
  • โ€”Bibliometric analyses

The model accepts:

  • โ€”title
  • โ€”abstract
  • โ€”title + abstract (recommended)

Training data

Training data combines approximately 20,000 documents sampled from multiple sources:

  • โ€”MAG/SciDocs
  • โ€”ERC panel datasets (publications and projects)
  • โ€”Elsevier Open Access publications

Each source provides its own disciplinary taxonomy. Categories corresponding to Social Sciences were mapped into a common binary classification problem. :contentReference[oaicite:1]{index=1}

Training procedure

Preprocessing

  • โ€”Input text: title + abstract
  • โ€”Maximum sequence length: 512 tokens
  • โ€”Tokenization using the SPECTER2 tokenizer

Training hyperparameters

  • โ€”learning_rate: 2e-5
  • โ€”trainbatchsize: 32
  • โ€”evalbatchsize: 32
  • โ€”num_epochs: 4
  • โ€”max_length: 512
  • โ€”optimizer: AdamW
  • โ€”metric for best model: F1 Macro

Evaluation results

MetricValue
Accuracy0.9670
F1 Micro0.9670
F1 Macro0.9480
F1 Weighted0.9670

Limitations

  • โ€”The model predicts whether a publication belongs to the Social Sciences domain only.
  • โ€”It does not distinguish between individual Social Sciences disciplines.
  • โ€”Labels are derived from mappings between different disciplinary taxonomies and should be interpreted as high-level domain assignments.

Framework versions

  • โ€”Transformers 4.57.1
  • โ€”PyTorch 2.8.0
  • โ€”Datasets 3.6.0
  • โ€”Tokenizers 0.22.1