medrxiv-clustering-s2s

<!-- adapted from https://github.com/huggingface/huggingface_hub/blob/v0.30.2/src/huggingface_hub/templates/datasetcard_template.md -->

<div align="center" style="padding: 40px 20px; background-color: white; border-radius: 12px; box-shadow: 0 2px 10px rgba(0, 0, 0, 0.05); max-width: 600px; margin: 0 auto;">
<h1 style="font-size: 3.5rem; color: #1a1a1a; margin: 0 0 20px 0; letter-spacing: 2px; font-weight: 700;">MedrxivClusteringS2S.v2</h1>
<div style="font-size: 1.5rem; color: #4a4a4a; margin-bottom: 5px; font-weight: 300;">An <a href="https://github.com/embeddings-benchmark/mteb" style="color: #2c5282; font-weight: 600; text-decoration: none;" onmouseover="this.style.textDecoration='underline'" onmouseout="this.style.textDecoration='none'">MTEB</a> dataset</div>
<div style="font-size: 0.9rem; color: #2c5282; margin-top: 10px;">Massive Text Embedding Benchmark</div>
</div>

Clustering of titles from medrxiv across 51 categories.

| | |
|---------------|---------------------------------------------|
| Task category | t2c |
| Domains | Academic, Medical, Written |
| Reference | https://api.medrxiv.org/ |

## How to evaluate on this task

You can evaluate an embedding model on this dataset using the following code:

```python
import mteb

task = mteb.get_tasks(["MedrxivClusteringS2S.v2"])
evaluator = mteb.MTEB(task)

model = mteb.get_model(YOUR_MODEL)
evaluator.run(model)
```

