/ Модель

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

Контекст
512 токенов
Параметры
137 млн
Архитектура
bert · dense
Опубликована
19 окт 2022

drug_combinations_lm_pubmedbert

Заполнение пропусковТекст → текст3 профессии

Англоязычная BERT-модель, дополнительно обученная на аннотациях PubMed о комбинированной лекарственной терапии для предсказания замаскированных слов и представления биомедицинского текста.

157скачиваний за 30 дней3лайкавеса обновлены20 окт 2022

Бенчмарки

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

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

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

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

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

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

allenai/drug_combinations_lm_pubmedbert — англоязычная модель семейства BERT от Allen Institute for AI, основанная на PubMedBERT. Её дополнительно предварительно обучили на аннотациях PubMed, вероятно посвящённых терапиям с несколькими препаратами, чтобы адаптировать представления модели к этой области. Модель содержит 136,7 млн параметров, использует архитектуру BertForMaskedLM и принимает последовательности длиной до 512 токенов.

Основное назначение — предсказывать замаскированные слова в биомедицинских текстах. В исследовании извлечения отношений между комбинациями препаратов она показала лучшие результаты среди контекстных энкодеров, выступая компонентом более крупной модели классификации отношений; сама по себе она не является готовой системой извлечения отношений. Для использования модели авторы просят цитировать работы PubMedBERT и датасета по извлечению n-арных отношений комбинаций препаратов. Лицензия в предоставленных данных не указана.

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

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

Ограничения

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

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

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

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

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

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

Открыть на Hugging Face

This is a version of PubmedBERT which has been domain-adapted (via additional pretraining) to a set of PubMed abstracts that likely discuss multiple-drug therapies. This model was the strongest contextualized encoder in the experiments in the paper "A Dataset for N-ary Relation Extraction of Drug Combinations", when used as a component of a larger relation classification model (also hosted here on Huggingface).

If you use this model, cite both

@misc{pubmedbert,
  author = {Yu Gu and Robert Tinn and Hao Cheng and Michael Lucas and Naoto Usuyama and Xiaodong Liu and Tristan Naumann and Jianfeng Gao and Hoifung Poon},
  title = {Domain-Specific Language Model Pretraining for Biomedical Natural Language Processing},
  year = {2020},
  eprint = {arXiv:2007.15779},
}

and

@inproceedings{Tiktinsky2022ADF,
    title = "A Dataset for N-ary Relation Extraction of Drug Combinations",
    author = "Tiktinsky, Aryeh and Viswanathan, Vijay and Niezni, Danna and Meron Azagury, Dana and Shamay, Yosi and Taub-Tabib, Hillel and Hope, Tom and Goldberg, Yoav",
    booktitle = "Proceedings of the 2022 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies",
    month = jul,
    year = "2022",
    address = "Seattle, United States",
    publisher = "Association for Computational Linguistics",
    url = "https://aclanthology.org/2022.naacl-main.233",
    doi = "10.18653/v1/2022.naacl-main.233",
    pages = "3190--3203",
}

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

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

GPU в ailist

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

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

Подбираем GPU…

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

0 моделей