/ Модель

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

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

cs224n-squad2.0-albert-base-v2

Ответы на вопросыТекст → текст3 профессии

Дообученная на SQuAD 2.0 английская ALBERT-модель отвечает на вопросы по контексту и определяет отсутствие ответа.

54скачиваний за 30 дней0лайковвеса обновлены11 дек 2020

Бенчмарки

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

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

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

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

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

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

elgeish/cs224n-squad2.0-albert-base-v2 — дообученная модель AlbertForQuestionAnswering на базе ALBERT для извлечения ответов из текста и определения случаев, когда ответ в контексте отсутствует. Её подготовил автор elgeish как базовый результат для студентов курса CS224n, чтобы сократить время на запуск экспериментов с набором SQuAD 2.0.

Модель содержит 136,3 млн параметров, использует PyTorch и принимает вопрос вместе с контекстом на английском языке; длина контекста — до 512 токенов. В README приведены результаты на 6078 примерах отложенной части dev-набора: exact match 78,94 и F1 81,77. Для обучения использовался официальный тренировочный набор SQuAD 2.0, а оценка и выбор модели проводились на случайной половине официального dev-набора.

Модель подходит для экспериментов и учебных проектов по извлечению ответов из текста. README предупреждает, что официальный dev-набор SQuAD 2.0 содержит тестовый набор проекта, поэтому нельзя использовать его или модели, отобранные с его помощью, при работе над этим проектом. Лицензия в предоставленных данных не указана.

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

  • извлечение ответа на вопрос из английского текста
  • определение, содержится ли ответ в данном контексте
  • создание базовой линии для учебного проекта по SQuAD 2.0

Ограничения

Модель ориентирована на английский язык и контексты длиной до 512 токенов. Для проекта CS224n README предписывает не использовать официальный dev-набор SQuAD 2.0 и модели, выбранные с его помощью; лицензия не указана в доступных данных.

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

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

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

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

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

Открыть на Hugging Face

CS224n SQuAD2.0 Project Dataset

The goal of this model is to save CS224n students GPU time when establishing baselines to beat for the Default Final Project. The training set used to fine-tune this model is the same as the official one; however, evaluation and model selection were performed using roughly half of the official dev set, 6078 examples, picked at random. The data files can be found at https://github.com/elgeish/squad/tree/master/data — this is the Winter 2020 version. Given that the official SQuAD2.0 dev set contains the project's test set, students must make sure not to use the official SQuAD2.0 dev set in any way — including the use of models fine-tuned on the official SQuAD2.0, since they used the official SQuAD2.0 dev set for model selection.

Results

{
  "exact": 78.94044093451794,
  "f1": 81.7724930324639,
  "total": 6078,
  "HasAns_exact": 76.28865979381443,
  "HasAns_f1": 82.20385314478195,
  "HasAns_total": 2910,
  "NoAns_exact": 81.37626262626263,
  "NoAns_f1": 81.37626262626263,
  "NoAns_total": 3168,
  "best_exact": 78.95689371503784,
  "best_exact_thresh": 0.0,
  "best_f1": 81.78894581298378,
  "best_f1_thresh": 0.0
}

Notable Arguments

{
  "do_lower_case": true,
  "doc_stride": 128,
  "fp16": false,
  "fp16_opt_level": "O1",
  "gradient_accumulation_steps": 24,
  "learning_rate": 3e-05,
  "max_answer_length": 30,
  "max_grad_norm": 1,
  "max_query_length": 64,
  "max_seq_length": 384,
  "model_name_or_path": "albert-base-v2",
  "model_type": "albert",
  "num_train_epochs": 3,
  "per_gpu_train_batch_size": 8,
  "save_steps": 5000,
  "seed": 42,
  "train_batch_size": 8,
  "version_2_with_negative": true,
  "warmup_steps": 0,
  "weight_decay": 0
}

Environment Setup

{
  "transformers": "2.5.1",
  "pytorch": "1.4.0=py3.6_cuda10.1.243_cudnn7.6.3_0",
  "python": "3.6.5=hc3d631a_2",
  "os": "Linux 4.15.0-1060-aws #62-Ubuntu SMP Tue Feb 11 21:23:22 UTC 2020 x86_64 x86_64 x86_64 GNU/Linux",
  "gpu": "Tesla V100-SXM2-16GB"
}

How to Cite

@misc{elgeish2020gestalt,
  title={Gestalt: a Stacking Ensemble for SQuAD2.0},
  author={Mohamed El-Geish},
  journal={arXiv e-prints},
  archivePrefix={arXiv},
  eprint={2004.07067},
  year={2020},
}

Related Models

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

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

GPU в ailist

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

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

Подбираем GPU…

Класс архитектуры
AlbertForQuestionAnswering
Ревизия весов
95a8cdd

0 моделей