Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks
Paper • 1908.10084 • Published • 15
How to use ThanhLe0125/e5-math-ebd with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("ThanhLe0125/e5-math-ebd")
sentences = [
"query: Cách tính chi phí biến đổi trong một bài toán cụ thể.",
"passage: # Bài toán tối ưu hình học\n\n## Dạng bài thường gặp\n1. **Chu vi cố định, tối ưu diện tích**\n2. **Diện tích cố định, tối ưu chu vi** \n3. **Vật liệu cố định, tối ưu thể tích**\n4. **Tối ưu kích thước hộp, hình trụ**\n\n## Ví dụ 1: Hình chữ nhật có chu vi cố định\nTrong tất cả hình chữ nhật có chu vi $2p$, tìm hình có diện tích lớn nhất.\n\n**Lời giải:**\nGọi hai cạnh là $x$ và $y$ với $x, y > 0$\nĐiều kiện: $2(x + y) = 2p \\Rightarrow y = p - x$\nMiền xác định: $0 < x < p$\n\nDiện tích: $S(x) = xy = x(p - x) = px - x^2$\n$S'(x) = p - 2x = 0 \\Rightarrow x = \\frac{p}{2}$\n\nKhi đó $y = p - \\frac{p}{2} = \\frac{p}{2}$\n\n**Kết luận:** Hình vuông có diện tích lớn nhất.\n\n## Ví dụ 2: Hộp không nắp từ tấm tôn\nTừ tấm tôn vuông cạnh $a$, cắt bỏ bốn góc là hình vuông cạnh $x$, gập thành hộp không nắp. Tìm $x$ để thể tích lớn nhất.\n\n**Lời giải:**\n- Đáy hộp: $(a-2x) \\times (a-2x)$\n- Chiều cao: $x$\n- Điều kiện: $0 < x < \\frac{a}{2}$\n\nThể tích: $V(x) = x(a-2x)^2$\n$V'(x) = (a-2x)^2 + x \\cdot 2(a-2x)(-2) = (a-2x)(a-6x)$\n$V'(x) = 0 \\Rightarrow x = \\frac{a}{6}$ (loại $x = \\frac{a}{2}$)\n\n**Kết luận:** $x = \\frac{a}{6}$ cho thể tích lớn nhất.",
"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}$",
"passage: # Bài toán tối ưu kinh tế\n\n## Các khái niệm cơ bản\n- **Doanh thu:** $R(x) = p \\cdot x$ (giá × số lượng)\n- **Chi phí:** $C(x) = $ chi phí cố định + chi phí biến đổi\n- **Lợi nhuận:** $P(x) = R(x) - C(x)$\n\n## Ví dụ: Tối ưu lợi nhuận\nMột công ty sản xuất $x$ sản phẩm/ngày với:\n- Chi phí: $C(x) = x^2 + 40x + 400$ (nghìn đồng)\n- Giá bán: 200 nghìn đồng/sản phẩm\n\nTìm mức sản xuất để lợi nhuận tối đa.\n\n**Lời giải:**\nDoanh thu: $R(x) = 200x$\nLợi nhuận: $P(x) = 200x - (x^2 + 40x + 400) = -x^2 + 160x - 400$\n\n$P'(x) = -2x + 160 = 0 \\Rightarrow x = 80$\n$P''(x) = -2 < 0$ nên $x = 80$ cho cực đại.\n\nLợi nhuận tối đa: $P(80) = -6400 + 12800 - 400 = 6000$ nghìn đồng.\n\n**Kết luận:** Sản xuất 80 sản phẩm/ngày để đạt lợi nhuận tối đa 6 triệu đồng.\n\n## Dạng bài khác\n- **Tối ưu giá bán** khi có mối quan hệ giữa giá và lượng tiêu thụ\n- **Tối ưu chi phí vận chuyển, sản xuất**\n- **Tối ưu thời gian với chi phí nhân công**"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4]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