Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks
Paper • 1908.10084 • Published • 13
How to use cuadron11/jina-reranker-v2-base-multilingual-1negative with sentence-transformers:
from sentence_transformers import CrossEncoder
model = CrossEncoder("cuadron11/jina-reranker-v2-base-multilingual-1negative", trust_remote_code=True)
query = "Which planet is known as the Red Planet?"
passages = [
"Venus is often called Earth's twin because of its similar size and proximity.",
"Mars, known for its reddish appearance, is often referred to as the Red Planet.",
"Jupiter, the largest planet in our solar system, has a prominent red spot.",
"Saturn, famous for its rings, is sometimes mistaken for the Red Planet."
]
scores = model.predict([(query, passage) for passage in passages])
print(scores)This is a Cross Encoder model finetuned from jinaai/jina-reranker-v2-base-multilingual using the sentence-transformers library. It computes scores for pairs of texts, which can be used for text reranking and semantic search.
First install the Sentence Transformers library:
pip install -U sentence-transformers
Then you can load this model and run inference.
from sentence_transformers import CrossEncoder
# Download from the 🤗 Hub
model = CrossEncoder("cuadron11/jina-reranker-v2-base-multilingual-1negative")
# Get scores for pairs of texts
pairs = [
['Zein da Euskal Hezkuntza Sistemaren etorkizunari buruzko hitzarmenaren oinarriak zehazteko ponentziaren helburua?', '[TOPIC: Txostena, etorkizuneko euskal hezkuntza-sistemari buruzko hitzarmenaren oinarriak zehazteko ponentziak egindakoa. Eztabaida eta behin betiko ebazpena]\n[GORROTXATEGI AZURMENDI, (EP-IU)]:\negitea. Dena dela, esan bezala, lehenik, etorkizunari begira jarri behar dugu gaur. Etorkizuna da orain kontuan hartu behar duguna; eta etorkizunean, halaber, eskerrak eman nahiko nizkizueke, fio naizelako arestian aipatu ditudan eragile guztiak, gainerako alderdi politikoak ere, bere interes propioak albo batera utzi eta interes orokorraren alde egiteko gai izango direla. Baita sailari ere, erdietsi ditugun akordioak, bestelako presiorik gabe, lantzeko gai eta arduradun izango delakoan. Betiere gizarte-mugimenduen eta (Date: 07.04.2022)'],
['Zein da Martínez Hierroren iritzia errelato partekatu bat eraikitzeari buruz Euskal Herrian?', '[TOPIC: Lehenbailehen eztabaidatzeko Euzko Abertzaleak eta Euskal Sozialistak talde parlamentarioek egindako legez besteko proposamena, ETAren armagabetzeari buruz. Eztabaida eta behin betiko ebazpena]\n[MARTÍNEZ HIERRO, (EP)]:\naurrera, egia da etorkizunera begiratu behar dela iragana ahaztu gabe, biolentzia deslegitimatuz, eta, baita ere, errelato bat ireki nahi badugu, errelato partekatu bat eraikitzea, eta errelato partekatu hori izatea oinarri bezala biolentzia honen deslegitimazioa. Eta lehen esan dudan bezala, ez bakarrik biolentzia zuzena edo armek sortu duten biolentzia, baizik eta, baita ere, oraindik gizartean badaukagun hainbat biolentzia forma ezberdin, biolentzia kulturala eta herri honetan oraindik instalatuta dauden hainbat gauza. (Date: 06.04.2017)'],
['Zein da Ertzaintzaren hasierako prestakuntzaren iraupena polizia-prozedurei eta autobabesari dagokionez?', '[TOPIC: Galdera, Miren Gallastegui Oyarzábal Euskal Sozialistak taldeko legebiltzarkideak Segurtasuneko sailburuari egina, Ertzaintzaren aurkako agresioak ekiditeko Segurtasun Sailak hartu beharreko neurriei buruz]\n[SEGURTASUNEKO SAILBURUAK (BELTRÁN DE HEREDIA ARRONIZ), (EA-NV)]:\nEskerrik asko, presidente andrea. Legebiltzarkideok, egun on berriro ere. Gallastegui andrea, segurtasun-neurriei buruz agerikoak diren gauza batzuk gogorarazi behar dizkizut: adibidez, ertzain guztiek hasieran prestakuntza handia jasotzen dutela polizia-prozedurei eta autobabesari buruz. Bederatzi hilabete akademian eta beste hainbeste polizia-unitateetan praktikak egiten. Hasierako prestakuntza hori sakondu eta zabaldu egiten da gero, ertzainen karreran zehar, eguneratze-ikastaroekin, birziklatze-ikastaroekin, espezializazio-ikastaroekin... Baina, era berean, garrantzitsuak dira oso ekintza (Date: 13.03.2015)'],
['Noiz izan zen kutxen legearen erreforma indarrean jarri zena?', "[TOPIC: Legegintzako herri-ekimeneko lege-proposamena, hidrokarburo ez-konbentzionalak erauzteko prozesuan eta haustura hidrauliko edo 'fracking'ean ingurumena babesteko neurri osagarriei buruz. Aintzat hartzeari buruzko eztabaida eta behin betiko ebazpena]\n[DAMBORENEA BASTERRECHEA, (PV-ETP)]:\ngara arazorik izan ez duten bakarrak, besteak beste bateratzeak arazo batzuk estali zituelako, eta nik uste hori ere guztiok dakigula. Esan duzu Arabako ahaldun nagusia akzioak sortzaileei bueltatzearen aldekoa dela... Tira, bat nator erabat Arabako ahaldun nagusiarekin. Are gehiago, jakin dezazun esango dizut joan den legealdian kutxen legearen erreforma, oraindik indarrean dagoena, hona ekarri zenean gure taldeak esan zuela fundazioak arautu behar zirela eta, jakina, erakunde sortzaileek han egon behar (Date: 04.12.2014)"],
['Zein da EAEko Auzitegi Nagusiaren ebazpena Osakidetzako aldi baterako kontratazio-zerrendak kudeatzeko eta egiteko irizpideak aldatzeari buruz?', '[TOPIC: Galdera, Laura Garrido Knörr Euskal Talde Popularreko legebiltzarkideak Osasuneko sailburuari egina, Osakidetzako aldi baterako kontratazio-zerrendak kudeatzeko eta egiteko irizpideak aldatzeari buruz]\n[BLANCO DE ANGULO, (EH Bildu)]:\nEskerrik asko. Arrazoi izango duzu agian, eta instrukzioak nahasi ditut. Ez dugu hitz egin ez zuk ez nik, eta agertu da, EAEko Auzitegi Nagusiaren ebazpenaz, nik aipatzen nuenaz. Egia da gogoko duzuela zuek agindua, baina guri ez zaigu gustatzen, ez zaigu inoiz gustatu, eta beti esan dugu. Ez zaigu gustatzen agindu hori, ezta 114 Dekretua ere, guk gai berarekin jarraitzen baitugu, eta da iruditzen zaigula urtebetez erroldatuta ez (Date: 21.02.2014)'],
]
scores = model.predict(pairs)
print(scores.shape)
# (5,)
# Or rank different texts based on similarity to a single text
ranks = model.rank(
'Zein da Euskal Hezkuntza Sistemaren etorkizunari buruzko hitzarmenaren oinarriak zehazteko ponentziaren helburua?',
[
'[TOPIC: Txostena, etorkizuneko euskal hezkuntza-sistemari buruzko hitzarmenaren oinarriak zehazteko ponentziak egindakoa. Eztabaida eta behin betiko ebazpena]\n[GORROTXATEGI AZURMENDI, (EP-IU)]:\negitea. Dena dela, esan bezala, lehenik, etorkizunari begira jarri behar dugu gaur. Etorkizuna da orain kontuan hartu behar duguna; eta etorkizunean, halaber, eskerrak eman nahiko nizkizueke, fio naizelako arestian aipatu ditudan eragile guztiak, gainerako alderdi politikoak ere, bere interes propioak albo batera utzi eta interes orokorraren alde egiteko gai izango direla. Baita sailari ere, erdietsi ditugun akordioak, bestelako presiorik gabe, lantzeko gai eta arduradun izango delakoan. Betiere gizarte-mugimenduen eta (Date: 07.04.2022)',
'[TOPIC: Lehenbailehen eztabaidatzeko Euzko Abertzaleak eta Euskal Sozialistak talde parlamentarioek egindako legez besteko proposamena, ETAren armagabetzeari buruz. Eztabaida eta behin betiko ebazpena]\n[MARTÍNEZ HIERRO, (EP)]:\naurrera, egia da etorkizunera begiratu behar dela iragana ahaztu gabe, biolentzia deslegitimatuz, eta, baita ere, errelato bat ireki nahi badugu, errelato partekatu bat eraikitzea, eta errelato partekatu hori izatea oinarri bezala biolentzia honen deslegitimazioa. Eta lehen esan dudan bezala, ez bakarrik biolentzia zuzena edo armek sortu duten biolentzia, baizik eta, baita ere, oraindik gizartean badaukagun hainbat biolentzia forma ezberdin, biolentzia kulturala eta herri honetan oraindik instalatuta dauden hainbat gauza. (Date: 06.04.2017)',
'[TOPIC: Galdera, Miren Gallastegui Oyarzábal Euskal Sozialistak taldeko legebiltzarkideak Segurtasuneko sailburuari egina, Ertzaintzaren aurkako agresioak ekiditeko Segurtasun Sailak hartu beharreko neurriei buruz]\n[SEGURTASUNEKO SAILBURUAK (BELTRÁN DE HEREDIA ARRONIZ), (EA-NV)]:\nEskerrik asko, presidente andrea. Legebiltzarkideok, egun on berriro ere. Gallastegui andrea, segurtasun-neurriei buruz agerikoak diren gauza batzuk gogorarazi behar dizkizut: adibidez, ertzain guztiek hasieran prestakuntza handia jasotzen dutela polizia-prozedurei eta autobabesari buruz. Bederatzi hilabete akademian eta beste hainbeste polizia-unitateetan praktikak egiten. Hasierako prestakuntza hori sakondu eta zabaldu egiten da gero, ertzainen karreran zehar, eguneratze-ikastaroekin, birziklatze-ikastaroekin, espezializazio-ikastaroekin... Baina, era berean, garrantzitsuak dira oso ekintza (Date: 13.03.2015)',
"[TOPIC: Legegintzako herri-ekimeneko lege-proposamena, hidrokarburo ez-konbentzionalak erauzteko prozesuan eta haustura hidrauliko edo 'fracking'ean ingurumena babesteko neurri osagarriei buruz. Aintzat hartzeari buruzko eztabaida eta behin betiko ebazpena]\n[DAMBORENEA BASTERRECHEA, (PV-ETP)]:\ngara arazorik izan ez duten bakarrak, besteak beste bateratzeak arazo batzuk estali zituelako, eta nik uste hori ere guztiok dakigula. Esan duzu Arabako ahaldun nagusia akzioak sortzaileei bueltatzearen aldekoa dela... Tira, bat nator erabat Arabako ahaldun nagusiarekin. Are gehiago, jakin dezazun esango dizut joan den legealdian kutxen legearen erreforma, oraindik indarrean dagoena, hona ekarri zenean gure taldeak esan zuela fundazioak arautu behar zirela eta, jakina, erakunde sortzaileek han egon behar (Date: 04.12.2014)",
'[TOPIC: Galdera, Laura Garrido Knörr Euskal Talde Popularreko legebiltzarkideak Osasuneko sailburuari egina, Osakidetzako aldi baterako kontratazio-zerrendak kudeatzeko eta egiteko irizpideak aldatzeari buruz]\n[BLANCO DE ANGULO, (EH Bildu)]:\nEskerrik asko. Arrazoi izango duzu agian, eta instrukzioak nahasi ditut. Ez dugu hitz egin ez zuk ez nik, eta agertu da, EAEko Auzitegi Nagusiaren ebazpenaz, nik aipatzen nuenaz. Egia da gogoko duzuela zuek agindua, baina guri ez zaigu gustatzen, ez zaigu inoiz gustatu, eta beti esan dugu. Ez zaigu gustatzen agindu hori, ezta 114 Dekretua ere, guk gai berarekin jarraitzen baitugu, eta da iruditzen zaigula urtebetez erroldatuta ez (Date: 21.02.2014)',
]
)
# [{'corpus_id': ..., 'score': ...}, {'corpus_id': ..., 'score': ...}, ...]
jina-reranker-v2-base-multilingual-1negativeCrossEncoderRerankingEvaluator with these parameters:{
"at_k": 10
}
| Metric | Value |
|---|---|
| map | 0.9999 |
| mrr@10 | 0.9999 |
| ndcg@10 | 1.0 |
query, positive, and negative| query | positive | negative | |
|---|---|---|---|
| type | string | string | string |
| details |
|
|
|
| query | positive | negative |
|---|---|---|
Zergatik ez zuen alde bozkatu EH Bilduk Lanbiden aztarna digitala ezartzeari buruzko proposamenean? |
[TOPIC: Euskal Legebiltzar Talde Popularrak egindako legez besteko proposamena, Lanbiden aztarna digitala ezartzeari buruz. Eztabaida eta behin betiko ebazpena] |
[TOPIC: Galdera, Amaia Martinez Grisaleña, Talde Mistoko legebiltzarkideak lehendakariari egina, ETAren kasu argitu gabeak direla eta, EBren ordezkaritzaren Gasteizko bisitari buruz] |
Zein urtetan baztertu zuen Urkullu lehendakariak aurkeztutako aurrekontuen lege-proiektua, babes faltagatik? |
[TOPIC: Galdera, Susana Corcuera Leunda Euskal Sozialistak taldeko legebiltzarkideak Ogasun eta Finantzetako sailburuari egina, aurrekontu-luzapenari buruzko legezkotasuna 2013an ez betetzeari buruz] |
Bazterketa horren ondorioz, bi lege-proiektu bidali dira; ikusiko dugu zer gertatzen den haiekin. Enpleguaren legea. |
Zein da Euskal Autonomia Erkidegoko Kultura Ondarearen Kontseiluaren osaera, antolamendua, funtzionamendua eta zereginen inguruan Eusko Jaurlaritzak dituen irizpideak? |
[TOPIC: Mozioa, Carmelo Barrio Baroja Euskal Talde Popularreko legebiltzarkideak aurkeztua, Eusko Jaurlaritzak Kultura Ondarearen Euskal Autonomia Erkidegoko Kontseiluaren osaerari, antolamenduari, funtzionamenduari eta zereginei dagokienez dituen irizpideen inguruan. Eztabaida eta behin betiko ebazpena] |
Euskal Kultura Ondarearen Erregistroa erregulatzen ari gara. Oinarrizko Babesa duten Kultura Ondasunen Erregistroa erregulatzen ari gara. |
CachedMultipleNegativesRankingLoss with these parameters:{
"scale": 10.0,
"num_negatives": 1,
"activation_fn": "torch.nn.modules.activation.Sigmoid",
"mini_batch_size": 32
}
query, positive, and negative| query | positive | negative | |
|---|---|---|---|
| type | string | string | string |
| details |
|
|
|
| query | positive | negative |
|---|---|---|
Zein da Euskal Hezkuntza Sistemaren etorkizunari buruzko hitzarmenaren oinarriak zehazteko ponentziaren helburua? |
[TOPIC: Txostena, etorkizuneko euskal hezkuntza-sistemari buruzko hitzarmenaren oinarriak zehazteko ponentziak egindakoa. Eztabaida eta behin betiko ebazpena] |
Honezkero ia urte eta erdia pasatu da hezkuntza-sistemaren oinarriak ezartzeko ponentzia eratu zela. |
Zein da Martínez Hierroren iritzia errelato partekatu bat eraikitzeari buruz Euskal Herrian? |
[TOPIC: Lehenbailehen eztabaidatzeko Euzko Abertzaleak eta Euskal Sozialistak talde parlamentarioek egindako legez besteko proposamena, ETAren armagabetzeari buruz. Eztabaida eta behin betiko ebazpena] |
Badakit, Martínez jauna, gogorra da. Baina aitor ezazu, hausnar ezazue. Aukera duzue Legebiltzar honetan egiteko, baina ez zaitez etorri argudio-zerrendarekin halakoari buruz... |
Zein da Ertzaintzaren hasierako prestakuntzaren iraupena polizia-prozedurei eta autobabesari dagokionez? |
[TOPIC: Galdera, Miren Gallastegui Oyarzábal Euskal Sozialistak taldeko legebiltzarkideak Segurtasuneko sailburuari egina, Ertzaintzaren aurkako agresioak ekiditeko Segurtasun Sailak hartu beharreko neurriei buruz] |
Gainera, prestakuntza berria jasotzen ari dira autobabeseko polizia-taktika eta -tekniketan, kriminalitate eta terrorismo mota berri hauei aurre egiteko polizia-taktika eta -tekniketan. |
CachedMultipleNegativesRankingLoss with these parameters:{
"scale": 10.0,
"num_negatives": 1,
"activation_fn": "torch.nn.modules.activation.Sigmoid",
"mini_batch_size": 32
}
eval_strategy: stepsper_device_train_batch_size: 32per_device_eval_batch_size: 32learning_rate: 2e-05warmup_ratio: 0.1load_best_model_at_end: Truebatch_sampler: no_duplicatesoverwrite_output_dir: Falsedo_predict: Falseeval_strategy: stepsprediction_loss_only: Trueper_device_train_batch_size: 32per_device_eval_batch_size: 32per_gpu_train_batch_size: Noneper_gpu_eval_batch_size: Nonegradient_accumulation_steps: 1eval_accumulation_steps: Nonetorch_empty_cache_steps: Nonelearning_rate: 2e-05weight_decay: 0.0adam_beta1: 0.9adam_beta2: 0.999adam_epsilon: 1e-08max_grad_norm: 1.0num_train_epochs: 3max_steps: -1lr_scheduler_type: linearlr_scheduler_kwargs: {}warmup_ratio: 0.1warmup_steps: 0log_level: passivelog_level_replica: warninglog_on_each_node: Truelogging_nan_inf_filter: Truesave_safetensors: Truesave_on_each_node: Falsesave_only_model: Falserestore_callback_states_from_checkpoint: Falseno_cuda: Falseuse_cpu: Falseuse_mps_device: Falseseed: 42data_seed: Nonejit_mode_eval: Falseuse_ipex: Falsebf16: Falsefp16: Falsefp16_opt_level: O1half_precision_backend: autobf16_full_eval: Falsefp16_full_eval: Falsetf32: Nonelocal_rank: 0ddp_backend: Nonetpu_num_cores: Nonetpu_metrics_debug: Falsedebug: []dataloader_drop_last: Falsedataloader_num_workers: 0dataloader_prefetch_factor: Nonepast_index: -1disable_tqdm: Falseremove_unused_columns: Truelabel_names: Noneload_best_model_at_end: Trueignore_data_skip: Falsefsdp: []fsdp_min_num_params: 0fsdp_config: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}fsdp_transformer_layer_cls_to_wrap: Noneaccelerator_config: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}deepspeed: Nonelabel_smoothing_factor: 0.0optim: adamw_torchoptim_args: Noneadafactor: Falsegroup_by_length: Falselength_column_name: lengthddp_find_unused_parameters: Noneddp_bucket_cap_mb: Noneddp_broadcast_buffers: Falsedataloader_pin_memory: Truedataloader_persistent_workers: Falseskip_memory_metrics: Trueuse_legacy_prediction_loop: Falsepush_to_hub: Falseresume_from_checkpoint: Nonehub_model_id: Nonehub_strategy: every_savehub_private_repo: Nonehub_always_push: Falsegradient_checkpointing: Falsegradient_checkpointing_kwargs: Noneinclude_inputs_for_metrics: Falseinclude_for_metrics: []eval_do_concat_batches: Truefp16_backend: autopush_to_hub_model_id: Nonepush_to_hub_organization: Nonemp_parameters: auto_find_batch_size: Falsefull_determinism: Falsetorchdynamo: Noneray_scope: lastddp_timeout: 1800torch_compile: Falsetorch_compile_backend: Nonetorch_compile_mode: Noneinclude_tokens_per_second: Falseinclude_num_input_tokens_seen: Falseneftune_noise_alpha: Noneoptim_target_modules: Nonebatch_eval_metrics: Falseeval_on_start: Falseuse_liger_kernel: Falseeval_use_gather_object: Falseaverage_tokens_across_devices: Falseprompts: Nonebatch_sampler: no_duplicatesmulti_dataset_batch_sampler: proportionalrouter_mapping: {}learning_rate_mapping: {}| Epoch | Step | Training Loss | Validation Loss | jina-reranker-v2-base-multilingual-1negative_ndcg@10 |
|---|---|---|---|---|
| 0.4591 | 500 | 0.0301 | 0.0030 | 0.9999 |
| 0.9183 | 1000 | 0.0032 | 0.0029 | 1.0000 |
| 1.3774 | 1500 | 0.0027 | 0.0029 | 1.0000 |
| 1.8365 | 2000 | 0.0025 | 0.0022 | 1.0 |
| 2.2957 | 2500 | 0.0021 | 0.0027 | 1.0000 |
| 2.7548 | 3000 | 0.0019 | 0.0026 | 1.0000 |
@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",
}
Base model
jinaai/jina-reranker-v2-base-multilingual