/ Модель

Характеристики

Контекст
512 токенов
Параметры
205 млн
Архитектура
bert · dense

russian-sensitive-topics

Классификация текстаТекст → label5 профессий

BERT-классификатор для выявления 18 чувствительных тем в русскоязычных текстах.

359скачиваний за 30 дней23лайкавеса обновлены18 мая 2021

Бенчмарки

Как модель справилась с задачами пользователей ailist и что о ней публикуют открытые лидерборды

Публичные оценки

Из открытых лидербордов. Сравнивайте результаты одного теста при одинаковых условиях — отсутствие оценки не значит, что модель хуже.

В подключённых источниках результатов для этой модели пока нет.

Что умеет модель

Описание составлено по карточке модели и метаданным Hugging Face

apanc/russian-sensitive-topics — русскоязычный классификатор чувствительных тем, обученный на расширенной версии датасета, описанного в статье для семинара Balto-Slavic NLP на EACL 2021. Модель построена на архитектуре BERT и содержит 205,1 млн параметров; она относит текст к комбинациям из 18 чувствительных тем.

Модель предназначена для многотемной классификации русскоязычных текстов, например для выявления упоминаний преступлений и наркотиков. Максимальная длина контекста — 512 токенов. В README приведены метрики по нескольким классам; авторы отмечают, что тестовая выборка состоит из вручную размеченной части данных, поэтому некоторые темы представлены в ней недостаточно хорошо. Модель доступна в форматах PyTorch, TensorFlow и Flax. Лицензия CC BY-NC-SA 4.0 разрешает использование с указанием авторства и на условиях распространения производных материалов, но запрещает коммерческое использование.

Сценарии применения

  • выявление упоминаний преступлений в русскоязычных текстах
  • поиск публикаций о наркотиках
  • многотемная разметка чувствительных тем
  • фильтрация и анализ русскоязычного пользовательского контента

Ограничения

Качество по отдельным темам различается: тестовая выборка содержит мало вручную размеченных примеров некоторых классов. Коммерческое использование запрещено лицензией CC BY-NC-SA 4.0.

Для каких профессий подходит5

Разделы каталога профессий, в задачах которых используется эта модель

Все профессии

Также подходит для

Карточка модели

README от автора на Hugging Face, как он его опубликовал

Открыть на Hugging Face

General concept of the model

This model is trained on the dataset of sensitive topics of the Russian language. The concept of sensitive topics is described in this article presented at the workshop for Balto-Slavic NLP at the EACL-2021 conference. Please note that this article describes the first version of the dataset, while the model is trained on the extended version of the dataset open-sourced on our GitHub or on kaggle. The properties of the dataset is the same as the one described in the article, the only difference is the size.

Instructions

The model predicts combinations of 18 sensitive topics described in the article. You can find step-by-step instructions for using the model here

Metrics

The dataset partially manually labeled samples and partially semi-automatically labeled samples. Learn more in our article. We tested the performance of the classifier only on the part of manually labeled data that is why some topics are not well represented in the test set.

precisionrecallf1-scoresupport
offline_crime0.650.550.6132
online_crime0.50.460.4837
drugs0.870.90.8887
gambling0.50.670.576
pornography0.730.590.65204
prostitution0.750.690.7291
slavery0.720.720.7340
suicide0.330.290.317
terrorism0.680.570.6247
weapons0.890.830.86138
body_shaming0.90.670.77109
health_shaming0.840.550.66108
politics0.680.540.6241
racism0.810.590.68204
religion0.940.720.81102
sexual_minorities0.690.460.55102
sexism0.660.640.65132
social_injustice0.560.370.45181
none0.620.670.64250
micro avg0.720.610.662218
macro avg0.70.60.642218
weighted avg0.730.610.662218
samples avg0.750.660.682218

Licensing Information

Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International License.

CC BY-NC-SA 4.0

Citation

If you find this repository helpful, feel free to cite our publication:

@inproceedings{babakov-etal-2021-detecting,
    title = "Detecting Inappropriate Messages on Sensitive Topics that Could Harm a Company{'}s Reputation",
    author = "Babakov, Nikolay  and
      Logacheva, Varvara  and
      Kozlova, Olga  and
      Semenov, Nikita  and
      Panchenko, Alexander",
    booktitle = "Proceedings of the 8th Workshop on Balto-Slavic Natural Language Processing",
    month = apr,
    year = "2021",
    address = "Kiyv, Ukraine",
    publisher = "Association for Computational Linguistics",
    url = "https://www.aclweb.org/anthology/2021.bsnlp-1.4",
    pages = "26--36",
    abstract = "Not all topics are equally {``}flammable{''} in terms of toxicity: a calm discussion of turtles or fishing less often fuels inappropriate toxic dialogues than a discussion of politics or sexual minorities. We define a set of sensitive topics that can yield inappropriate and toxic messages and describe the methodology of collecting and labelling a dataset for appropriateness. While toxicity in user-generated data is well-studied, we aim at defining a more fine-grained notion of inappropriateness. The core of inappropriateness is that it can harm the reputation of a speaker. This is different from toxicity in two respects: (i) inappropriateness is topic-related, and (ii) inappropriate message is not toxic but still unacceptable. We collect and release two datasets for Russian: a topic-labelled dataset and an appropriateness-labelled dataset. We also release pre-trained classification models trained on this data.",
}

Как запустить

На своей GPU в ailist. Цена — по текущим предложениям провайдеров

GPU в ailist

Универсальный рантайм · text-classification

Можно запустить

Подбираем GPU…

Класс архитектуры
BertForSequenceClassification
Ревизия весов
a5deed3

0 моделей