Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks
Paper
•
1908.10084
•
Published
•
10
This is a sentence-transformers model finetuned from intfloat/multilingual-e5-base. It maps sentences & paragraphs to a 768-dimensional dense vector space and can be used for semantic textual similarity, semantic search, paraphrase mining, text classification, clustering, and more.
SentenceTransformer(
(0): Transformer({'max_seq_length': 256, 'do_lower_case': False}) with Transformer model: XLMRobertaModel
(1): Pooling({'word_embedding_dimension': 768, 'pooling_mode_cls_token': False, 'pooling_mode_mean_tokens': True, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False, 'pooling_mode_weightedmean_tokens': False, 'pooling_mode_lasttoken': False, 'include_prompt': True})
(2): Normalize()
)
First install the Sentence Transformers library:
pip install -U sentence-transformers
Then you can load this model and run inference.
from sentence_transformers import SentenceTransformer
# Download from the 🤗 Hub
model = SentenceTransformer("ThanhLe0125/e5-math-ebd")
# Run inference
sentences = [
'query: Trong định nghĩa điểm cực đại, khoảng (a; b) có vai trò gì?',
"passage: # Cực trị của hàm số\n\n## Định nghĩa\nCho hàm số $y = f(x)$ xác định trên tập $D$ và $x_0 \\in D$.\n\n**Cực đại:** $x_0$ được gọi là điểm cực đại nếu tồn tại khoảng $(a; b)$ chứa $x_0$ sao cho $f(x) \\leq f(x_0)$ với mọi $x \\in (a; b) \\cap D$.\n\n**Cực tiểu:** $x_0$ được gọi là điểm cực tiểu nếu tồn tại khoảng $(a; b)$ chứa $x_0$ sao cho $f(x) \\geq f(x_0)$ với mọi $x \\in (a; b) \\cap D$.\n\n## Điều kiện cần (Định lý Fermat)\nNếu hàm số $f(x)$ có đạo hàm tại $x_0$ và $x_0$ là điểm cực trị thì $f'(x_0) = 0$.\n\n## Điều kiện đủ\n**Dấu của đạo hàm bậc nhất:**\n- Nếu $f'(x) > 0$ trên $(x_0 - h; x_0)$ và $f'(x) < 0$ trên $(x_0; x_0 + h)$ thì $x_0$ là điểm cực đại.\n- Nếu $f'(x) < 0$ trên $(x_0 - h; x_0)$ và $f'(x) > 0$ trên $(x_0; x_0 + h)$ thì $x_0$ là điểm cực tiểu.\n\n**Đạo hàm bậc hai:**\n- Nếu $f'(x_0) = 0$ và $f''(x_0) > 0$ thì $x_0$ là điểm cực tiểu.\n- Nếu $f'(x_0) = 0$ và $f''(x_0) < 0$ thì $x_0$ là điểm cực đại.",
'passage: # Các phép toán vectơ trong không gian\n\n## Phép cộng vectơ\n**Quy tắc hình bình hành:** Đặt hai vectơ chung gốc, vectơ tổng là đường chéo hình bình hành.\n\n**Quy tắc tam giác:** $\\overrightarrow{AB} + \\overrightarrow{BC} = \\overrightarrow{AC}$\n\n**Tính chất:**\n- Giao hoán: $\\vec{a} + \\vec{b} = \\vec{b} + \\vec{a}$\n- Kết hợp: $(\\vec{a} + \\vec{b}) + \\vec{c} = \\vec{a} + (\\vec{b} + \\vec{c})$\n- Phần tử trung hòa: $\\vec{a} + \\vec{0} = \\vec{a}$\n\n## Phép trừ vectơ\n$\\vec{a} - \\vec{b} = \\vec{a} + (-\\vec{b})$\n\n**Ý nghĩa:** $\\overrightarrow{AB} - \\overrightarrow{AC} = \\overrightarrow{CB}$\n\n## Phép nhân vectơ với số thực\nCho vectơ $\\vec{a}$ và số thực $k$:\n- Nếu $k > 0$: $k\\vec{a}$ cùng hướng với $\\vec{a}$, độ lớn $|k\\vec{a}| = k|\\vec{a}|$\n- Nếu $k < 0$: $k\\vec{a}$ ngược hướng với $\\vec{a}$, độ lớn $|k\\vec{a}| = |k||\\vec{a}|$ \n- Nếu $k = 0$: $k\\vec{a} = \\vec{0}$\n\n**Tính chất:**\n- $k(\\vec{a} + \\vec{b}) = k\\vec{a} + k\\vec{b}$\n- $(k + h)\\vec{a} = k\\vec{a} + h\\vec{a}$\n- $(kh)\\vec{a} = k(h\\vec{a})$\n\n## Ví dụ ứng dụng\nCho hình hộp $ABCD.EFGH$. Chứng minh: $\\overrightarrow{AG} = \\overrightarrow{AB} + \\overrightarrow{AD} + \\overrightarrow{AE}$\n\n**Lời giải:**\n$\\overrightarrow{AG} = \\overrightarrow{AB} + \\overrightarrow{BG}$\n$= \\overrightarrow{AB} + \\overrightarrow{BC} + \\overrightarrow{CG}$ \n$= \\overrightarrow{AB} + \\overrightarrow{AD} + \\overrightarrow{AE}$',
]
embeddings = model.encode(sentences)
print(embeddings.shape)
# [3, 768]
# Get the similarity scores for the embeddings
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [3, 3]
ranking_evaluatorBinaryClassificationEvaluator| Metric | Value |
|---|---|
| cosine_accuracy | 0.955 |
| cosine_accuracy_threshold | 0.3222 |
| cosine_f1 | 0.9333 |
| cosine_f1_threshold | 0.2707 |
| cosine_precision | 0.9265 |
| cosine_recall | 0.9403 |
| cosine_ap | 0.9869 |
| cosine_mcc | 0.8994 |
sentence_0, sentence_1, and sentence_2| sentence_0 | sentence_1 | sentence_2 | |
|---|---|---|---|
| type | string | string | string |
| details |
|
|
|
| sentence_0 | sentence_1 | sentence_2 |
|---|---|---|
query: Cách tính phương sai khi có các giá trị ngoại lệ trong tập dữ liệu? |
passage: # Phương sai và độ lệch chuẩn |
passage: # Cấp số nhân |
query: Cách tính đạo hàm của một hàm số đa thức như f(x) = x^3 - 3x^2 + 2? |
passage: # Tính đơn điệu của hàm số |
passage: # Hệ trục tọa độ Oxyz trong không gian |
query: Trong bài toán thực tế về tăng lương, yếu tố nào đóng vai trò là công sai? |
passage: # Tổng n số hạng đầu của cấp số cộng |
passage: # Công thức tọa độ cơ bản trong không gian |
MultipleNegativesRankingLoss with these parameters:{
"scale": 20.0,
"similarity_fct": "cos_sim"
}
eval_strategy: stepsper_device_train_batch_size: 4per_device_eval_batch_size: 4num_train_epochs: 20multi_dataset_batch_sampler: round_robinoverwrite_output_dir: Falsedo_predict: Falseeval_strategy: stepsprediction_loss_only: Trueper_device_train_batch_size: 4per_device_eval_batch_size: 4per_gpu_train_batch_size: Noneper_gpu_eval_batch_size: Nonegradient_accumulation_steps: 1eval_accumulation_steps: Nonetorch_empty_cache_steps: Nonelearning_rate: 5e-05weight_decay: 0.0adam_beta1: 0.9adam_beta2: 0.999adam_epsilon: 1e-08max_grad_norm: 1num_train_epochs: 20max_steps: -1lr_scheduler_type: linearlr_scheduler_kwargs: {}warmup_ratio: 0.0warmup_steps: 0log_level: passivelog_level_replica: warninglog_on_each_node: Truelogging_nan_inf_filter: Truesave_safetensors: Truesave_on_each_node: Falsesave_only_model: Falserestore_callback_states_from_checkpoint: Falseno_cuda: Falseuse_cpu: Falseuse_mps_device: Falseseed: 42data_seed: Nonejit_mode_eval: Falseuse_ipex: Falsebf16: Falsefp16: Falsefp16_opt_level: O1half_precision_backend: autobf16_full_eval: Falsefp16_full_eval: Falsetf32: Nonelocal_rank: 0ddp_backend: Nonetpu_num_cores: Nonetpu_metrics_debug: Falsedebug: []dataloader_drop_last: Falsedataloader_num_workers: 0dataloader_prefetch_factor: Nonepast_index: -1disable_tqdm: Falseremove_unused_columns: Truelabel_names: Noneload_best_model_at_end: Falseignore_data_skip: Falsefsdp: []fsdp_min_num_params: 0fsdp_config: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}tp_size: 0fsdp_transformer_layer_cls_to_wrap: Noneaccelerator_config: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}deepspeed: Nonelabel_smoothing_factor: 0.0optim: adamw_torchoptim_args: Noneadafactor: Falsegroup_by_length: Falselength_column_name: lengthddp_find_unused_parameters: Noneddp_bucket_cap_mb: Noneddp_broadcast_buffers: Falsedataloader_pin_memory: Truedataloader_persistent_workers: Falseskip_memory_metrics: Trueuse_legacy_prediction_loop: Falsepush_to_hub: Falseresume_from_checkpoint: Nonehub_model_id: Nonehub_strategy: every_savehub_private_repo: Nonehub_always_push: Falsegradient_checkpointing: Falsegradient_checkpointing_kwargs: Noneinclude_inputs_for_metrics: Falseinclude_for_metrics: []eval_do_concat_batches: Truefp16_backend: autopush_to_hub_model_id: Nonepush_to_hub_organization: Nonemp_parameters: auto_find_batch_size: Falsefull_determinism: Falsetorchdynamo: Noneray_scope: lastddp_timeout: 1800torch_compile: Falsetorch_compile_backend: Nonetorch_compile_mode: Noneinclude_tokens_per_second: Falseinclude_num_input_tokens_seen: Falseneftune_noise_alpha: Noneoptim_target_modules: Nonebatch_eval_metrics: Falseeval_on_start: Falseuse_liger_kernel: Falseeval_use_gather_object: Falseaverage_tokens_across_devices: Falseprompts: Nonebatch_sampler: batch_samplermulti_dataset_batch_sampler: round_robin| Epoch | Step | Training Loss | ranking_evaluator_cosine_ap |
|---|---|---|---|
| 0.8299 | 200 | - | 0.9543 |
| 1.0 | 241 | - | 0.9699 |
| 1.6598 | 400 | - | 0.9702 |
| 2.0 | 482 | - | 0.9744 |
| 2.0747 | 500 | 0.7119 | - |
| 2.4896 | 600 | - | 0.9765 |
| 3.0 | 723 | - | 0.9852 |
| 3.3195 | 800 | - | 0.9822 |
| 4.0 | 964 | - | 0.9829 |
| 4.1494 | 1000 | 0.4462 | 0.9813 |
| 4.9793 | 1200 | - | 0.9775 |
| 5.0 | 1205 | - | 0.9768 |
| 5.8091 | 1400 | - | 0.9843 |
| 6.0 | 1446 | - | 0.9869 |
@inproceedings{reimers-2019-sentence-bert,
title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
author = "Reimers, Nils and Gurevych, Iryna",
booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
month = "11",
year = "2019",
publisher = "Association for Computational Linguistics",
url = "https://arxiv.org/abs/1908.10084",
}
@misc{henderson2017efficient,
title={Efficient Natural Language Response Suggestion for Smart Reply},
author={Matthew Henderson and Rami Al-Rfou and Brian Strope and Yun-hsuan Sung and Laszlo Lukacs and Ruiqi Guo and Sanjiv Kumar and Balint Miklos and Ray Kurzweil},
year={2017},
eprint={1705.00652},
archivePrefix={arXiv},
primaryClass={cs.CL}
}
Base model
intfloat/multilingual-e5-base