Model reference · open weights
fin-mpnet is an open-weight embedding model from mukaj. AxForge deploys and operates it for you on dedicated EU-owned hardware — with the licence handled where one is required.
Available as managed deployment — configured and operated for you on dedicated EU hardware, quoted per deployment.
What it is
| Released by | mukaj |
|---|---|
| Type | Embedding models |
| Task | Embeddings |
| Parameters (lead) | 109M |
| Context | 514 tokens |
| Runs with | sentence-transformers |
| Released | 2024-01-17 |
| Popularity | 6k downloads / month |
| Licence | Unknown |
About
full evaluation not complete
This is a fine-tuned sentence-transformers model: It maps sentences & paragraphs to a 768 dimensional dense vector space and can be used for tasks like clustering or semantic search.
This model aims to be very strong on Financial Document Retrieval Tasks, while trying to maintain as much generalized performance as possible.
| FiQA | SciFact | AmazonReviews | OnlineBankingIntent | ArguAna | |
|---|---|---|---|---|---|
| fin-mpnet-base | 79.91 | 65.40 | 29.12 | 80.25 | 49.11 |
| all-mpnet-base-v2 | 49.96 | 65.57 | 31.92 | 81.86 | 46.52 |
| previous SoTA | 56.59 | - | - | - | - |
v0.1 shows SoTA results on FiQA Test set while other non-financial benchmarks only drop a few small % and improvement in others.
Using this model becomes easy when you have sentence-transformers installed:
pip install -U sentence-transformers
Then you can use the model like this:
from sentence_transformers import SentenceTransformer
sentences = ["This is an example sentence", "Each sentence is converted"]
model = SentenceTransformer('mukaj/fin-mpnet-base')
embeddings = model.encode(sentences)
print(embeddings)
Model was evaluated during training only on the new finance QA examples, as such only financial relevant benchmarks were evaluated on for v0.1 [FiQA-2018, BankingClassification77]
The model currently shows the highest FiQA Retrieval score on the test set, on the MTEB Leaderboard (https://huggingface.co/spaces/mteb/leaderboard)
The model will have likely suffered some performance on other benchmarks, i.e. BankingClassification77 has dropped from 81.6 to 80.25, this will be addressed for v0.2 and full evaluation on all sets will be run.
"sentence-transformers/all-mpnet-base-v2" was fine-tuned on 150k+ financial document QA examples using MNR Loss.
From the published model card. Full card on the HuggingFace links in the sidebar.
Benchmarks
As published on the model card — the maker's own numbers, not measured by AxForge.
| Task | Dataset | Metric | Score |
|---|---|---|---|
| Classification | MTEB AmazonReviewsClassification (en) | accuracy | 29.128 |
| Classification | MTEB AmazonReviewsClassification (en) | f1 | 28.657 |
| Retrieval | MTEB ArguAna | map_at_1 | 24.111 |
| Retrieval | MTEB ArguAna | map_at_10 | 40.083 |
| Retrieval | MTEB ArguAna | map_at_100 | 41.201 |
| Retrieval | MTEB ArguAna | map_at_1000 | 41.215 |
| Retrieval | MTEB ArguAna | map_at_3 | 35.325 |
| Retrieval | MTEB ArguAna | map_at_5 | 37.796 |
| Retrieval | MTEB ArguAna | mrr_at_1 | 25.036 |
| Retrieval | MTEB ArguAna | mrr_at_10 | 40.436 |
| Retrieval | MTEB ArguAna | mrr_at_100 | 41.554 |
| Retrieval | MTEB ArguAna | mrr_at_1000 | 41.568 |
| Retrieval | MTEB ArguAna | mrr_at_3 | 35.645 |
| Retrieval | MTEB ArguAna | mrr_at_5 | 38.141 |
| Retrieval | MTEB ArguAna | ndcg_at_1 | 24.111 |
| Retrieval | MTEB ArguAna | ndcg_at_10 | 49.112 |
| Retrieval | MTEB ArguAna | ndcg_at_100 | 53.670 |
| Retrieval | MTEB ArguAna | ndcg_at_1000 | 53.944 |
| Retrieval | MTEB ArguAna | ndcg_at_3 | 39.035 |
| Retrieval | MTEB ArguAna | ndcg_at_5 | 43.503 |
| Retrieval | MTEB ArguAna | precision_at_1 | 24.111 |
| Retrieval | MTEB ArguAna | precision_at_10 | 7.817 |
| Retrieval | MTEB ArguAna | precision_at_100 | 0.976 |
| Retrieval | MTEB ArguAna | precision_at_1000 | 0.100 |
Using it via the API
Once AxForge deploys fin-mpnet for you, it answers on the OpenAI-compatible API — the same base URL and keys as every other model. (fin-mpnet below is illustrative; you get the exact model name on deployment.)
$ curl -sS https://api.axforge.ai/v1/embeddings \
-H "Authorization: Bearer $AXFORGE_API_KEY" \
-H "Content-Type: application/json" \
-d '{"model":"fin-mpnet","input":"text to embed"}'
Create an account — your API key is available in the console. 3M free tokens every 30 days with every new account.