CoolFace
Modelpublic

vabatista/sbert-mpnet-base-bm25-hard-neg

sourceHugging Faceapache-2.0updated 3y agoView on Hugging Face
0likes73downloads
Model Card

This similarity model was trained for 2 epochs based on the sentence-transformers/all-mpnet-base-v2. We used the SQuAD 2.0 dataset to compare similarity between questions and sentences containing the answer to the question. We employed the MultipleNegativesRankingLoss as the objective function. To generate negative examples, our strategy involved using BM25 to retrieve similar examples from all sentences in the dataset that did not contain the answer.

As a result, we improved the retrieval of correct sentences (containing the answer) for a question, as measured by Mean Reciprocal Rank (MRR@10) metric. We tested on 7 different SQuAD like datasets.

ModelDropBioASQHotpotNewsTextbookNQTrivia
Baseline Dense0.370.120.280.290.170.290.47
Baseline BM250.360.150.340.330.180.180.47
Azure Hybrid Semantic0.350.180.29
This model0.390.160.340.360.190.280.49