/ Модель

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

Контекст
514 токенов
Параметры
152 млн
Архитектура
roberta · dense

roberta-base-STS-B

Классификация текстаТекст → label

Дообученная RoBERTa-base оценивает семантическое сходство пар предложений в задаче STS-B.

106скачиваний за 30 дней0лайковвеса обновлены20 мая 2021

Бенчмарки

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

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

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

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

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

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

Модель textattack/roberta-base-STS-B — дообученная TextAttack модель RoBERTa-base для оценки семантического сходства двух предложений (STS-B) как задачи регрессии. Она принимает текстовую пару и предсказывает оценку сходства; обучалась на наборе GLUE с функцией потерь mean squared error. Лучший результат на оценочной выборке — корреляция Пирсона 0,911 после четырёх эпох обучения.

Модель содержит около 151,8 млн параметров, поддерживает входные последовательности длиной до 514 токенов по метаданным; при дообучении использовалась максимальная длина 128. Доступны веса PyTorch и Flax. README не сообщает сведения о лицензии, поэтому условия использования по имеющимся данным определить нельзя.

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

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

Ограничения

Обучена для англоязычного бенчмарка STS-B; перенос на другие языки и домены не подтверждён. При дообучении использовалась максимальная длина последовательности 128 токенов, что может ограничивать оценку длинных текстов.

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

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

Открыть на Hugging Face

TextAttack Model Card

This roberta-base model was fine-tuned for sequence classification using TextAttack and the glue dataset loaded using the nlp library. The model was fine-tuned for 5 epochs with a batch size of 8, a learning rate of 2e-05, and a maximum sequence length of 128. Since this was a regression task, the model was trained with a mean squared error loss function. The best score the model achieved on this task was 0.9108696741479216, as measured by the eval set pearson correlation, found after 4 epochs.

For more information, check out TextAttack on Github.

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

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

GPU в ailist

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

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

Подбираем GPU…

Класс архитектуры
RobertaForSequenceClassification
Ревизия весов
3bea43e

0 моделей