Update README.md
Browse files
README.md
CHANGED
|
@@ -1,199 +1,270 @@
|
|
| 1 |
---
|
| 2 |
-
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
---
|
| 5 |
|
| 6 |
-
#
|
| 7 |
|
| 8 |
-
|
| 9 |
|
|
|
|
| 10 |
|
|
|
|
| 11 |
|
| 12 |
-
|
| 13 |
|
| 14 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 15 |
|
| 16 |
-
|
| 17 |
|
| 18 |
-
|
| 19 |
|
| 20 |
-
|
| 21 |
-
- **Funded by [optional]:** [More Information Needed]
|
| 22 |
-
- **Shared by [optional]:** [More Information Needed]
|
| 23 |
-
- **Model type:** [More Information Needed]
|
| 24 |
-
- **Language(s) (NLP):** [More Information Needed]
|
| 25 |
-
- **License:** [More Information Needed]
|
| 26 |
-
- **Finetuned from model [optional]:** [More Information Needed]
|
| 27 |
|
| 28 |
-
|
| 29 |
|
| 30 |
-
|
|
|
|
| 31 |
|
| 32 |
-
|
| 33 |
-
- **Paper [optional]:** [More Information Needed]
|
| 34 |
-
- **Demo [optional]:** [More Information Needed]
|
| 35 |
|
| 36 |
-
|
| 37 |
|
| 38 |
-
|
| 39 |
|
| 40 |
-
|
| 41 |
|
| 42 |
-
|
| 43 |
|
| 44 |
-
|
| 45 |
|
| 46 |
-
|
| 47 |
|
| 48 |
-
|
| 49 |
|
| 50 |
-
|
| 51 |
|
| 52 |
-
|
| 53 |
|
| 54 |
-
|
| 55 |
|
| 56 |
-
|
|
|
|
|
|
|
|
|
|
| 57 |
|
| 58 |
-
|
| 59 |
|
| 60 |
-
|
| 61 |
|
| 62 |
-
|
| 63 |
|
| 64 |
-
|
|
|
|
|
|
|
| 65 |
|
| 66 |
-
|
| 67 |
|
| 68 |
-
|
|
|
|
| 69 |
|
| 70 |
-
|
|
|
|
| 71 |
|
| 72 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 73 |
|
| 74 |
-
|
|
|
|
| 75 |
|
| 76 |
-
|
|
|
|
|
|
|
| 77 |
|
| 78 |
-
###
|
| 79 |
|
| 80 |
-
|
|
|
|
|
|
|
| 81 |
|
| 82 |
-
|
| 83 |
|
| 84 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 85 |
|
| 86 |
-
|
|
|
|
| 87 |
|
| 88 |
-
|
|
|
|
|
|
|
| 89 |
|
| 90 |
-
|
|
|
|
| 91 |
|
|
|
|
|
|
|
| 92 |
|
| 93 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 94 |
|
| 95 |
-
|
| 96 |
|
| 97 |
-
|
|
|
|
|
|
|
|
|
|
| 98 |
|
| 99 |
-
|
| 100 |
|
| 101 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 102 |
|
| 103 |
-
|
|
|
|
| 104 |
|
| 105 |
-
|
|
|
|
|
|
|
| 106 |
|
| 107 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 108 |
|
| 109 |
-
|
| 110 |
|
| 111 |
-
|
| 112 |
|
| 113 |
-
|
|
|
|
| 114 |
|
| 115 |
-
|
|
|
|
| 116 |
|
| 117 |
-
|
|
|
|
| 118 |
|
| 119 |
-
|
|
|
|
| 120 |
|
| 121 |
-
|
|
|
|
| 122 |
|
| 123 |
-
|
| 124 |
|
| 125 |
-
|
| 126 |
|
| 127 |
-
|
|
|
|
|
|
|
| 128 |
|
| 129 |
-
|
| 130 |
|
| 131 |
-
|
| 132 |
|
|
|
|
| 133 |
|
|
|
|
| 134 |
|
| 135 |
-
|
| 136 |
|
| 137 |
-
|
| 138 |
|
| 139 |
-
|
| 140 |
|
| 141 |
-
|
| 142 |
|
| 143 |
-
|
| 144 |
|
| 145 |
-
|
|
|
|
|
|
|
| 146 |
|
| 147 |
-
|
| 148 |
-
- **Hours used:** [More Information Needed]
|
| 149 |
-
- **Cloud Provider:** [More Information Needed]
|
| 150 |
-
- **Compute Region:** [More Information Needed]
|
| 151 |
-
- **Carbon Emitted:** [More Information Needed]
|
| 152 |
|
| 153 |
-
|
|
|
|
|
|
|
| 154 |
|
| 155 |
-
|
| 156 |
|
| 157 |
-
|
| 158 |
|
| 159 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 160 |
|
| 161 |
-
|
| 162 |
|
| 163 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 164 |
|
| 165 |
-
|
| 166 |
|
| 167 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 168 |
|
| 169 |
-
|
| 170 |
|
| 171 |
-
|
| 172 |
-
|
| 173 |
-
<!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
|
| 174 |
-
|
| 175 |
-
**BibTeX:**
|
| 176 |
-
|
| 177 |
-
[More Information Needed]
|
| 178 |
-
|
| 179 |
-
**APA:**
|
| 180 |
-
|
| 181 |
-
[More Information Needed]
|
| 182 |
-
|
| 183 |
-
## Glossary [optional]
|
| 184 |
-
|
| 185 |
-
<!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
|
| 186 |
-
|
| 187 |
-
[More Information Needed]
|
| 188 |
-
|
| 189 |
-
## More Information [optional]
|
| 190 |
-
|
| 191 |
-
[More Information Needed]
|
| 192 |
-
|
| 193 |
-
## Model Card Authors [optional]
|
| 194 |
-
|
| 195 |
-
[More Information Needed]
|
| 196 |
-
|
| 197 |
-
## Model Card Contact
|
| 198 |
-
|
| 199 |
-
[More Information Needed]
|
|
|
|
| 1 |
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
language:
|
| 4 |
+
- ru
|
| 5 |
+
base_model:
|
| 6 |
+
- FractalGPT/SbertDistil
|
| 7 |
+
- intfloat/multilingual-e5-large-instruct
|
| 8 |
+
library_name: sentence-transformers
|
| 9 |
+
tags:
|
| 10 |
+
- sentence-transformers
|
| 11 |
+
- sentence-similarity
|
| 12 |
+
- feature-extraction
|
| 13 |
+
- embeddings
|
| 14 |
+
- distillation
|
| 15 |
+
- nli
|
| 16 |
+
- masl
|
| 17 |
+
- task-specification
|
| 18 |
+
- agent
|
| 19 |
+
pipeline_tag: sentence-similarity
|
| 20 |
---
|
| 21 |
|
| 22 |
+
# SbertDistilV2
|
| 23 |
|
| 24 |
+
**Автор:** Потанин М. В.
|
| 25 |
|
| 26 |
+
## Описание модели
|
| 27 |
|
| 28 |
+
SbertDistilV2 — компактная модель эмбеддингов, специализированная для работы с задачами NLI (Natural Language Interface) и преобразования команд в формат MASL (Multi-agent system language). Модель получена путём двухэтапного обучения с применением дистилляции знаний и дообучения на специализированном датасете.
|
| 29 |
|
| 30 |
+
### Ключевые характеристики
|
| 31 |
|
| 32 |
+
- **Базовая модель**: `FractalGPT/SbertDistil`
|
| 33 |
+
- **Модель-учитель**: `intfloat/multilingual-e5-large-instruct`
|
| 34 |
+
- **Размерность эмбеддингов**: 384 (против 1024 у учителя)
|
| 35 |
+
- **Язык**: Русский
|
| 36 |
+
- **Специализация**: NLI задачи, MASL спецификации
|
| 37 |
|
| 38 |
+
## Архитектура обучения
|
| 39 |
|
| 40 |
+
Обучение проводилось в **два этапа**:
|
| 41 |
|
| 42 |
+
### Этап 1: Дистилляция знаний
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 43 |
|
| 44 |
+
Передача знаний от большой модели-учителя (`multilingual-e5-large-instruct`, 1024 dim) к компактной модели-ученику (`SbertDistil`, 384 dim).
|
| 45 |
|
| 46 |
+
**Метод согласования размерностей:**
|
| 47 |
+
- SVD-проекция для сжатия 1024-мерных эмбеддингов учителя в 384-мерное пространство ученика с минимальными потерями информации
|
| 48 |
|
| 49 |
+
**Комбинированная функция потерь:**
|
|
|
|
|
|
|
| 50 |
|
| 51 |
+
$$\mathcal{L}_{\text{total}} = \alpha \cdot \mathcal{L}_{\text{MSE}} + \beta \cdot \mathcal{L}_{\text{cosine}} + \gamma \cdot \mathcal{L}_{\text{InfoNCE}}$$
|
| 52 |
|
| 53 |
+
Где:
|
| 54 |
|
| 55 |
+
$$\mathcal{L}_{\text{MSE}}$$ — среднеквадратичная ошибка для прямого согласования эмбеддингов (вес: **1.0**)
|
| 56 |
|
| 57 |
+
$$\mathcal{L}_{\text{cosine}}$$ — косинусное сходство для сохранения направлений векторов (вес: **0.5**)
|
| 58 |
|
| 59 |
+
$$\mathcal{L}_{\text{InfoNCE}}$$ — контрастное обучение для улучшения различимости представлений (вес: **0.3**)
|
| 60 |
|
| 61 |
+
**Датасет:** Запросы на естественном языке
|
| 62 |
|
| 63 |
+
### Этап 2: Дообучение на MASL
|
| 64 |
|
| 65 |
+
Специализация модели на задачах преобразования NL-команд в MASL спецификации.
|
| 66 |
|
| 67 |
+
**Датасет:** Пары "запрос на естественном языке / MASL структура" (9 999 примеров)
|
| 68 |
|
| 69 |
+
## Датасет
|
| 70 |
|
| 71 |
+
Модель дообучена на синтетическом датасете NLI Task Specification Dataset (MASL):
|
| 72 |
+
- **Размер**: 9 999 примеров
|
| 73 |
+
- **Структура**: пары "команда → JSON структура"
|
| 74 |
+
- **Формат**: Входные и вых��дные коннекторы с типами данных, семантическими категориями и областями знаний
|
| 75 |
|
| 76 |
+
Подробнее о датасете см. документацию к [NLI Dataset](https://huggingface.co/datasets/Ponimash/nli_dataset).
|
| 77 |
|
| 78 |
+
## Использование
|
| 79 |
|
| 80 |
+
### Установка
|
| 81 |
|
| 82 |
+
```bash
|
| 83 |
+
pip install sentence-transformers
|
| 84 |
+
```
|
| 85 |
|
| 86 |
+
### Базовое использование
|
| 87 |
|
| 88 |
+
```python
|
| 89 |
+
from sentence_transformers import SentenceTransformer
|
| 90 |
|
| 91 |
+
# Загрузка модели
|
| 92 |
+
model = SentenceTransformer("FractalGPT/SbertDistilV2")
|
| 93 |
|
| 94 |
+
# Пример команд
|
| 95 |
+
commands = [
|
| 96 |
+
"Преобразуй эту аудиозапись в текстовый формат",
|
| 97 |
+
"Создай презентацию с анализом медицинских данных",
|
| 98 |
+
"Ответь на вопрос о квантовой физике голосом"
|
| 99 |
+
]
|
| 100 |
|
| 101 |
+
# Получение эмбеддингов
|
| 102 |
+
embeddings = model.encode(commands)
|
| 103 |
|
| 104 |
+
print(f"Размерность эмбеддингов: {embeddings.shape[1]}")
|
| 105 |
+
print(f"Количество векторов: {embeddings.shape[0]}")
|
| 106 |
+
```
|
| 107 |
|
| 108 |
+
### Поиск похожих команд
|
| 109 |
|
| 110 |
+
```python
|
| 111 |
+
from sentence_transformers import SentenceTransformer, util
|
| 112 |
+
import torch
|
| 113 |
|
| 114 |
+
model = SentenceTransformer("FractalGPT/SbertDistilV2")
|
| 115 |
|
| 116 |
+
# База команд
|
| 117 |
+
commands_db = [
|
| 118 |
+
"Преобразуй речь в текст",
|
| 119 |
+
"Создай презентацию по физике",
|
| 120 |
+
"Сделай аудио ответ на вопрос",
|
| 121 |
+
"Проанализируй медицинский документ",
|
| 122 |
+
"Переведи аудио в текстовый файл"
|
| 123 |
+
]
|
| 124 |
|
| 125 |
+
# Запрос пользователя
|
| 126 |
+
query = "Транскрибируй голосовое сообщение в txt"
|
| 127 |
|
| 128 |
+
# Получение эмбеддингов
|
| 129 |
+
query_embedding = model.encode(query, convert_to_tensor=True)
|
| 130 |
+
db_embeddings = model.encode(commands_db, convert_to_tensor=True)
|
| 131 |
|
| 132 |
+
# Вычисление косинусного сходства
|
| 133 |
+
cosine_scores = util.cos_sim(query_embedding, db_embeddings)[0]
|
| 134 |
|
| 135 |
+
# Сортировка результатов
|
| 136 |
+
results = torch.argsort(cosine_scores, descending=True)
|
| 137 |
|
| 138 |
+
print(f"Запрос: {query}\n")
|
| 139 |
+
print("Наиболее похожие команды:")
|
| 140 |
+
for idx in results[:3]:
|
| 141 |
+
print(f" {commands_db[idx]} (Score: {cosine_scores[idx]:.4f})")
|
| 142 |
+
```
|
| 143 |
|
| 144 |
+
### Кластеризация команд
|
| 145 |
|
| 146 |
+
```python
|
| 147 |
+
from sentence_transformers import SentenceTransformer
|
| 148 |
+
from sklearn.cluster import KMeans
|
| 149 |
+
import numpy as np
|
| 150 |
|
| 151 |
+
model = SentenceTransformer("FractalGPT/SbertDistilV2")
|
| 152 |
|
| 153 |
+
commands = [
|
| 154 |
+
"Преобразуй аудио в текст",
|
| 155 |
+
"Транскрибируй голос в txt",
|
| 156 |
+
"Создай презентацию по биологии",
|
| 157 |
+
"Сделай pptx с анализом данных",
|
| 158 |
+
"Ответь голосом на вопрос",
|
| 159 |
+
"Озвучь ответ на запрос"
|
| 160 |
+
]
|
| 161 |
|
| 162 |
+
# Получение эмбеддингов
|
| 163 |
+
embeddings = model.encode(commands)
|
| 164 |
|
| 165 |
+
# Кластеризация (3 кластера: транскрипция, презентации, голосовые ответы)
|
| 166 |
+
kmeans = KMeans(n_clusters=3, random_state=42)
|
| 167 |
+
clusters = kmeans.fit_predict(embeddings)
|
| 168 |
|
| 169 |
+
# Группировка по кластерам
|
| 170 |
+
for i in range(3):
|
| 171 |
+
cluster_commands = [cmd for cmd, cluster in zip(commands, clusters) if cluster == i]
|
| 172 |
+
print(f"\nКластер {i+1}:")
|
| 173 |
+
for cmd in cluster_commands:
|
| 174 |
+
print(f" - {cmd}")
|
| 175 |
+
```
|
| 176 |
|
| 177 |
+
## Применение
|
| 178 |
|
| 179 |
+
Модель оптимизирована для следующих задач:
|
| 180 |
|
| 181 |
+
### 1. Семантический поиск команд
|
| 182 |
+
Поиск похожих команд в базе знаний мультиагентной системы для переиспользования существующих решений.
|
| 183 |
|
| 184 |
+
### 2. Кластеризация задач
|
| 185 |
+
Группировка схожих запросов для оптимизации маршрутизации в агентных системах.
|
| 186 |
|
| 187 |
+
### 3. Классификация намерений
|
| 188 |
+
Определение типа задачи (транскрипция, генерация отчётов, голосовые ответы и т.д.) по векторному представлению.
|
| 189 |
|
| 190 |
+
### 4. Ранжирование инструментов
|
| 191 |
+
Поиск наиболее подходящих инструментов для выполнения пользовательской команды на основе семантического сходства.
|
| 192 |
|
| 193 |
+
### 5. Дедупликация запросов
|
| 194 |
+
Выявление дублирующихся или очень похожих запросов для оптимизации обработки.
|
| 195 |
|
| 196 |
+
## Технические детали
|
| 197 |
|
| 198 |
+
### Оптимизации обучения
|
| 199 |
|
| 200 |
+
- **Mixed Precision Training**: использование bfloat16/float16 для ускорения обучения
|
| 201 |
+
- **Gradient Accumulation**: накопление градиентов для эффективной работы с большими батчами
|
| 202 |
+
- **SVD Projection**: оптимальное сжатие размерности с сохранением максимальной дисперсии
|
| 203 |
|
| 204 |
+
### Математическое обоснование
|
| 205 |
|
| 206 |
+
**MSE Loss** (прямое согласование):
|
| 207 |
|
| 208 |
+
$$\mathcal{L}_{\text{MSE}} = \frac{1}{N} \sum_{i=1}^{N} \|\text{SVD}(\mathbf{h}_{\text{teacher}}^{(i)}) - \mathbf{h}_{\text{student}}^{(i)}\|^2$$
|
| 209 |
|
| 210 |
+
**Cosine Similarity Loss** (сохранение направлений):
|
| 211 |
|
| 212 |
+
$$\mathcal{L}_{\text{cosine}} = 1 - \frac{1}{N} \sum_{i=1}^{N} \frac{\mathbf{h}_{\text{teacher}}^{(i)} \cdot \mathbf{h}_{\text{student}}^{(i)}}{\|\mathbf{h}_{\text{teacher}}^{(i)}\| \|\mathbf{h}_{\text{student}}^{(i)}\|}$$
|
| 213 |
|
| 214 |
+
**InfoNCE Loss** (контрастное обучение):
|
| 215 |
|
| 216 |
+
$$\mathcal{L}_{\text{InfoNCE}} = -\log \frac{\exp(\text{sim}(\mathbf{h}_i, \mathbf{h}_i^+) / \tau)}{\sum_{j=1}^{N} \exp(\text{sim}(\mathbf{h}_i, \mathbf{h}_j) / \tau)}$$
|
| 217 |
|
| 218 |
+
Где $\tau$ — температурный параметр контрастного обучения.
|
| 219 |
|
| 220 |
+
## Ограничения
|
| 221 |
|
| 222 |
+
- Модель оптимизирована для **русского языка**
|
| 223 |
+
- Специализирована на задачах NLI и MASL, может показывать сниженное качество на общих задачах эмбеддингов
|
| 224 |
+
- Размерность 384 может быть недостаточной для очень сложных семантических различий
|
| 225 |
|
| 226 |
+
## Связанные модели
|
|
|
|
|
|
|
|
|
|
|
|
|
| 227 |
|
| 228 |
+
- [Qwen2.5-NLI-7B](https://huggingface.co/Ponimash/Qwen2.5-nli-7b) — генеративная модель для создания MASL спецификаций
|
| 229 |
+
- [FractalGPT/SbertDistil](https://huggingface.co/FractalGPT/SbertDistil) — базовая модель
|
| 230 |
+
- [multilingual-e5-large-instruct](https://huggingface.co/intfloat/multilingual-e5-large-instruct) — модель-учитель
|
| 231 |
|
| 232 |
+
## Примеры результатов
|
| 233 |
|
| 234 |
+
### Пример 1: Транскрипция
|
| 235 |
|
| 236 |
+
```python
|
| 237 |
+
query = "Преобразуй голосовое сообщение в текст"
|
| 238 |
+
similar = [
|
| 239 |
+
"Транскрибируй аудио в txt",
|
| 240 |
+
"Переведи речь в текстовый формат",
|
| 241 |
+
"Распознай речь и сохрани в файл"
|
| 242 |
+
]
|
| 243 |
+
```
|
| 244 |
|
| 245 |
+
### Пример 2: Генерация отчётов
|
| 246 |
|
| 247 |
+
```python
|
| 248 |
+
query = "Создай презентацию с анализом данных"
|
| 249 |
+
similar = [
|
| 250 |
+
"Сделай pptx с результатами исследования",
|
| 251 |
+
"Подготовь слайды с выводами",
|
| 252 |
+
"Оформи отчёт в виде презентации"
|
| 253 |
+
]
|
| 254 |
+
```
|
| 255 |
|
| 256 |
+
## Цитирование
|
| 257 |
|
| 258 |
+
```bibtex
|
| 259 |
+
@misc{sbertdistilv2-nli-masl,
|
| 260 |
+
author = {Потанин, М. В.},
|
| 261 |
+
title = {SbertDistilV2-NLI-MASL: Compact Embeddings for Natural Language Interface Tasks},
|
| 262 |
+
year = {2024},
|
| 263 |
+
publisher = {HuggingFace},
|
| 264 |
+
howpublished = {\url{https://huggingface.co/FractalGPT/SbertDistilV2}}
|
| 265 |
+
}
|
| 266 |
+
```
|
| 267 |
|
| 268 |
+
## Лицензия
|
| 269 |
|
| 270 |
+
Модель распространяется под лицензией Apache 2.0.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|