/ Модель

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

Контекст
512 токенов
Параметры
534 млн
Лицензия
MIT
Архитектура
deberta-v2 · dense
Библиотека
transformers
Опубликована
16 авг 2024

Political_DEBATE_large_v1.0

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

Модель DeBERTa-v2 для zero-shot и few-shot классификации политических текстов по позиции автора, тематике, языку ненависти и событиям.

1 266скачиваний за 30 дней25лайковвеса обновлены13 дек 2024

Бенчмарки

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

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

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

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

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

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

Political DEBATE large — модель классификации текстов на основе DeBERTa-v2, разработанная Майклом Бернхэмом и соавторами. Она дообучена на наборе PolNLI для классификации политических документов в режиме zero-shot и few-shot: определения позиции автора, выявления языка ненависти, извлечения событий и тематической классификации. Модель сопоставляет текст с короткой гипотезой, заданной пользователем, и оценивает, следует ли эта гипотеза из текста.

Модель содержит 533,8 млн параметров, принимает текстовые входы, поддерживает английский язык и имеет контекст до 512 токенов. Данные обучения включают публикации в соцсетях, новости, законопроекты Конгресса и краткие описания судебных дел. Для few-shot настройки README рекомендует использовать не менее 10 примеров, предпочтительно около 25, и небольшие пакеты по 1–2 примера.

Модель предназначена для анализа политических текстов из разных источников; качество зависит от формулировки гипотезы и близости задачи к обучающим данным. Для supervised-классификации по задачам, явно представленным в обучении, авторы рекомендуют базовую модель. Лицензия — MIT.

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

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

Ограничения

Результат зависит от краткости и точности пользовательской гипотезы; для few-shot настройки рекомендуются как минимум 10 примеров, а для сложных или разнообразных задач может потребоваться больше данных. Оценка в карточке преимущественно основана на тестовом наборе PolNLI.

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

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

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

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

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

Открыть на Hugging Face

Model Details

Political DEBATE (DeBERTa Algorithm for Textual Entailment) is an NLI classifier trained for zero-shot and few-shot classification of political documents.

Zero-shot tutorial
Few-shot tutorial

The model was trained using Moritz Laurer's zero shot model as the foundation, and then trained further using the PolNLI dataset. The model was trained for four classification categories:

  1. Stance detection (i.e. opinion mining).
  2. Hatespeech detection.
  3. Event extraction.
  4. Topic classification.

The PolNLI dataset contains documents from social media, news outlets, congressional bills, court case summaries, and more. Classification should work well across a broad set of document sources and subjects, but for best performance refer to the recommendations section.

If you use this model or the PolNLI data set please cite:

@article{burnham2024political,
  title={Political DEBATE: Efficient Zero-shot and Few-shot Classifiers for Political Text},
  author={Burnham, Michael and Kahn, Kayla and Wang, Ryan Yank and Peng, Rachel X},
  journal={arXiv preprint arXiv:2409.02078},
  year={2024}
}

Using NLI Classifiers

NLI classifiers work by pairing a document (AKA the 'premise') with a 'hypothesis', and determining if the hypothesis is true given the content of the document. The hypothesis is supplied by the user and can be thought of as a simplified version of a prompt for an LLM. It's best to keep hypotheses short and simple with a single classification criteria. If a more complicated hypothesis is necessary, consider few-shot training. For more detailed reading on using NLI classifiers see:

@article{burnham2024stance,
  title={Stance detection: a practical guide to classifying political beliefs in text},
  author={Burnham, Michael},
  journal={Political Science Research and Methods},
  year={2024},
  publisher={Cambridge University Press}
}

Recommendations

  1. Use the Political DEBATE large model for zero-shot and few-shot applications unless your use case is explicitly in the training data, in which case the base model should perform well. The base model is more appropriate for using the model as a supervised classifier.
  2. For best results, format hypotheses similar to the PolNLI dataset. Ex:
  • "This text is about {}"
  • "The author of this text believes {}"
  1. For few-shot training, use a small batch size of 1-2.
  2. Use a minimum of 10 samples for few-shot training, but we generally recommend 25 because the variance of outcomes at 10 can be quite large. You will see more benefit from more samples the more difficult your classification task and varied your data.
  3. Train for 3-5 epochs for few-shot learning.

Evaluation

Evaluation is primarily conducted on the PolNLI test set. No hypotheses in the PolNLI test set are present in the training data in order to estimate zero-shot performance.

Overal performance on the PolNLI test set:

Performance on the PolNLI test set by task:

Few-shot performance on open-text survey answers vs. Llama 3.1 8B (unquantized):

Few-shot performance vs. and Electra classifier trained on 2,000 documents:

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

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

GPU в ailist

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

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

Подбираем GPU…

Класс архитектуры
DebertaV2ForSequenceClassification
Ревизия весов
1a3aff1

0 моделей