Bge M3 Msmarco V3 Sbert

🚀 基於BAAI/bge-m3的句子轉換器

這是一個基於 sentence-transformers 庫，從 BAAI/bge-m3 微調而來的模型。它可以將句子和段落映射到一個1024維的密集向量空間，可用於語義文本相似度計算、語義搜索、釋義挖掘、文本分類、聚類等任務。

🚀 快速開始

直接使用（Sentence Transformers）

首先，安裝 Sentence Transformers 庫：

pip install -U sentence-transformers

然後，你可以加載該模型並進行推理：

from sentence_transformers import SentenceTransformer

# 從 🤗 Hub 下載模型
model = SentenceTransformer("BlackBeenie/bge-m3-msmarco-v3-sbert")
# 運行推理
sentences = [
    'who is christopher kyle',
    'Chris Kyle American Sniper. Christopher Scott Kyle was born and raised in Texas and was a United States Navy SEAL from 1999 to 2009. He is currently known as the most successful sniper in American military history. According to his book American Sniper, he had 160 confirmed kills (which was from 255 claimed kills).',
    "'American Sniper' Chris Kyle's wife thanks audiences for 'watching the hard stuff'. Taya Kyle has told of her gratitude to audiences for supporting the film about her dead husband Chris Kyle, a Navy Seal played by Bradley Cooper.",
]
embeddings = model.encode(sentences)
print(embeddings.shape)
# [3, 1024]

# 獲取嵌入向量的相似度分數
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [3, 3]

✨ 主要特性

• 語義映射：能夠將句子和段落映射到1024維的密集向量空間，方便進行語義相關任務。
• 多任務支持：可用於語義文本相似度計算、語義搜索、釋義挖掘、文本分類、聚類等多種任務。

📦 安裝指南

安裝 Sentence Transformers 庫：

pip install -U sentence-transformers

💻 使用示例

基礎用法

from sentence_transformers import SentenceTransformer

# 從 🤗 Hub 下載模型
model = SentenceTransformer("BlackBeenie/bge-m3-msmarco-v3-sbert")
# 運行推理
sentences = [
    'who is christopher kyle',
    'Chris Kyle American Sniper. Christopher Scott Kyle was born and raised in Texas and was a United States Navy SEAL from 1999 to 2009. He is currently known as the most successful sniper in American military history. According to his book American Sniper, he had 160 confirmed kills (which was from 255 claimed kills).',
    "'American Sniper' Chris Kyle's wife thanks audiences for 'watching the hard stuff'. Taya Kyle has told of her gratitude to audiences for supporting the film about her dead husband Chris Kyle, a Navy Seal played by Bradley Cooper.",
]
embeddings = model.encode(sentences)
print(embeddings.shape)
# [3, 1024]

# 獲取嵌入向量的相似度分數
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [3, 3]

📚 詳細文檔

模型詳情

模型描述

模型來源

• 文檔：Sentence Transformers 文檔
• 倉庫：GitHub 上的 Sentence Transformers
• Hugging Face：Hugging Face 上的 Sentence Transformers

完整模型架構

SentenceTransformer(
  (0): Transformer({'max_seq_length': 8192, 'do_lower_case': False}) with Transformer model: XLMRobertaModel 
  (1): Pooling({'word_embedding_dimension': 1024, '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})
)

訓練詳情

訓練數據集

未命名數據集

• 大小：498,970 個訓練樣本
• 列：sentence_0、sentence_1 和 sentence_2
• 基於前 1000 個樣本的近似統計信息：

  	sentence_0	sentence_1	sentence_2
  類型	字符串	字符串	字符串
  詳情	最小：4 個標記 平均：9.93 個標記 最大：37 個標記	最小：17 個標記 平均：90.01 個標記 最大：239 個標記	最小：16 個標記 平均：86.47 個標記 最大：229 個標記

• 樣本：

  sentence_0	sentence_1	sentence_2
  how much does it cost to paint a interior house	Interior House Painting Cost Factors. Generally, it will take a minimum of two gallons of paint to cover a room. At the highest end, paint will cost anywhere between $30 and $60 per gallon and come in three different finishes: flat, semi-gloss or high-gloss.Flat finishes are the least shiny and are best suited for areas requiring frequent cleaning.rovide a few details about your project and receive competitive quotes from local pros. The average national cost to paint a home interior is $1,671, with most homeowners spending between $966 and $2,426.	Question DetailsAsked on 3/12/2014. Guest_... How much does it cost per square foot to paint the interior of a house? We just bought roughly a 1500 sg ft townhouse and want to get the entire house, including ceilings painted (including a roughly 400 sq ft finished basement not included in square footage).
  when is s corp taxes due	If you form a corporate entity for your small business, regardless of whether it's taxed as a C or S corporation, a tax return must be filed with the Internal Revenue Service on its due date each year. Corporate tax returns are always due on the 15th day of the third month following the close of the tax year. The actual day that the tax return filing deadline falls on, however, isn't the same for every corporation.	But if you haven’t, don’t panic: the majority of forms aren’t due quite yet. Most tax forms have an annual January 31 due date. Your tax forms are considered on time if the form is properly addressed and mailed on or before that date. If the regular due date falls on a Saturday, Sunday, or legal holiday – which is the case in 2015 for both January and February due dates – issuers have until the next business day.
  what are disaccharides	Disaccharides are formed when two monosaccharides are joined together and a molecule of water is removed, a process known as dehydration reaction. For example; milk sugar (lactose) is made from glucose and galactose whereas the sugar from sugar cane and sugar beets (sucrose) is made from glucose and fructose.altose, another notable disaccharide, is made up of two glucose molecules. The two monosaccharides are bonded via a dehydration reaction (also called a condensation reaction or dehydration synthesis) that leads to the loss of a molecule of water and formation of a glycosidic bond.	Other disaccharides include (diagrams p. 364): Sucrose, common table sugar, has a glycosidic bond linking the anomeric hydroxyls of glucose and fructose. Because the configuration at the anomeric carbon of glucose is a (O points down from the ring), the linkage is designated a(12).

• 損失：MultipleNegativesRankingLoss，參數如下：

  {
      "scale": 20.0,
      "similarity_fct": "cos_sim"
  }
  

訓練超參數

非默認超參數

• eval_strategy：按步驟評估
• per_device_train_batch_size：32
• per_device_eval_batch_size：32
• num_train_epochs：5
• fp16：True
• multi_dataset_batch_sampler：輪詢

所有超參數

訓練日誌

框架版本

• Python：3.11.11
• Sentence Transformers：3.4.1
• Transformers：4.48.3
• PyTorch：2.5.1+cu124
• Accelerate：1.3.0
• Datasets：3.3.2
• Tokenizers：0.21.0

📄 引用

BibTeX

Sentence Transformers

@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",
}

MultipleNegativesRankingLoss

@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}
}