Finetuned Ce Climate Multineg V1

🚀 基於 cross-encoder/ms-marco-MiniLM-L12-v2 的交叉編碼器

這是一個基於 Cross Encoder 的模型，它在 climate-cross-encoder-mixed-neg-v3 數據集上，使用 sentence-transformers 庫對 cross-encoder/ms-marco-MiniLM-L12-v2 進行微調得到。該模型可以為文本對計算分數，可用於文本重排序和語義搜索。

🚀 快速開始

本模型可用於計算文本對的分數，進而實現文本重排序和語義搜索。下面將詳細介紹如何使用該模型。

✨ 主要特性

• 基於 cross-encoder/ms-marco-MiniLM-L12-v2 進行微調，在 climate-cross-encoder-mixed-neg-v3 數據集上訓練。
• 能夠計算文本對的分數，用於文本重排序和語義搜索。
• 支持使用 sentence-transformers 庫進行推理和微調。

📦 安裝指南

首先，你需要安裝 sentence-transformers 庫：

pip install -U sentence-transformers

💻 使用示例

基礎用法

以下是如何加載模型並進行推理的示例：

from sentence_transformers import CrossEncoder

# 從 Hugging Face Hub 下載模型
model = CrossEncoder("CharlesPing/finetuned-ce-climate-multineg-v1")
# 獲取文本對的分數
pairs = [
    ['Scientific analysis of past climates\xa0shows that greenhouse gasses, principally CO2,\xa0have controlled most ancient\xa0climate changes.', 'Greenhouse gases, in particular carbon dioxide and methane, played a significant role during the Eocene in controlling the surface temperature.'],
    ['Scientific analysis of past climates\xa0shows that greenhouse gasses, principally CO2,\xa0have controlled most ancient\xa0climate changes.', 'Climatic geomorphology is of limited use to study recent (Quaternary, Holocene) large climate changes since there are seldom discernible in the geomorphological record.'],
    ['Scientific analysis of past climates\xa0shows that greenhouse gasses, principally CO2,\xa0have controlled most ancient\xa0climate changes.', 'There is also a close correlation between CO2 and temperature, where CO2 has a strong control over global temperatures in Earth history.'],
    ['Scientific analysis of past climates\xa0shows that greenhouse gasses, principally CO2,\xa0have controlled most ancient\xa0climate changes.', 'While scientists knew of past climate change such as the ice ages, the concept of climate as unchanging was useful in the development of a general theory of what determines climate.'],
    ['Scientific analysis of past climates\xa0shows that greenhouse gasses, principally CO2,\xa0have controlled most ancient\xa0climate changes.', 'Some long term modifications along the history of the planet have been significant, such as the incorporation of oxygen to the atmosphere.'],
]
scores = model.predict(pairs)
print(scores.shape)
# (5,)

# 或者根據與單個文本的相似度對不同文本進行排序
ranks = model.rank(
    'Scientific analysis of past climates\xa0shows that greenhouse gasses, principally CO2,\xa0have controlled most ancient\xa0climate changes.',
    [
        'Greenhouse gases, in particular carbon dioxide and methane, played a significant role during the Eocene in controlling the surface temperature.',
        'Climatic geomorphology is of limited use to study recent (Quaternary, Holocene) large climate changes since there are seldom discernible in the geomorphological record.',
        'There is also a close correlation between CO2 and temperature, where CO2 has a strong control over global temperatures in Earth history.',
        'While scientists knew of past climate change such as the ice ages, the concept of climate as unchanging was useful in the development of a general theory of what determines climate.',
        'Some long term modifications along the history of the planet have been significant, such as the incorporation of oxygen to the atmosphere.',
    ]
)
# [{'corpus_id': ..., 'score': ...}, {'corpus_id': ..., 'score': ...}, ...]

📚 詳細文檔

模型詳情

模型描述

模型資源

• 文檔：Sentence Transformers 文檔
• 文檔：Cross Encoder 文檔
• 代碼倉庫：GitHub 上的 Sentence Transformers
• Hugging Face：Hugging Face 上的交叉編碼器

評估

指標

交叉編碼器重排序