<!-- Datasets want link to arxiv in readme to autolink dataset with paper -->
To learn more about how to run models on `mteb` task check out the [GitHub repitory](https://github.com/embeddings-benchmark/mteb).

## Citation

If you use this dataset, please cite the dataset as well as [mteb](https://github.com/embeddings-benchmark/mteb), as this dataset likely includes additional processing as a part of the [MMTEB Contribution](https://github.com/embeddings-benchmark/mteb/tree/main/docs/mmteb).

```bibtex

@article{enevoldsen2025mmtebmassivemultilingualtext,
title={MMTEB: Massive Multilingual Text Embedding Benchmark},
author={Kenneth Enevoldsen and Isaac Chung and Imene Kerboua and Márton Kardos and Ashwin Mathur and David Stap and Jay Gala and Wissam Siblini and Dominik Krzemiński and Genta Indra Winata and Saba Sturua and Saiteja Utpala and Mathieu Ciancone and Marion Schaeffer and Gabriel Sequeira and Diganta Misra and Shreeya Dhakal and Jonathan Rystrøm and Roman Solomatin and Ömer Çağatan and Akash Kundu and Martin Bernstorff and Shitao Xiao and Akshita Sukhlecha and Bhavish Pahwa and Rafał Poświata and Kranthi Kiran GV and Shawon Ashraf and Daniel Auras and Björn Plüster and Jan Philipp Harries and Loïc Magne and Isabelle Mohr and Mariya Hendriksen and Dawei Zhu and Hippolyte Gisserot-Boukhlef and Tom Aarsen and Jan Kostkan and Konrad Wojtasik and Taemin Lee and Marek Šuppa and Crystina Zhang and Roberta Rocca and Mohammed Hamdy and Andrianos Michail and John Yang and Manuel Faysse and Aleksei Vatolin and Nandan Thakur and Manan Dey and Dipam Vasani and Pranjal Chitale and Simone Tedeschi and Nguyen Tai and Artem Snegirev and Michael Günther and Mengzhou Xia and Weijia Shi and Xing Han Lù and Jordan Clive and Gayatri Krishnakumar and Anna Maksimova and Silvan Wehrli and Maria Tikhonova and Henil Panchal and Aleksandr Abramov and Malte Ostendorff and Zheng Liu and Simon Clematide and Lester James Miranda and Alena Fenogenova and Guangyu Song and Ruqiya Bin Safi and Wen-Ding Li and Alessia Borghini and Federico Cassano and Hongjin Su and Jimmy Lin and Howard Yen and Lasse Hansen and Sara Hooker and Chenghao Xiao and Vaibhav Adlakha and Orion Weller and Siva Reddy and Niklas Muennighoff},
publisher = {arXiv},
journal={arXiv preprint arXiv:2502.13595},
year={2025},
url={https://arxiv.org/abs/2502.13595},
doi = {10.48550/arXiv.2502.13595},
}

@article{muennighoff2022mteb,
author = {Muennighoff, Niklas and Tazi, Nouamane and Magne, Lo{\"\i}c and Reimers, Nils},
title = {MTEB: Massive Text Embedding Benchmark},
publisher = {arXiv},
journal={arXiv preprint arXiv:2210.07316},
year = {2022}
url = {https://arxiv.org/abs/2210.07316},
doi = {10.48550/ARXIV.2210.07316},
}
```

# Dataset Statistics
<details>
<summary> Dataset Statistics</summary>

The following code contains the descriptive statistics from the task. These can also be obtained using:

```python
import mteb

task = mteb.get_task("MedrxivClusteringS2S.v2")

desc_stats = task.metadata.descriptive_stats
```

```json
{
"test": {
"num_samples": 37500,
"number_of_characters": 4301276,
"min_text_length": 18,
"average_text_length": 114.70069333333333,
"max_text_length": 339,
"min_labels_per_text": 6,
"average_labels_per_text": 1.0,
"max_labels_per_text": 8830,
"unique_labels": 51,
"labels": {
"epidemiology": {
"count": 6656
},
"public and global health": {
"count": 3595
},
"oncology": {
"count": 845
},
"allergy and immunology": {
"count": 464
},
"orthopedics": {
"count": 104
},
"health informatics": {
"count": 1107
},
"occupational and environmental health": {
"count": 415
},
"infectious diseases": {
"count": 8830
},
"genetic and genomic medicine": {
"count": 1918
},
"health policy": {
"count": 527
},
"gastroenterology": {
"count": 343
},
"radiology and imaging": {
"count": 541
},
"pain medicine": {
"count": 121
},
"neurology": {
"count": 1773
},
"primary care research": {
"count": 232
},
"rheumatology": {
"count": 189
},
"endocrinology": {
"count": 419
},
"hematology": {
"count": 202
},
"addiction medicine": {
"count": 178
},
"pediatrics": {
"count": 589
},
"cardiovascular medicine": {
"count": 855
},
"obstetrics and gynecology": {
"count": 373
},
"health systems and quality improvement": {
"count": 491
},
"nephrology": {
"count": 241
},
"respiratory medicine": {
"count": 482
},
"geriatric medicine": {
"count": 169
},
"dentistry and oral medicine": {
"count": 159
},
"psychiatry and clinical psychology": {
"count": 1781
},
"nutrition": {
"count": 240
},
"intensive care and critical care medicine": {
"count": 368
},
"rehabilitation medicine and physical therapy": {
"count": 322
},
"otolaryngology": {
"count": 166
},
"nursing": {
"count": 93
},
"transplantation": {
"count": 118
},
"health economics": {
"count": 327
},
"sports medicine": {
"count": 180
},
"hiv aids": {
"count": 363
},
"dermatology": {
"count": 98
},
"pathology": {
"count": 223
},
"emergency medicine": {
"count": 191
},
"pharmacology and therapeutics": {
"count": 221
},
"ophthalmology": {
"count": 220
},
"medical ethics": {
"count": 46
},
"palliative medicine": {
"count": 45
},
"sexual and reproductive health": {
"count": 156
},
"medical education": {
"count": 203
},
"surgery": {
"count": 162
},
"urology": {
"count": 65
},
"anesthesia": {
"count": 72
},
"toxicology": {
"count": 16
},
"forensic medicine": {
"count": 6
}
}
}
}
```

</details>

---
*This dataset card was automatically generated using [MTEB](https://github.com/embeddings-benchmark/mteb)*

<!-- adapted from https://github.com/huggingface/huggingface_hub/blob/v0.30.2/src/huggingface_hub/templates/datasetcard_template.md -->
<div align="center" style="padding: 40px 20px; background-color: white; border-radius: 12px; box-shadow: 0 2px 10px rgba(0, 0, 0, 0.05); max-width: 600px; margin: 0 auto;">
<h1 style="font-size: 3.5rem; color: #1a1a1a; margin: 0 0 20px 0; letter-spacing: 2px; font-weight: 700;">MedrxivClusteringS2S.v2</h1>
<div style="font-size: 1.5rem; color: #4a4a4a; margin-bottom: 5px; font-weight: 300;">一款隶属于大规模文本嵌入基准（Massive Text Embedding Benchmark，MTEB）的数据集</div>
<div style="font-size: 0.9rem; color: #2c5282; margin-top: 10px;">大规模文本嵌入基准</div>
</div>

针对medRxiv平台的论文标题，按51个类别开展聚类任务。

| | |
|---------------|---------------------------------------------|
| 任务类别 | t2c |
| 领域 | 学术、医学、书面文本 |
| 参考来源 | https://api.medrxiv.org/ |

## 任务评估方法

您可通过下述代码在该数据集上评估嵌入模型：

python
import mteb

task = mteb.get_tasks(["MedrxivClusteringS2S.v2"])
evaluator = mteb.MTEB(task)

model = mteb.get_model(YOUR_MODEL)
evaluator.run(model)

<!-- Datasets want link to arxiv in readme to autolink dataset with paper -->
若需了解如何在MTEB任务中运行模型，请访问其[GitHub仓库](https://github.com/embeddings-benchmark/mteb)。

## 引用说明

若您使用本数据集，请同时引用该数据集与[MTEB](https://github.com/embeddings-benchmark/mteb)的相关文献——本数据集大概率经过了[MMTEB贡献项目](https://github.com/embeddings-benchmark/mteb/tree/main/docs/mmteb)中的额外处理流程。

bibtex

@article{enevoldsen2025mmtebmassivemultilingualtext,
title={MMTEB: Massive Multilingual Text Embedding Benchmark},
author={Kenneth Enevoldsen and Isaac Chung and Imene Kerboua and Márton Kardos and Ashwin Mathur and David Stap and Jay Gala and Wissam Siblini and Dominik Krzemiński and Genta Indra Winata and Saba Sturua and Saiteja Utpala and Mathieu Ciancone and Marion Schaeffer and Gabriel Sequeira and Diganta Misra and Shreeya Dhakal and Jonathan Rystrøm and Roman Solomatin and Ömer Çağatan and Akash Kundu and Martin Bernstorff and Shitao Xiao and Akshita Sukhlecha and Bhavish Pahwa and Rafał Poświata and Kranthi Kiran GV and Shawon Ashraf and Daniel Auras and Björn Plüster and Jan Philipp Harries and Lo{"i"c Magne and Isabelle Mohr and Mariya Hendriksen and Dawei Zhu and Hippolyte Gisserot-Boukhlef and Tom Aarsen and Jan Kostkan and Konrad Wojtasik and Taemin Lee and Marek Šuppa and Crystina Zhang and Roberta Rocca and Mohammed Hamdy and Andrianos Michail and John Yang and Manuel Faysse and Aleksei Vatolin and Nandan Thakur and Manan Dey and Dipam Vasani and Pranjal Chitale and Simone Tedeschi and Nguyen Tai and Artem Snegirev and Michael Günther and Mengzhou Xia and Weijia Shi and Xing Han Lù and Jordan Clive and Gayatri Krishnakumar and Anna Maksimova and Silvan Wehrli and Maria Tikhonova and Henil Panchal and Aleksandr Abramov and Malte Ostendorff and Zheng Liu and Simon Clematide and Lester James Miranda and Alena Fenogenova and Guangyu Song and Ruqiya Bin Safi and Wen-Ding Li and Alessia Borghini and Federico Cassano and Hongjin Su and Jimmy Lin and Howard Yen and Lasse Hansen and Sara Hooker and Chenghao Xiao and Vaibhav Adlakha and Orion Weller and Siva Reddy and Niklas Muennighoff},
publisher = {arXiv},
journal={arXiv preprint arXiv:2502.13595},
year={2025},
url={https://arxiv.org/abs/2502.13595},
doi = {10.48550/arXiv.2502.13595},
}

@article{muennighoff2022mteb,
author = {Muennighoff, Niklas and Tazi, Nouamane and Magne, Lo{"i"c} and Reimers, Nils},
title = {MTEB: Massive Text Embedding Benchmark},
publisher = {arXiv},
journal={arXiv preprint arXiv:2210.07316},
year = {2022}
url = {https://arxiv.org/abs/2210.07316},
doi = {10.48550/ARXIV.2210.07316},
}

# 数据集统计信息
<details>
<summary>数据集统计信息</summary>

下述代码展示了本任务的描述性统计数据，您也可以通过以下代码获取这些数据：

python
import mteb

task = mteb.get_task("MedrxivClusteringS2S.v2")

desc_stats = task.metadata.descriptive_stats

json
{
"test": {
"num_samples": 37500,
"number_of_characters": 4301276,
"min_text_length": 18,
"average_text_length": 114.70069333333333,
"max_text_length": 339,
"min_labels_per_text": 6,
"average_labels_per_text": 1.0,
"max_labels_per_text": 8830,
"unique_labels": 51,
"labels": {
"epidemiology": {
"count": 6656
},
"public and global health": {
"count": 3595
},
"oncology": {
"count": 845
},
"allergy and immunology": {
"count": 464
},
"orthopedics": {
"count": 104
},
"health informatics": {
"count": 1107
},
"occupational and environmental health": {
"count": 415
},
"infectious diseases": {
"count": 8830
},
"genetic and genomic medicine": {
"count": 1918
},
"health policy": {
"count": 527
},
"gastroenterology": {
"count": 343
},
"radiology and imaging": {
"count": 541
},
"pain medicine": {
"count": 121
},
"neurology": {
"count": 1773
},
"primary care research": {
"count": 232
},
"rheumatology": {
"count": 189
},
"endocrinology": {
"count": 419
},
"hematology": {
"count": 202
},
"addiction medicine": {
"count": 178
},
"pediatrics": {
"count": 589
},
"cardiovascular medicine": {
"count": 855
},
"obstetrics and gynecology": {
"count": 373
},
"health systems and quality improvement": {
"count": 491
},
"nephrology": {
"count": 241
},
"respiratory medicine": {
"count": 482
},
"geriatric medicine": {
"count": 169
},
"dentistry and oral medicine": {
"count": 159
},
"psychiatry and clinical psychology": {
"count": 1781
},
"nutrition": {
"count": 240
},
"intensive care and critical care medicine": {
"count": 368
},
"rehabilitation medicine and physical therapy": {
"count": 322
},
"otolaryngology": {
"count": 166
},
"nursing": {
"count": 93
},
"transplantation": {
"count": 118
},
"health economics": {
"count": 327
},
"sports medicine": {
"count": 180
},
"hiv aids": {
"count": 363
},
"dermatology": {
"count": 98
},
"pathology": {
"count": 223
},
"emergency medicine": {
"count": 191
},
"pharmacology and therapeutics": {
"count": 221
},
"ophthalmology": {
"count": 220
},
"medical ethics": {
"count": 46
},
"palliative medicine": {
"count": 45
},
"sexual and reproductive health": {
"count": 156
},
"medical education": {
"count": 203
},
"surgery": {
"count": 162
},
"urology": {
"count": 65
},
"anesthesia": {
"count": 72
},
"toxicology": {
"count": 16
},
"forensic medicine": {
"count": 6
}
}
}
}

</details>

---
*本数据集卡片由[MTEB](https://github.com/embeddings-benchmark/mteb)自动生成*