File size: 2,756 Bytes
9400b5b
 
 
 
 
f50ebd0
9400b5b
 
 
f50ebd0
9400b5b
 
f50ebd0
 
 
9400b5b
 
 
f50ebd0
9400b5b
f50ebd0
 
 
 
9400b5b
 
 
f50ebd0
 
 
 
 
 
 
9400b5b
f50ebd0
9400b5b
f50ebd0
 
 
9400b5b
f50ebd0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9400b5b
 
 
f50ebd0
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
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
---
language:
- ru
license: apache-2.0
library_name: sentence-transformers
pipeline_tag: sentence-similarity
base_model: sentence-transformers/all-MiniLM-L6-v2
datasets:
- deepvk/ru-HNP
- ai-forever/rubq-retrieval
tags:
- sentence-transformers
- embeddings
- semantic-search
- triplet-loss
- russian
---

# all-MiniLM-L6-v2 — адаптация на ru-HNP через TripletLoss

Модель получена из `sentence-transformers/all-MiniLM-L6-v2` дообучением на русскоязычном
`deepvk/ru-HNP`. Использовано 20000 исходных строк,
20000 валидных триплетов, `query` как anchor, `pos[0]` как positive
и `neg[0]` как hard negative.

## Обучение

- TripletLoss, cosine distance, margin = 0.2
- batch size = 16
- epochs = 3
- steps per epoch = 1250
- total steps = 3750
- warmup steps = 375 (10.00%)
- seed = 42

## Оценка на ai-forever/rubq-retrieval

Обе модели оценены одной функцией на полном корпусе и всех
1692 запросах с положительными qrels.
Использованы L2-нормированные эмбеддинги и `faiss.IndexFlatIP`.

| Метрика | Базовая модель | Дообученная модель | Абсолютное изменение | Относительное изменение |
|---|---:|---:|---:|---:|
| Recall@5 | 0.027194 | 0.021298 | -0.005896 | -21.68% |
| Recall@10 | 0.035025 | 0.029375 | -0.005650 | -16.13% |

## UMAP

Silhouette рассчитан в отдельной двумерной UMAP-проекции
`ai-forever/headline-classification` (`test[:1000]`, `text`/`label_text`):

- базовая модель: -0.082124
- дообученная модель: -0.086733
- изменение: -0.004609

UMAP — диагностическая 2D-проекция, а не retrieval-метрика. Изменение Silhouette
в UMAP не доказывает улучшение поиска в исходном 384-мерном пространстве.

## Ограничения

`ru-HNP` преимущественно содержит симметричные парафразы и сложные непарафразы,
тогда как RubQ — несимметричная задача «короткий вопрос → документ». Совпадение
языка не гарантирует совпадение предметной области, типа задачи, длины или стиля.
Результат относится к одному seed, одному набору гиперпараметров и первым 20 000
строкам обучающего датасета.