GLiDRE Multi v2

GLiDRE v2 — мультиязычный чекпоинт от CEA LIST-IA для извлечения связей на уровне документов. Это облегчённая модель для zero-shot классификации отношений: ей можно передать текст или полный документ и список меток. Построена на mdeberta-v3-base и gte-multilingual-base.

🧐 Ключевые особенности

• 🫡 Zero-shot: модель классифицирует невиданные связи прямо по тексту.
• 🔥 Вход: поддерживаются токенизированный текст и полные документы.
• 🧐 Архитектура: несколько функций потерь и возможность менять компоненты.
• 🌞 Мультиязычность: 12 языков, включая английский, французский, немецкий, испанский и арабский.

🚀 Пример

from glidre import GLiDRE
model = GLiDRE.from_pretrained('cea-list-ia/glidre_multi_v2')
text = 'The Loud Tour was the fourth overall and third world concert tour by Barbadian recording artist Rihanna.'
labels = ['COUNTRY_OF_CITIZENSHIP', 'PUBLICATION_DATE', 'PART_OF']
mentions = [{'id': 0, 'type': 'LOC', 'mentions': [{'value': 'Barbadian', 'start': 69, 'end': 78}]}, {'id': 1, 'type': 'PER', 'mentions': [{'value': 'Rihanna', 'start': 96, 'end': 103}]}]
relations = model.predict_entities(text=text, labels=labels, mentions=mentions, threshold=0.3, multi_label=False)
print(relations)

Подробнее

Основа: microsoft/mdeberta-v3-base и Alibaba-NLP/gte-multilingual-base.
Дообучение: поддержка DocRED и Re-DocRED, несколько loss-функций.
Ограничения: нужны упоминания сущностей с индексами и список меток; порог задаётся вручную. Цифр бенчмарков и требований к VRAM в карточке нет.