Edit model card

Model Card for RoBERTa Social Roles Classifier

This model is a token classifier that extracts social roles from explicit expressions of self-identification in sentences, e.g. I am a designer, entrepreneur, and mother.

Model Details

Model Description

We continue pretraining RoBERTa-base for 10 epochs on individuals' about pages, which is a subset of Common Crawl and can be accessed here.

Then, we finetuned on hand-annotated token-level labels as described in this paper. We use a train-dev-test split of 600/200/200 labeled sentences.

Our definition of "roles" or "occupations" on about pages is any singular noun referring to the subject of the bio. The roles and occupations can be ones that the subject actively participated in the past, e.g. Throughout my life I have been a teacher, a startup founder, and a seashell collector.

Subject of the about page

  • First person biographies: the subject is I, me, my, mine.
  • Third person biographies: we assume the bio’s subject is the main person referenced in the excerpt sentence.

Positive examples of self-identification

  • I am a chef, author, and mom living in Virginia.
  • As an award-winning geologist, Sebastian has given talks around the world.
  • Knitter, blogger, & dreamer. In the last example above, the sentence’s relation to the subject of the bio is implied rather than stated.

Negative examples

  • My wife loves beekeeping as well.
  • Janice works hard to accommodate every client.

Language(s) (NLP): English

License: Apache 2.0

Uses

We use tagged social roles in web pages to assess the social impact of LLM pretraining data curation decisions. Text linked to descriptions of their creators can also facilitate other areas of research, including self-presentation and language variation.

Evaluation

On our test set, we achieve a precision score of 0.856, recall score of 0.945, and F1 score of 0.898.

Citation

@misc{lucy2024aboutme,
      title={AboutMe: Using Self-Descriptions in Webpages to Document the Effects of English Pretraining Data Filters}, 
      author={Li Lucy and Suchin Gururangan and Luca Soldaini and Emma Strubell and David Bamman and Lauren Klein and Jesse Dodge},
      year={2024},
      eprint={2401.06408},
      archivePrefix={arXiv},
      primaryClass={cs.CL}
}

Contact

lucy3_li@berkeley.edu

Downloads last month
3
Inference Examples
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.