• 數據集：climate-rerank-multineg
• 使用 CrossEncoderRerankingEvaluator 進行評估，參數如下：

  {
      "at_k": 1,
      "always_rerank_positives": false
  }
  

訓練詳情

訓練數據集

climate-cross-encoder-mixed-neg-v3

• 數據集：climate-cross-encoder-mixed-neg-v3，版本為 cd49b57
• 大小：41,052 個訓練樣本
• 列：query、doc 和 label
• 基於前 1000 個樣本的近似統計信息：

  	查詢	文檔	標籤
  類型	字符串	字符串	浮點數
  詳情	最小長度：49 個字符 平均長度：140.03 個字符 最大長度：306 個字符	最小長度：4 個字符 平均長度：136.03 個字符 最大長度：731 個字符	最小值：0.0 平均值：0.09 最大值：1.0

• 樣本：

  查詢	文檔	標籤
  “A leading Canadian authority on polar bears, Mitch Taylor, said: ‘We’re seeing an increase in bears that’s really unprecedented, and in places where we’re seeing a decrease in the population	Warnings about the future of the polar bear are often contrasted with the fact that worldwide population estimates have increased over the past 50 years and are relatively stable today.	1.0
  “A leading Canadian authority on polar bears, Mitch Taylor, said: ‘We’re seeing an increase in bears that’s really unprecedented, and in places where we’re seeing a decrease in the population	Species distribution models of recent years indicate that the deer tick, known as "I. scapularis," is pushing its distribution to higher latitudes of the Northeastern United States and Canada, as well as pushing and maintaining populations in the South Central and Northern Midwest regions of the United States.	0.0
  “A leading Canadian authority on polar bears, Mitch Taylor, said: ‘We’re seeing an increase in bears that’s really unprecedented, and in places where we’re seeing a decrease in the population	Bear and deer are among the animals present.	0.0

• 損失函數：BinaryCrossEntropyLoss，參數如下：

  {
      "activation_fn": "torch.nn.modules.linear.Identity",
      "pos_weight": null
  }
  

評估數據集

climate-cross-encoder-mixed-neg-v3

• 數據集：climate-cross-encoder-mixed-neg-v3，版本為 cd49b57
• 大小：4,290 個評估樣本
• 列：query、doc 和 label
• 基於前 1000 個樣本的近似統計信息：

  	查詢	文檔	標籤
  類型	字符串	字符串	浮點數
  詳情	最小長度：39 個字符 平均長度：116.67 個字符 最大長度：240 個字符	最小長度：18 個字符 平均長度：132.92 個字符 最大長度：731 個字符	最小值：0.0 平均值：0.09 最大值：1.0

• 樣本：

  查詢	文檔	標籤
  Scientific analysis of past climates shows that greenhouse gasses, principally CO2, have controlled most ancient climate changes.	Greenhouse gases, in particular carbon dioxide and methane, played a significant role during the Eocene in controlling the surface temperature.	1.0
  Scientific analysis of past climates shows that greenhouse gasses, principally CO2, have controlled most ancient climate changes.	Climatic geomorphology is of limited use to study recent (Quaternary, Holocene) large climate changes since there are seldom discernible in the geomorphological record.	0.0
  Scientific analysis of past climates shows that greenhouse gasses, principally CO2, have controlled most ancient climate changes.	There is also a close correlation between CO2 and temperature, where CO2 has a strong control over global temperatures in Earth history.	0.0

• 損失函數：BinaryCrossEntropyLoss，參數如下：

  {
      "activation_fn": "torch.nn.modules.linear.Identity",
      "pos_weight": null
  }
  

訓練超參數

非默認超參數

• eval_strategy：steps
• per_device_train_batch_size：16
• per_device_eval_batch_size：32
• learning_rate：2e-05
• warmup_ratio：0.1
• fp16：True
• load_best_model_at_end：True

所有超參數

訓練日誌

• 加粗行表示保存的檢查點。

框架版本

• Python：3.11.12
• Sentence Transformers：4.1.0
• Transformers：4.51.3
• PyTorch：2.6.0+cu124
• Accelerate：1.6.0
• Datasets：3.6.0
• Tokenizers：0.21.1

📄 引用

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