MattStammers commited on
Commit
962956c
·
verified ·
1 Parent(s): 2efa1a7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +197 -197
README.md CHANGED
@@ -1,198 +1,198 @@
1
- ---
2
- license: cc-by-nc-4.0
3
- language:
4
- - en
5
- metrics:
6
- - precision
7
- - recall
8
- - brier_score
9
- - f1
10
- - matthews_correlation
11
- base_model:
12
- - sentence-transformers/all-mpnet-base-v2
13
- tags:
14
- - IBD
15
- - cohort_identification
16
- - case_finding
17
- ---
18
- # Model Card for BioClinicalBERT IBD
19
-
20
- The model classifies documents as either IBD or Not IBD
21
-
22
- ## Model Details
23
-
24
- ### Model Description
25
-
26
- As above. This is a model trained to detect IBD patients from clinical text
27
-
28
- - **Developed by:** Matt Stammers
29
- - **Funded by:** University Hospital Foundation NHS Trust
30
- - **Shared by:** Matt Stammers - SETT Data and AI Clinical Lead
31
- - **Model type:** BERT Transformer
32
- - **Language(s) (NLP):** English
33
- - **License:** cc-by-nc-4.0
34
- - **Finetuned from model:** sentence-transformers/all-mpnet-base-v2
35
-
36
- ### Model Sources
37
-
38
- - **Repository:** https://huggingface.co/MattStammers/SBERT_IBD
39
- - **Paper:** Arxiv (Pending)
40
- - **Demo:** https://huggingface.co/spaces/MattStammers/IBD_Cohort_Identification
41
-
42
- ## Uses
43
-
44
- For document classification tasks to differentiate between documents likely to be diagnostic of IBD and those unlikely to be diagnostic of IBD.
45
-
46
- ### Direct Use
47
-
48
- This model can be used directly at [Cohort Identification Demo](https://huggingface.co/spaces/MattStammers/IBD_Cohort_Identification)
49
-
50
- ### Downstream Use
51
-
52
- Others can build on this model and improve it but only for non-commercial purposes.
53
-
54
- ### Out-of-Scope Use
55
-
56
- This model is less powerful (in terms of F1 Score) when making predictions at the patient level by 1-2%. It can be used for this purpose but with care.
57
-
58
- ## Bias, Risks, and Limitations
59
-
60
- This model does not have any major known biases. It only has some slight bias tendencies.
61
-
62
- ### Recommendations
63
-
64
- This model might work well in any population but has not yet been tested in this regard as such.
65
-
66
- ## How to Get Started with the Model
67
-
68
- Use the code below to get started with the model.
69
-
70
- The model is best used with the transformers library.
71
-
72
- ## Training Details
73
-
74
- ### Training Data
75
-
76
- The model was trained on fully pseudonymised clinical information at UHSFT which was carefully labelled by a consultant (attending) physician and evaluated against a randomly selected internal holdout set.
77
-
78
- ### Training Procedure
79
-
80
- See the paper for more information on the training procedure
81
-
82
- #### Training Hyperparameters
83
-
84
- - **Training regime:** fp32
85
-
86
- #### Speeds, Sizes, Times
87
-
88
- This model (part of a set of models) took 213.55 minutes to train
89
-
90
- ## Evaluation
91
-
92
- The model was internally validated against a holdout set
93
-
94
- ### Testing Data, Factors & Metrics
95
-
96
- #### Testing Data
97
-
98
- The testing data cannot be revealed due to IG regulations and to remain compliant with GDPR, only the resulting model can be
99
-
100
- #### Factors
101
-
102
- IBD vs Not-IBD
103
-
104
- #### Metrics
105
-
106
- Full evaluation metrics are available in the paper with a summary below
107
-
108
- ### Results
109
-
110
- | Model | Doc Coverage | Accuracy | Precision | Recall | Specificity | NPV | F1 Score | MCC |
111
- |-------|------------------|-----------------------------------|-----------------------------------|-----------------------------------|----------------------------------|----------------------------------|----------------------------------|------------------------------------|
112
- | SBERT | 768 (100.00%) | 89.67% (CI: 86.64% - 92.08%) | 88.81% (CI: 85.44% - 91.48%) | 99.20% (CI: 97.68% - 99.73%) | 56.48% (CI: 47.07% - 65.45%) | 95.31% (CI: 87.10% - 98.39%) | 93.72% (CI: 92.50% - 94.90%) | 0.6844 (CI: 0.6120 - 0.7545) |
113
-
114
-
115
- #### Summary
116
-
117
- Overall performance of the model is high with an F1 Score of >94% on our internal holdout set.
118
-
119
- ## Environmental Impact
120
-
121
- Training the model used 2.01kWh of energy emmitting 416.73 grams of CO2
122
-
123
- - **Hardware Type:** L40S
124
- - **Hours used:** 0.1
125
- - **Carbon Emitted:** 0.009 Kg CO2
126
-
127
- ## Citation
128
-
129
- Arxiv (Pending)
130
-
131
- ## Glossary
132
-
133
- | Term | Description |
134
- |-------------------------------------|-------------|
135
- | **Accuracy** | The percentage of results that were correct among all results from the system. Calc: (TP + TN) / (TP + FP + TN + FN). |
136
- | **Precision (PPV)** | Also called positive predictive value (PPV), it is the percentage of true positive results among all results that the system flagged as positive. Calc: TP / (TP + FP). |
137
- | **Negative Predictive Value (NPV)** | The percentage of results that were true negative (TN) among all results that the system flagged as negative. Calc: TN / (TN + FN). |
138
- | **Recall** | Also called sensitivity. The percentage of results flagged positive among all results that should have been obtained. Calc: TP / (TP + FN). |
139
- | **Specificity** | The percentage of results that were flagged negative among all negative results. Calc: TN / (TN + FP). |
140
- | **F1-Score** | The harmonic mean of PPV/precision and sensitivity/recall. Calc: 2 × (Precision × Recall) / (Precision + Recall). Moderately useful in the context of class imbalance. |
141
- | **Matthews’ Correlation Coefficient (MCC)** | A statistical measure used to evaluate the quality of binary classifications. Unlike other metrics, MCC considers all four categories of a confusion matrix. Calc: (TP × TN − FP × FN) / √((TP + FP)(TP + FN)(TN + FP)(TN + FN)). |
142
- | **Precision / Recall AUC** | Represents the area under the Precision-Recall curve, which plots Precision against Recall at various threshold settings. It is more resistant to class imbalance than alternatives like AUROC. |
143
- | **Demographic Parity (DP)** | Demographic Parity, also known as Statistical Parity, requires that the probability of a positive prediction is the same across different demographic groups. Calc: DP = P(Ŷ=1∣A=a) = P(Ŷ=1∣A=b). This figure is given as an absolute difference where positive values suggest the more privileged group gains and negative values the reverse. |
144
- | **Equal Opportunity (EO)** | Equal Opportunity focuses on equalising the true positive rates across groups. Among those who truly belong to the positive class, the model should predict positive outcomes at equal rates across groups. Calc: EO = P(Ŷ=1∣Y=1, A=a) = P(Ŷ=1∣Y=1, A=b). A higher value indicates a bias against the more vulnerable group. |
145
- | **Disparate Impact (DI)** | Divides the protected group’s positive prediction rate by that of the most-favoured group. If the ratio is below 0.8 or above 1.25, disparate impact is considered present. Calc: DI = P(Ŷ=1∣A=unfavoured) / P(Ŷ=1∣A=favoured). Values outside 0.8–1.25 range suggest bias. |
146
- | **Execution Time / Energy / CO₂ Emissions** | Measured in minutes and total energy consumption in kilowatt-hours (kWh), which is then converted to CO₂ emissions using a factor of 0.20705 Kg CO₂e per kWh. |
147
-
148
- ## Model Card Authors
149
-
150
- Matt Stammers - Computational Gastroenterologist
151
-
152
- ## Model Card Contact
153
-
154
- m.stammers@soton.ac.uk
155
-
156
- ## Base Training Data for the Model
157
-
158
- #### Training data
159
-
160
- We use the concatenation from multiple datasets to fine-tune our model. The total number of sentence pairs is above 1 billion sentences.
161
- We sampled each dataset given a weighted probability which configuration is detailed in the `data_config.json` file.
162
-
163
-
164
- | Dataset | Paper | Number of training tuples |
165
- |--------------------------------------------------------|:----------------------------------------:|:--------------------------:|
166
- | [Reddit comments (2015-2018)](https://github.com/PolyAI-LDN/conversational-datasets/tree/master/reddit) | [paper](https://arxiv.org/abs/1904.06472) | 726,484,430 |
167
- | [S2ORC](https://github.com/allenai/s2orc) Citation pairs (Abstracts) | [paper](https://aclanthology.org/2020.acl-main.447/) | 116,288,806 |
168
- | [WikiAnswers](https://github.com/afader/oqa#wikianswers-corpus) Duplicate question pairs | [paper](https://doi.org/10.1145/2623330.2623677) | 77,427,422 |
169
- | [PAQ](https://github.com/facebookresearch/PAQ) (Question, Answer) pairs | [paper](https://arxiv.org/abs/2102.07033) | 64,371,441 |
170
- | [S2ORC](https://github.com/allenai/s2orc) Citation pairs (Titles) | [paper](https://aclanthology.org/2020.acl-main.447/) | 52,603,982 |
171
- | [S2ORC](https://github.com/allenai/s2orc) (Title, Abstract) | [paper](https://aclanthology.org/2020.acl-main.447/) | 41,769,185 |
172
- | [Stack Exchange](https://huggingface.co/datasets/flax-sentence-embeddings/stackexchange_xml) (Title, Body) pairs | - | 25,316,456 |
173
- | [Stack Exchange](https://huggingface.co/datasets/flax-sentence-embeddings/stackexchange_xml) (Title+Body, Answer) pairs | - | 21,396,559 |
174
- | [Stack Exchange](https://huggingface.co/datasets/flax-sentence-embeddings/stackexchange_xml) (Title, Answer) pairs | - | 21,396,559 |
175
- | [MS MARCO](https://microsoft.github.io/msmarco/) triplets | [paper](https://doi.org/10.1145/3404835.3462804) | 9,144,553 |
176
- | [GOOAQ: Open Question Answering with Diverse Answer Types](https://github.com/allenai/gooaq) | [paper](https://arxiv.org/pdf/2104.08727.pdf) | 3,012,496 |
177
- | [Yahoo Answers](https://www.kaggle.com/soumikrakshit/yahoo-answers-dataset) (Title, Answer) | [paper](https://proceedings.neurips.cc/paper/2015/hash/250cf8b51c773f3f8dc8b4be867a9a02-Abstract.html) | 1,198,260 |
178
- | [Code Search](https://huggingface.co/datasets/code_search_net) | - | 1,151,414 |
179
- | [COCO](https://cocodataset.org/#home) Image captions | [paper](https://link.springer.com/chapter/10.1007%2F978-3-319-10602-1_48) | 828,395|
180
- | [SPECTER](https://github.com/allenai/specter) citation triplets | [paper](https://doi.org/10.18653/v1/2020.acl-main.207) | 684,100 |
181
- | [Yahoo Answers](https://www.kaggle.com/soumikrakshit/yahoo-answers-dataset) (Question, Answer) | [paper](https://proceedings.neurips.cc/paper/2015/hash/250cf8b51c773f3f8dc8b4be867a9a02-Abstract.html) | 681,164 |
182
- | [Yahoo Answers](https://www.kaggle.com/soumikrakshit/yahoo-answers-dataset) (Title, Question) | [paper](https://proceedings.neurips.cc/paper/2015/hash/250cf8b51c773f3f8dc8b4be867a9a02-Abstract.html) | 659,896 |
183
- | [SearchQA](https://huggingface.co/datasets/search_qa) | [paper](https://arxiv.org/abs/1704.05179) | 582,261 |
184
- | [Eli5](https://huggingface.co/datasets/eli5) | [paper](https://doi.org/10.18653/v1/p19-1346) | 325,475 |
185
- | [Flickr 30k](https://shannon.cs.illinois.edu/DenotationGraph/) | [paper](https://transacl.org/ojs/index.php/tacl/article/view/229/33) | 317,695 |
186
- | [Stack Exchange](https://huggingface.co/datasets/flax-sentence-embeddings/stackexchange_xml) Duplicate questions (titles) | | 304,525 |
187
- | AllNLI ([SNLI](https://nlp.stanford.edu/projects/snli/) and [MultiNLI](https://cims.nyu.edu/~sbowman/multinli/) | [paper SNLI](https://doi.org/10.18653/v1/d15-1075), [paper MultiNLI](https://doi.org/10.18653/v1/n18-1101) | 277,230 |
188
- | [Stack Exchange](https://huggingface.co/datasets/flax-sentence-embeddings/stackexchange_xml) Duplicate questions (bodies) | | 250,519 |
189
- | [Stack Exchange](https://huggingface.co/datasets/flax-sentence-embeddings/stackexchange_xml) Duplicate questions (titles+bodies) | | 250,460 |
190
- | [Sentence Compression](https://github.com/google-research-datasets/sentence-compression) | [paper](https://www.aclweb.org/anthology/D13-1155/) | 180,000 |
191
- | [Wikihow](https://github.com/pvl/wikihow_pairs_dataset) | [paper](https://arxiv.org/abs/1810.09305) | 128,542 |
192
- | [Altlex](https://github.com/chridey/altlex/) | [paper](https://aclanthology.org/P16-1135.pdf) | 112,696 |
193
- | [Quora Question Triplets](https://quoradata.quora.com/First-Quora-Dataset-Release-Question-Pairs) | - | 103,663 |
194
- | [Simple Wikipedia](https://cs.pomona.edu/~dkauchak/simplification/) | [paper](https://www.aclweb.org/anthology/P11-2117/) | 102,225 |
195
- | [Natural Questions (NQ)](https://ai.google.com/research/NaturalQuestions) | [paper](https://transacl.org/ojs/index.php/tacl/article/view/1455) | 100,231 |
196
- | [SQuAD2.0](https://rajpurkar.github.io/SQuAD-explorer/) | [paper](https://aclanthology.org/P18-2124.pdf) | 87,599 |
197
- | [TriviaQA](https://huggingface.co/datasets/trivia_qa) | - | 73,346 |
198
  | **Total** | | **1,170,060,424** |
 
1
+ ---
2
+ license: cc-by-nc-4.0
3
+ language:
4
+ - en
5
+ metrics:
6
+ - precision
7
+ - recall
8
+ - brier_score
9
+ - f1
10
+ - matthews_correlation
11
+ base_model:
12
+ - sentence-transformers/all-mpnet-base-v2
13
+ tags:
14
+ - IBD
15
+ - cohort_identification
16
+ - case_finding
17
+ ---
18
+ # Model Card for BioClinicalBERT IBD
19
+
20
+ The model classifies documents as either IBD or Not IBD
21
+
22
+ ## Model Details
23
+
24
+ ### Model Description
25
+
26
+ As above. This is a model trained to detect IBD patients from clinical text
27
+
28
+ - **Developed by:** Matt Stammers
29
+ - **Funded by:** University Hospital Foundation NHS Trust
30
+ - **Shared by:** Matt Stammers - SETT Data and AI Clinical Lead
31
+ - **Model type:** BERT Transformer
32
+ - **Language(s) (NLP):** English
33
+ - **License:** cc-by-nc-4.0
34
+ - **Finetuned from model:** sentence-transformers/all-mpnet-base-v2
35
+
36
+ ### Model Sources
37
+
38
+ - **Repository:** https://huggingface.co/MattStammers/SBERT_IBD
39
+ - **Paper:** MedRxiv- [MedRxiv Paper](https://www.medrxiv.org/content/10.1101/2025.07.06.25330961v1)
40
+ - **Demo:** https://huggingface.co/spaces/MattStammers/IBD_Cohort_Identification
41
+
42
+ ## Uses
43
+
44
+ For document classification tasks to differentiate between documents likely to be diagnostic of IBD and those unlikely to be diagnostic of IBD.
45
+
46
+ ### Direct Use
47
+
48
+ This model can be used directly at [Cohort Identification Demo](https://huggingface.co/spaces/MattStammers/IBD_Cohort_Identification)
49
+
50
+ ### Downstream Use
51
+
52
+ Others can build on this model and improve it but only for non-commercial purposes.
53
+
54
+ ### Out-of-Scope Use
55
+
56
+ This model is less powerful (in terms of F1 Score) when making predictions at the patient level by 1-2%. It can be used for this purpose but with care.
57
+
58
+ ## Bias, Risks, and Limitations
59
+
60
+ This model does not have any major known biases. It only has some slight bias tendencies.
61
+
62
+ ### Recommendations
63
+
64
+ This model might work well in any population but has not yet been tested in this regard as such.
65
+
66
+ ## How to Get Started with the Model
67
+
68
+ Use the code below to get started with the model.
69
+
70
+ The model is best used with the transformers library.
71
+
72
+ ## Training Details
73
+
74
+ ### Training Data
75
+
76
+ The model was trained on fully pseudonymised clinical information at UHSFT which was carefully labelled by a consultant (attending) physician and evaluated against a randomly selected internal holdout set.
77
+
78
+ ### Training Procedure
79
+
80
+ See the paper for more information on the training procedure
81
+
82
+ #### Training Hyperparameters
83
+
84
+ - **Training regime:** fp32
85
+
86
+ #### Speeds, Sizes, Times
87
+
88
+ This model (part of a set of models) took 213.55 minutes to train
89
+
90
+ ## Evaluation
91
+
92
+ The model was internally validated against a holdout set
93
+
94
+ ### Testing Data, Factors & Metrics
95
+
96
+ #### Testing Data
97
+
98
+ The testing data cannot be revealed due to IG regulations and to remain compliant with GDPR, only the resulting model can be
99
+
100
+ #### Factors
101
+
102
+ IBD vs Not-IBD
103
+
104
+ #### Metrics
105
+
106
+ Full evaluation metrics are available in the paper with a summary below
107
+
108
+ ### Results
109
+
110
+ | Model | Doc Coverage | Accuracy | Precision | Recall | Specificity | NPV | F1 Score | MCC |
111
+ |-------|------------------|-----------------------------------|-----------------------------------|-----------------------------------|----------------------------------|----------------------------------|----------------------------------|------------------------------------|
112
+ | SBERT | 768 (100.00%) | 89.67% (CI: 86.64% - 92.08%) | 88.81% (CI: 85.44% - 91.48%) | 99.20% (CI: 97.68% - 99.73%) | 56.48% (CI: 47.07% - 65.45%) | 95.31% (CI: 87.10% - 98.39%) | 93.72% (CI: 92.50% - 94.90%) | 0.6844 (CI: 0.6120 - 0.7545) |
113
+
114
+
115
+ #### Summary
116
+
117
+ Overall performance of the model is high with an F1 Score of >94% on our internal holdout set.
118
+
119
+ ## Environmental Impact
120
+
121
+ Training the model used 2.01kWh of energy emmitting 416.73 grams of CO2
122
+
123
+ - **Hardware Type:** L40S
124
+ - **Hours used:** 0.1
125
+ - **Carbon Emitted:** 0.009 Kg CO2
126
+
127
+ ## Citation
128
+ Stammers M, Gwiggner M, Nouraei R, Metcalf C, Batchelor J. From Rule-Based to DeepSeek R1: A Robust Comparative Evaluation of Fifty Years of Natural Language Processing (NLP) Models To Identify Inflammatory Bowel Disease Cohorts. medRxiv. 2025:2025-07.
129
+ MedRxiv- [MedRxiv Paper](https://www.medrxiv.org/content/10.1101/2025.07.06.25330961v1)
130
+
131
+ ## Glossary
132
+
133
+ | Term | Description |
134
+ |-------------------------------------|-------------|
135
+ | **Accuracy** | The percentage of results that were correct among all results from the system. Calc: (TP + TN) / (TP + FP + TN + FN). |
136
+ | **Precision (PPV)** | Also called positive predictive value (PPV), it is the percentage of true positive results among all results that the system flagged as positive. Calc: TP / (TP + FP). |
137
+ | **Negative Predictive Value (NPV)** | The percentage of results that were true negative (TN) among all results that the system flagged as negative. Calc: TN / (TN + FN). |
138
+ | **Recall** | Also called sensitivity. The percentage of results flagged positive among all results that should have been obtained. Calc: TP / (TP + FN). |
139
+ | **Specificity** | The percentage of results that were flagged negative among all negative results. Calc: TN / (TN + FP). |
140
+ | **F1-Score** | The harmonic mean of PPV/precision and sensitivity/recall. Calc: 2 × (Precision × Recall) / (Precision + Recall). Moderately useful in the context of class imbalance. |
141
+ | **Matthews’ Correlation Coefficient (MCC)** | A statistical measure used to evaluate the quality of binary classifications. Unlike other metrics, MCC considers all four categories of a confusion matrix. Calc: (TP × TN − FP × FN) / √((TP + FP)(TP + FN)(TN + FP)(TN + FN)). |
142
+ | **Precision / Recall AUC** | Represents the area under the Precision-Recall curve, which plots Precision against Recall at various threshold settings. It is more resistant to class imbalance than alternatives like AUROC. |
143
+ | **Demographic Parity (DP)** | Demographic Parity, also known as Statistical Parity, requires that the probability of a positive prediction is the same across different demographic groups. Calc: DP = P(Ŷ=1∣A=a) = P(Ŷ=1∣A=b). This figure is given as an absolute difference where positive values suggest the more privileged group gains and negative values the reverse. |
144
+ | **Equal Opportunity (EO)** | Equal Opportunity focuses on equalising the true positive rates across groups. Among those who truly belong to the positive class, the model should predict positive outcomes at equal rates across groups. Calc: EO = P(Ŷ=1∣Y=1, A=a) = P(Ŷ=1∣Y=1, A=b). A higher value indicates a bias against the more vulnerable group. |
145
+ | **Disparate Impact (DI)** | Divides the protected group’s positive prediction rate by that of the most-favoured group. If the ratio is below 0.8 or above 1.25, disparate impact is considered present. Calc: DI = P(Ŷ=1∣A=unfavoured) / P(Ŷ=1∣A=favoured). Values outside 0.8–1.25 range suggest bias. |
146
+ | **Execution Time / Energy / CO₂ Emissions** | Measured in minutes and total energy consumption in kilowatt-hours (kWh), which is then converted to CO₂ emissions using a factor of 0.20705 Kg CO₂e per kWh. |
147
+
148
+ ## Model Card Authors
149
+
150
+ Matt Stammers - Computational Gastroenterologist
151
+
152
+ ## Model Card Contact
153
+
154
+ m.stammers@soton.ac.uk
155
+
156
+ ## Base Training Data for the Model
157
+
158
+ #### Training data
159
+
160
+ We use the concatenation from multiple datasets to fine-tune our model. The total number of sentence pairs is above 1 billion sentences.
161
+ We sampled each dataset given a weighted probability which configuration is detailed in the `data_config.json` file.
162
+
163
+
164
+ | Dataset | Paper | Number of training tuples |
165
+ |--------------------------------------------------------|:----------------------------------------:|:--------------------------:|
166
+ | [Reddit comments (2015-2018)](https://github.com/PolyAI-LDN/conversational-datasets/tree/master/reddit) | [paper](https://arxiv.org/abs/1904.06472) | 726,484,430 |
167
+ | [S2ORC](https://github.com/allenai/s2orc) Citation pairs (Abstracts) | [paper](https://aclanthology.org/2020.acl-main.447/) | 116,288,806 |
168
+ | [WikiAnswers](https://github.com/afader/oqa#wikianswers-corpus) Duplicate question pairs | [paper](https://doi.org/10.1145/2623330.2623677) | 77,427,422 |
169
+ | [PAQ](https://github.com/facebookresearch/PAQ) (Question, Answer) pairs | [paper](https://arxiv.org/abs/2102.07033) | 64,371,441 |
170
+ | [S2ORC](https://github.com/allenai/s2orc) Citation pairs (Titles) | [paper](https://aclanthology.org/2020.acl-main.447/) | 52,603,982 |
171
+ | [S2ORC](https://github.com/allenai/s2orc) (Title, Abstract) | [paper](https://aclanthology.org/2020.acl-main.447/) | 41,769,185 |
172
+ | [Stack Exchange](https://huggingface.co/datasets/flax-sentence-embeddings/stackexchange_xml) (Title, Body) pairs | - | 25,316,456 |
173
+ | [Stack Exchange](https://huggingface.co/datasets/flax-sentence-embeddings/stackexchange_xml) (Title+Body, Answer) pairs | - | 21,396,559 |
174
+ | [Stack Exchange](https://huggingface.co/datasets/flax-sentence-embeddings/stackexchange_xml) (Title, Answer) pairs | - | 21,396,559 |
175
+ | [MS MARCO](https://microsoft.github.io/msmarco/) triplets | [paper](https://doi.org/10.1145/3404835.3462804) | 9,144,553 |
176
+ | [GOOAQ: Open Question Answering with Diverse Answer Types](https://github.com/allenai/gooaq) | [paper](https://arxiv.org/pdf/2104.08727.pdf) | 3,012,496 |
177
+ | [Yahoo Answers](https://www.kaggle.com/soumikrakshit/yahoo-answers-dataset) (Title, Answer) | [paper](https://proceedings.neurips.cc/paper/2015/hash/250cf8b51c773f3f8dc8b4be867a9a02-Abstract.html) | 1,198,260 |
178
+ | [Code Search](https://huggingface.co/datasets/code_search_net) | - | 1,151,414 |
179
+ | [COCO](https://cocodataset.org/#home) Image captions | [paper](https://link.springer.com/chapter/10.1007%2F978-3-319-10602-1_48) | 828,395|
180
+ | [SPECTER](https://github.com/allenai/specter) citation triplets | [paper](https://doi.org/10.18653/v1/2020.acl-main.207) | 684,100 |
181
+ | [Yahoo Answers](https://www.kaggle.com/soumikrakshit/yahoo-answers-dataset) (Question, Answer) | [paper](https://proceedings.neurips.cc/paper/2015/hash/250cf8b51c773f3f8dc8b4be867a9a02-Abstract.html) | 681,164 |
182
+ | [Yahoo Answers](https://www.kaggle.com/soumikrakshit/yahoo-answers-dataset) (Title, Question) | [paper](https://proceedings.neurips.cc/paper/2015/hash/250cf8b51c773f3f8dc8b4be867a9a02-Abstract.html) | 659,896 |
183
+ | [SearchQA](https://huggingface.co/datasets/search_qa) | [paper](https://arxiv.org/abs/1704.05179) | 582,261 |
184
+ | [Eli5](https://huggingface.co/datasets/eli5) | [paper](https://doi.org/10.18653/v1/p19-1346) | 325,475 |
185
+ | [Flickr 30k](https://shannon.cs.illinois.edu/DenotationGraph/) | [paper](https://transacl.org/ojs/index.php/tacl/article/view/229/33) | 317,695 |
186
+ | [Stack Exchange](https://huggingface.co/datasets/flax-sentence-embeddings/stackexchange_xml) Duplicate questions (titles) | | 304,525 |
187
+ | AllNLI ([SNLI](https://nlp.stanford.edu/projects/snli/) and [MultiNLI](https://cims.nyu.edu/~sbowman/multinli/) | [paper SNLI](https://doi.org/10.18653/v1/d15-1075), [paper MultiNLI](https://doi.org/10.18653/v1/n18-1101) | 277,230 |
188
+ | [Stack Exchange](https://huggingface.co/datasets/flax-sentence-embeddings/stackexchange_xml) Duplicate questions (bodies) | | 250,519 |
189
+ | [Stack Exchange](https://huggingface.co/datasets/flax-sentence-embeddings/stackexchange_xml) Duplicate questions (titles+bodies) | | 250,460 |
190
+ | [Sentence Compression](https://github.com/google-research-datasets/sentence-compression) | [paper](https://www.aclweb.org/anthology/D13-1155/) | 180,000 |
191
+ | [Wikihow](https://github.com/pvl/wikihow_pairs_dataset) | [paper](https://arxiv.org/abs/1810.09305) | 128,542 |
192
+ | [Altlex](https://github.com/chridey/altlex/) | [paper](https://aclanthology.org/P16-1135.pdf) | 112,696 |
193
+ | [Quora Question Triplets](https://quoradata.quora.com/First-Quora-Dataset-Release-Question-Pairs) | - | 103,663 |
194
+ | [Simple Wikipedia](https://cs.pomona.edu/~dkauchak/simplification/) | [paper](https://www.aclweb.org/anthology/P11-2117/) | 102,225 |
195
+ | [Natural Questions (NQ)](https://ai.google.com/research/NaturalQuestions) | [paper](https://transacl.org/ojs/index.php/tacl/article/view/1455) | 100,231 |
196
+ | [SQuAD2.0](https://rajpurkar.github.io/SQuAD-explorer/) | [paper](https://aclanthology.org/P18-2124.pdf) | 87,599 |
197
+ | [TriviaQA](https://huggingface.co/datasets/trivia_qa) | - | 73,346 |
198
  | **Total** | | **1,170,060,424** |