🚀 基于GooAQ训练的ModernBERT-base模型

这是一个基于Cross Encoder的模型，它使用sentence-transformers库，从cross-encoder/ms-marco-MiniLM-L6-v2微调而来。该模型可以计算文本对的得分，可用于文本重排序和语义搜索。

🚀 快速开始

此模型可用于计算文本对的得分，适用于文本重排序和语义搜索任务。下面将介绍如何使用此模型。

✨ 主要特性

• 文本重排序：能够为文本对计算得分，从而实现文本重排序。
• 语义搜索：可用于语义搜索任务，帮助用户更精准地找到相关文本。

📦 安装指南

首先，你需要安装Sentence Transformers库：

pip install -U sentence-transformers

💻 使用示例

基础用法

安装好库之后，你可以加载此模型并进行推理：

from sentence_transformers import CrossEncoder

# 从🤗 Hub下载模型
model = CrossEncoder("ayushexel/reranker-ms-marco-MiniLM-L6-v2-gooaq-bce")
# 获取文本对的得分
pairs = [
    ['what does it mean when you get a sharp pain in your left arm?', 'Pain in the left arm A pain in your left arm could mean you have a bone or joint injury, a pinched nerve, or a problem with your heart. Read on to learn more about the causes of left arm pain and what symptoms could signal a serious problem.'],
    ['what does it mean when you get a sharp pain in your left arm?', "In this Article Whether it's throbbing, aching, or sharp, everyone has been in pain. The uncomfortable sensation is a red flag. Pain in your armpit could mean that you've simply strained a muscle, which is eased with ice and rest. It could also be a sign of more serious conditions, like an infection or breast cancer."],
    ['what does it mean when you get a sharp pain in your left arm?', 'Sharp: When you feel a sudden, intense spike of pain, that qualifies as “sharp.” Sharp pain may also fit the descriptors cutting and shooting. Stabbing: Like sharp pain, stabbing pain occurs suddenly and intensely. However, stabbing pain may fade and reoccur many times.'],
    ['what does it mean when you get a sharp pain in your left arm?', 'Symptoms. A herniated disc in the neck can cause neck pain, radiating arm pain, shoulder pain, and numbness or tingling in the arm or hand. The quality and type of pain can vary from dull, aching, and difficult to localize to sharp, burning, and easy to pinpoint.'],
    ['what does it mean when you get a sharp pain in your left arm?', 'Injuries or trauma to any part of the arm or shoulder, including bone fractures, joint dislocations, and muscle strains and sprains, are common causes of arm pain. Sometimes diseases that affect other organs in the body, like peripheral vascular disease or arthritis, can be the cause of pain in the arm.'],
]
scores = model.predict(pairs)
print(scores.shape)
# (5,)

# 或者根据与单个文本的相似度对不同文本进行排序
ranks = model.rank(
    'what does it mean when you get a sharp pain in your left arm?',
    [
        'Pain in the left arm A pain in your left arm could mean you have a bone or joint injury, a pinched nerve, or a problem with your heart. Read on to learn more about the causes of left arm pain and what symptoms could signal a serious problem.',
        "In this Article Whether it's throbbing, aching, or sharp, everyone has been in pain. The uncomfortable sensation is a red flag. Pain in your armpit could mean that you've simply strained a muscle, which is eased with ice and rest. It could also be a sign of more serious conditions, like an infection or breast cancer.",
        'Sharp: When you feel a sudden, intense spike of pain, that qualifies as “sharp.” Sharp pain may also fit the descriptors cutting and shooting. Stabbing: Like sharp pain, stabbing pain occurs suddenly and intensely. However, stabbing pain may fade and reoccur many times.',
        'Symptoms. A herniated disc in the neck can cause neck pain, radiating arm pain, shoulder pain, and numbness or tingling in the arm or hand. The quality and type of pain can vary from dull, aching, and difficult to localize to sharp, burning, and easy to pinpoint.',
        'Injuries or trauma to any part of the arm or shoulder, including bone fractures, joint dislocations, and muscle strains and sprains, are common causes of arm pain. Sometimes diseases that affect other organs in the body, like peripheral vascular disease or arthritis, can be the cause of pain in the arm.',
    ]
)
# [{'corpus_id': ..., 'score': ...}, {'corpus_id': ..., 'score': ...}, ...]

📚 详细文档

模型详情

模型描述

模型来源

• 文档：Sentence Transformers文档
• 文档：Cross Encoder文档
• 代码仓库：GitHub上的Sentence Transformers
• Hugging Face：Hugging Face上的Cross Encoders

评估

指标

交叉编码器重排序（数据集：gooaq-dev）

使用CrossEncoderRerankingEvaluator进行评估，参数如下：

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

交叉编码器重排序（数据集：NanoMSMARCO_R100、NanoNFCorpus_R100和NanoNQ_R100）

使用CrossEncoderRerankingEvaluator进行评估，参数如下：

{
    "at_k": 10,
    "always_rerank_positives": true
}

交叉编码器Nano BEIR（数据集：NanoBEIR_R100_mean）

使用CrossEncoderNanoBEIREvaluator进行评估，参数如下：

{
    "dataset_names": [
        "msmarco",
        "nfcorpus",
        "nq"
    ],
    "rerank_k": 100,
    "at_k": 10,
    "always_rerank_positives": true
}

训练详情

训练数据集

未命名数据集

• 规模：2,223,773个训练样本
• 列名：question、answer和label
• 基于前1000个样本的近似统计信息： | | 问题 | 答案 | 标签 | |------|------|------|------| | 类型 | 字符串 | 字符串 | 整数 | | 详情 |
  • 最小：19个字符
  • 平均：45.87个字符
  • 最大：88个字符
  |
  • 最小：61个字符
  • 平均：253.13个字符
  • 最大：374个字符
  |
  • 0：约86.70%
  • 1：约13.30%
  |
• 样本示例： | 问题 | 答案 | 标签 | |------|------|------| | what does it mean when you get a sharp pain in your left arm? | Pain in the left arm A pain in your left arm could mean you have a bone or joint injury, a pinched nerve, or a problem with your heart. Read on to learn more about the causes of left arm pain and what symptoms could signal a serious problem. | 1 | | what does it mean when you get a sharp pain in your left arm? | In this Article Whether it's throbbing, aching, or sharp, everyone has been in pain. The uncomfortable sensation is a red flag. Pain in your armpit could mean that you've simply strained a muscle, which is eased with ice and rest. It could also be a sign of more serious conditions, like an infection or breast cancer. | 0 | | what does it mean when you get a sharp pain in your left arm? | Sharp: When you feel a sudden, intense spike of pain, that qualifies as “sharp.” Sharp pain may also fit the descriptors cutting and shooting. Stabbing: Like sharp pain, stabbing pain occurs suddenly and intensely. However, stabbing pain may fade and reoccur many times. | 0 |
• 损失函数：BinaryCrossEntropyLoss，参数如下：

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

训练超参数

非默认超参数

• eval_strategy：按步骤评估
• per_device_train_batch_size：2048
• per_device_eval_batch_size：2048
• learning_rate：2e-05
• warmup_ratio：0.1
• seed：12
• bf16：True
• dataloader_num_workers：12
• load_best_model_at_end：True

所有超参数

训练日志

注：加粗行表示保存的检查点。

框架版本

• Python：3.11.0
• Sentence Transformers：4.0.1
• Transformers：4.50.3
• PyTorch：2.6.0+cu124
• Accelerate：1.5.2
• Datasets：3.5.0
• Tokenizers：0.21.1

📄 许可证

本模型使用apache-2.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",
}