kdhole/reranker-msmarco-v1.1-ettin-encoder-1b-ranknet

CrossEncoder based on jhu-clsp/ettin-encoder-1b

This is a Cross Encoder model finetuned from jhu-clsp/ettin-encoder-1b on the ms_marco dataset using the sentence-transformers library. It computes scores for pairs of texts, which can be used for text reranking and semantic search.

Model Details

Model Description

• Model Type: Cross Encoder
• Base model: jhu-clsp/ettin-encoder-1b
• Maximum Sequence Length: 7999 tokens
• Number of Output Labels: 1 label
• Training Dataset:
  • ms_marco
• Language: en

Model Sources

• Documentation: Sentence Transformers Documentation
• Documentation: Cross Encoder Documentation
• Repository: Sentence Transformers on GitHub
• Hugging Face: Cross Encoders on Hugging Face

Usage

Direct Usage (Sentence Transformers)

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("kdhole/reranker-msmarco-v1.1-ettin-encoder-1b-ranknet")
# Get scores for pairs of texts
pairs = [
    ['define monogenic trait', 'An allele is a version of a gene. For example, in fruitflies there is a gene which determines eye colour: one allele gives red eyes, and another gives white eyes; it is the same *gene*, just different versions of that gene. A monogenic trait is one which is encoded by a single gene. e.g. - cystic fibrosis in humans. There is a single gene which determines this trait: the wild-type allele is healthy, while the disease allele gives you cystic fibrosis'],
    ['define monogenic trait', 'Abstract. Monogenic inheritance refers to genetic control of a phenotype or trait by a single gene. For a monogenic trait, mutations in one (dominant) or both (recessive) copies of the gene are sufficient for the trait to be expressed. Digenic inheritance refers to mutation on two genes interacting to cause a genetic phenotype or disease. Triallelic inheritance is a special case of digenic inheritance that requires homozygous mutations at one locus and heterozygous mutations at a second locus to express a phenotype.'],
    ['define monogenic trait', 'A trait that is controlled by a group of nonallelic genes. Supplement. Polygenic traits are controlled by two or more than two genes (usually by many different genes) at different loci on different chromosomes. These genes are described as polygenes.'],
    ['define monogenic trait', "Monogenic Disorders (Single Abnormal Gene). Monogenic autosomal dominant disorders occur through the inheritance of a single copy of a defective gene. These disorders are the result of a single defective gene on the autosomes. They are inherited according to Mendel's Laws (Mendelian disorders). The mutation can be spontaneous and where there is no previous family history. Inheritance patterns can be autosomal dominant, autosomal recessive or X-linked recessive."],
    ['define monogenic trait', 'Adj. 1. monogenic-of or relating to an inheritable character that is controlled by a single pair of genes. genetic science, genetics-the branch of biology that studies heredity and variation in organisms. heritable, inheritable-capable of being inherited; inheritable traits such as eye color; an inheritable title. monogenic. adj. 1. (Genetics) genetics of or relating to an inherited character difference that is controlled by a single gene. 2. (Biology) (of animals) producing offspring of one sex. (ˌmɒn əˈdʒɛn ɪk).'],
]
scores = model.predict(pairs)
print(scores.shape)
# (5,)

# Or rank different texts based on similarity to a single text
ranks = model.rank(
    'define monogenic trait',
    [
        'An allele is a version of a gene. For example, in fruitflies there is a gene which determines eye colour: one allele gives red eyes, and another gives white eyes; it is the same *gene*, just different versions of that gene. A monogenic trait is one which is encoded by a single gene. e.g. - cystic fibrosis in humans. There is a single gene which determines this trait: the wild-type allele is healthy, while the disease allele gives you cystic fibrosis',
        'Abstract. Monogenic inheritance refers to genetic control of a phenotype or trait by a single gene. For a monogenic trait, mutations in one (dominant) or both (recessive) copies of the gene are sufficient for the trait to be expressed. Digenic inheritance refers to mutation on two genes interacting to cause a genetic phenotype or disease. Triallelic inheritance is a special case of digenic inheritance that requires homozygous mutations at one locus and heterozygous mutations at a second locus to express a phenotype.',
        'A trait that is controlled by a group of nonallelic genes. Supplement. Polygenic traits are controlled by two or more than two genes (usually by many different genes) at different loci on different chromosomes. These genes are described as polygenes.',
        "Monogenic Disorders (Single Abnormal Gene). Monogenic autosomal dominant disorders occur through the inheritance of a single copy of a defective gene. These disorders are the result of a single defective gene on the autosomes. They are inherited according to Mendel's Laws (Mendelian disorders). The mutation can be spontaneous and where there is no previous family history. Inheritance patterns can be autosomal dominant, autosomal recessive or X-linked recessive.",
        'Adj. 1. monogenic-of or relating to an inheritable character that is controlled by a single pair of genes. genetic science, genetics-the branch of biology that studies heredity and variation in organisms. heritable, inheritable-capable of being inherited; inheritable traits such as eye color; an inheritable title. monogenic. adj. 1. (Genetics) genetics of or relating to an inherited character difference that is controlled by a single gene. 2. (Biology) (of animals) producing offspring of one sex. (ˌmɒn əˈdʒɛn ɪk).',
    ]
)
# [{'corpus_id': ..., 'score': ...}, {'corpus_id': ..., 'score': ...}, ...]

Evaluation

Metrics

Cross Encoder Reranking

• Datasets: NanoMSMARCO_R100, NanoNFCorpus_R100 and NanoNQ_R100
• Evaluated with CrossEncoderRerankingEvaluator with these parameters:

  {
      "at_k": 10,
      "always_rerank_positives": true
  }
  

Metric	NanoMSMARCO_R100	NanoNFCorpus_R100	NanoNQ_R100
map	0.5930 (+0.1034)	0.3981 (+0.1371)	0.6555 (+0.2359)
mrr@10	0.5839 (+0.1064)	0.5919 (+0.0921)	0.6748 (+0.2481)
ndcg@10	0.6416 (+0.1012)	0.4444 (+0.1194)	0.7099 (+0.2092)

Cross Encoder Nano BEIR

• Dataset: NanoBEIR_R100_mean
• Evaluated with CrossEncoderNanoBEIREvaluator with these parameters:

  {
      "dataset_names": [
          "msmarco",
          "nfcorpus",
          "nq"
      ],
      "rerank_k": 100,
      "at_k": 10,
      "always_rerank_positives": true
  }
  

Metric	Value
map	0.5488 (+0.1588)
mrr@10	0.6169 (+0.1489)
ndcg@10	0.5986 (+0.1433)

Training Details

Training Dataset

ms_marco

• Dataset: ms_marco at a47ee7a
• Size: 78,704 training samples
• Columns: query, docs, and labels
• Approximate statistics based on the first 1000 samples:

  	query	docs	labels
  type	string	list	list
  details	min: 11 characters mean: 32.93 characters max: 95 characters	min: 3 elements mean: 6.50 elements max: 10 elements	min: 3 elements mean: 6.50 elements max: 10 elements

• Samples:

  query	docs	labels
  what does vegan mean	['A vegan, a person who practices veganism, is an individual who actively avoids the use of animal products for food, clothing or any other purpose. As with many diets and lifestyles, not all vegans approach animal product avoidance in the same ways. For example, some vegans completely avoid all animal by-products, while others consider it acceptable to use honey, silk, and other by-products produced from insects.', 'Fruitarian: Eats only raw fruit, including raw nuts and seeds. Vegan. Does not eat dairy products, eggs, or any other animal product. So in a nutshell, a vegetarian diet excludes flesh, but includes other animal products: A vegan diet is one that excludes all animal products. And I have to say that I have met very few vegans who stop with what they put in their mouths. ', 'Animal Ingredients and Their Alternatives. Adopting a vegan diet means saying “no” to cruelty to animals and environmental destruction and “yes” to compassion and good health. It also means paying attent...	[1, 0, 0, 0, 0, ...]
  difference between viral and bacterial conjunctivitis symptoms	["Viral and bacterial conjunctivitis. Viral conjunctivitis and bacterial conjunctivitis may affect one or both eyes. Viral conjunctivitis usually produces a watery discharge. Bacterial conjunctivitis often produces a thicker, yellow-green discharge. Both types can be associated with colds or symptoms of a respiratory infection, such as a sore throat. Both viral and bacterial types are very contagious. They are spread through direct or indirect contact with the eye secretions of someone who's infected", 'A Honor Society of Nursing (STTI) answered. Viral and bacterial conjunctivitis are similar, but differ in several key ways. First, bacterial conjunctivitis can be cured with antibiotics, while the viral form cannot. Second, there is a slight variation in symptoms. With viral conjunctivitis, the discharge from the eye is clearer and less thick than with the bacterial infection. Viral conjunctivitis can also cause painful swelling in the lymph node nearest the ear, a symptom not experienc...	[1, 0, 0, 0, 0, ...]
  can single member llc be taxed as s corp	['A single-member limited liability company, as a solely owned LLC is called, gives the owner a choice of how to be taxed -- as a sole proprietorship, an S corporation or a C corporation. The legal structure of the business itself doesn’t change with any of the choices. Under an S corporation classification, a single-member LLC needs to have a large enough profit in excess of the owner’s salary to realize any tax savings on passive income.', 'An S corp may own up to 100 percent of an LLC, or limited liability company. While all but single-member LLCs cannot be shareholders in S corporations, the reverse -- an S corporation owning an LLC -- is legal. The similarity of tax treatment for S corps and LLCs eliminates most of the common concerns about IRS issues. There is, however, one way for an LLC to own stock in an S corp. A single member LLC, taxed as a sole proprietorship, is called a disregarded entity by the IRS. Treated like an unincorporated individual, this LLC could own stock in ...	[1, 0, 0, 0, 0, ...]

• Loss: RankNetLoss with these parameters:

  {
      "k": null,
      "sigma": 1.0,
      "eps": 1e-10,
      "reduction_log": "binary",
      "activation_fn": "torch.nn.modules.linear.Identity",
      "mini_batch_size": 16
  }
  

Evaluation Dataset

ms_marco

• Dataset: ms_marco at a47ee7a
• Size: 1,000 evaluation samples
• Columns: query, docs, and labels
• Approximate statistics based on the first 1000 samples:

  	query	docs	labels
  type	string	list	list
  details	min: 11 characters mean: 33.63 characters max: 99 characters	min: 3 elements mean: 6.50 elements max: 10 elements	min: 3 elements mean: 6.50 elements max: 10 elements

• Samples:

  query	docs	labels
  define monogenic trait	['An allele is a version of a gene. For example, in fruitflies there is a gene which determines eye colour: one allele gives red eyes, and another gives white eyes; it is the same gene, just different versions of that gene. A monogenic trait is one which is encoded by a single gene. e.g. - cystic fibrosis in humans. There is a single gene which determines this trait: the wild-type allele is healthy, while the disease allele gives you cystic fibrosis', 'Abstract. Monogenic inheritance refers to genetic control of a phenotype or trait by a single gene. For a monogenic trait, mutations in one (dominant) or both (recessive) copies of the gene are sufficient for the trait to be expressed. Digenic inheritance refers to mutation on two genes interacting to cause a genetic phenotype or disease. Triallelic inheritance is a special case of digenic inheritance that requires homozygous mutations at one locus and heterozygous mutations at a second locus to express a phenotype.', 'A trait that is ...	[1, 1, 0, 0, 0, ...]
  behavioral theory definition	["Not to be confused with Behavioralism. Behaviorism (or behaviourism) is an approach to psychology that focuses on an individual's behavior. It combines elements of philosophy, methodology, and psychological theory", 'The initial assumption is that behavior can be explained and further described using behavioral theories. For instance, John Watson and B.F. Skinner advocate the theory that behavior can be acquired through conditioning. Also known as general behavior theory. BEHAVIOR THEORY: Each behavioral theory is an advantage to learning, because it provides teachers with a new and different approach.. No related posts. ', 'behaviorism. noun be·hav·ior·ism. : a school of psychology that takes the objective evidence of behavior (as measured responses to stimuli) as the only concern of its research and the only basis of its theory without reference to conscious experience—compare cognitive psychology. : a school of psychology that takes the objective evidence of behavior (as measured ...	[1, 0, 0, 0, 0, ...]
  What is a disease that is pleiotropic?	['Unsourced material may be challenged and removed. (September 2013). Pleiotropy occurs when one gene influences two or more seemingly unrelated phenotypic traits, an example being phenylketonuria, which is a human disease that affects multiple systems but is caused by one gene defect. Consequently, a mutation in a pleiotropic gene may have an effect on some or all traits simultaneously. The underlying mechanism is that the gene codes for a product that is, for example, used by various cells, or has a signaling function on various targets. A classic example of pleiotropy is the human disease phenylketonuria (PKU).', 'Pleiotropic, autosomal dominant disorder affecting connective tissue: Related Diseases. Pleiotropic, autosomal dominant disorder affecting connective tissue: Pleiotropic, autosomal dominant disorder affecting connective tissue is listed as a type of (or associated with) the following medical conditions in our database: 1 Heart conditions. Office of Rare Diseases (ORD) of ...	[1, 0, 0, 0, 0, ...]

• Loss: RankNetLoss with these parameters:

  {
      "k": null,
      "sigma": 1.0,
      "eps": 1e-10,
      "reduction_log": "binary",
      "activation_fn": "torch.nn.modules.linear.Identity",
      "mini_batch_size": 16
  }
  

Training Hyperparameters

Non-Default Hyperparameters

• eval_strategy: steps
• per_device_train_batch_size: 16
• per_device_eval_batch_size: 16
• learning_rate: 2e-05
• num_train_epochs: 1
• warmup_ratio: 0.1
• seed: 12
• bf16: True
• load_best_model_at_end: True

All Hyperparameters

Click to expand

• overwrite_output_dir: False
• do_predict: False
• eval_strategy: steps
• prediction_loss_only: True
• per_device_train_batch_size: 16
• per_device_eval_batch_size: 16
• per_gpu_train_batch_size: None
• per_gpu_eval_batch_size: None
• gradient_accumulation_steps: 1
• eval_accumulation_steps: None
• torch_empty_cache_steps: None
• learning_rate: 2e-05
• weight_decay: 0.0
• adam_beta1: 0.9
• adam_beta2: 0.999
• adam_epsilon: 1e-08
• max_grad_norm: 1.0
• num_train_epochs: 1
• max_steps: -1
• lr_scheduler_type: linear
• lr_scheduler_kwargs: {}
• warmup_ratio: 0.1
• warmup_steps: 0
• log_level: passive
• log_level_replica: warning
• log_on_each_node: True
• logging_nan_inf_filter: True
• save_safetensors: True
• save_on_each_node: False
• save_only_model: False
• restore_callback_states_from_checkpoint: False
• no_cuda: False
• use_cpu: False
• use_mps_device: False
• seed: 12
• data_seed: None
• jit_mode_eval: False
• use_ipex: False
• bf16: True
• fp16: False
• fp16_opt_level: O1
• half_precision_backend: auto
• bf16_full_eval: False
• fp16_full_eval: False
• tf32: None
• local_rank: 0
• ddp_backend: None
• tpu_num_cores: None
• tpu_metrics_debug: False
• debug: []
• dataloader_drop_last: False
• dataloader_num_workers: 0
• dataloader_prefetch_factor: None
• past_index: -1
• disable_tqdm: False
• remove_unused_columns: True
• label_names: None
• load_best_model_at_end: True
• ignore_data_skip: False
• fsdp: []
• fsdp_min_num_params: 0
• fsdp_config: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}
• fsdp_transformer_layer_cls_to_wrap: None
• accelerator_config: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}
• parallelism_config: None
• deepspeed: None
• label_smoothing_factor: 0.0
• optim: adamw_torch_fused
• optim_args: None
• adafactor: False
• group_by_length: False
• length_column_name: length
• ddp_find_unused_parameters: None
• ddp_bucket_cap_mb: None
• ddp_broadcast_buffers: False
• dataloader_pin_memory: True
• dataloader_persistent_workers: False
• skip_memory_metrics: True
• use_legacy_prediction_loop: False
• push_to_hub: False
• resume_from_checkpoint: None
• hub_model_id: None
• hub_strategy: every_save
• hub_private_repo: None
• hub_always_push: False
• hub_revision: None
• gradient_checkpointing: False
• gradient_checkpointing_kwargs: None
• include_inputs_for_metrics: False
• include_for_metrics: []
• eval_do_concat_batches: True
• fp16_backend: auto
• push_to_hub_model_id: None
• push_to_hub_organization: None
• mp_parameters:
• auto_find_batch_size: False
• full_determinism: False
• torchdynamo: None
• ray_scope: last
• ddp_timeout: 1800
• torch_compile: False
• torch_compile_backend: None
• torch_compile_mode: None
• include_tokens_per_second: False
• include_num_input_tokens_seen: False
• neftune_noise_alpha: None
• optim_target_modules: None
• batch_eval_metrics: False
• eval_on_start: False
• use_liger_kernel: False
• liger_kernel_config: None
• eval_use_gather_object: False
• average_tokens_across_devices: False
• prompts: None
• batch_sampler: batch_sampler
• multi_dataset_batch_sampler: proportional
• router_mapping: {}
• learning_rate_mapping: {}

Training Logs

Epoch	Step	Training Loss	Validation Loss	NanoMSMARCO_R100_ndcg@10	NanoNFCorpus_R100_ndcg@10	NanoNQ_R100_ndcg@10	NanoBEIR_R100_mean_ndcg@10
-1	-1	-	-	0.0000 (-0.5404)	0.2648 (-0.0602)	0.0388 (-0.4618)	0.1012 (-0.3541)
0.0002	1	0.9652	-	-	-	-	-
0.0203	100	1.0264	0.9387	0.1026 (-0.4379)	0.2550 (-0.0701)	0.1438 (-0.3569)	0.1671 (-0.2883)
0.0407	200	0.8354	0.7220	0.4720 (-0.0685)	0.3046 (-0.0204)	0.5663 (+0.0657)	0.4476 (-0.0077)
0.0610	300	0.7116	0.6593	0.4998 (-0.0406)	0.3404 (+0.0153)	0.5996 (+0.0990)	0.4799 (+0.0245)
0.0813	400	0.691	0.6539	0.5280 (-0.0124)	0.3542 (+0.0292)	0.6559 (+0.1553)	0.5127 (+0.0573)
0.1016	500	0.6317	0.6430	0.5017 (-0.0387)	0.3499 (+0.0249)	0.6520 (+0.1513)	0.5012 (+0.0458)
0.1220	600	0.658	0.6208	0.5725 (+0.0321)	0.3557 (+0.0307)	0.7198 (+0.2192)	0.5494 (+0.0940)
0.1423	700	0.6203	0.6104	0.5646 (+0.0242)	0.3989 (+0.0739)	0.7385 (+0.2379)	0.5674 (+0.1120)
0.1626	800	0.6221	0.6590	0.6029 (+0.0625)	0.4109 (+0.0859)	0.6997 (+0.1991)	0.5712 (+0.1158)
0.1830	900	0.6436	0.5981	0.6002 (+0.0597)	0.4210 (+0.0960)	0.7268 (+0.2262)	0.5827 (+0.1273)
0.2033	1000	0.6021	0.5989	0.5218 (-0.0187)	0.3943 (+0.0692)	0.6009 (+0.1002)	0.5056 (+0.0503)
0.2236	1100	0.6021	0.5833	0.5702 (+0.0298)	0.3885 (+0.0635)	0.6771 (+0.1764)	0.5453 (+0.0899)
0.2440	1200	0.5786	0.5696	0.5641 (+0.0237)	0.4210 (+0.0959)	0.6776 (+0.1770)	0.5543 (+0.0989)
0.2643	1300	0.6133	0.5733	0.6007 (+0.0603)	0.4098 (+0.0848)	0.7000 (+0.1994)	0.5702 (+0.1148)
0.2846	1400	0.6104	0.5701	0.6181 (+0.0777)	0.4114 (+0.0864)	0.7054 (+0.2047)	0.5783 (+0.1229)
0.3049	1500	0.5994	0.5712	0.5970 (+0.0566)	0.3949 (+0.0698)	0.6462 (+0.1456)	0.5460 (+0.0907)
0.3253	1600	0.5955	0.5596	0.5969 (+0.0565)	0.4288 (+0.1038)	0.7105 (+0.2099)	0.5787 (+0.1234)
0.3456	1700	0.5865	0.5680	0.6235 (+0.0830)	0.4385 (+0.1134)	0.6703 (+0.1696)	0.5774 (+0.1220)
0.3659	1800	0.5715	0.5913	0.6255 (+0.0850)	0.4379 (+0.1129)	0.6873 (+0.1867)	0.5836 (+0.1282)
0.3863	1900	0.5516	0.5707	0.6015 (+0.0611)	0.4060 (+0.0810)	0.6783 (+0.1777)	0.5620 (+0.1066)
0.4066	2000	0.5693	0.5601	0.6045 (+0.0640)	0.4038 (+0.0788)	0.6512 (+0.1505)	0.5531 (+0.0978)
0.4269	2100	0.6041	0.5516	0.6178 (+0.0774)	0.4206 (+0.0955)	0.7024 (+0.2018)	0.5803 (+0.1249)
0.4472	2200	0.5753	0.5499	0.6358 (+0.0954)	0.4076 (+0.0826)	0.6954 (+0.1948)	0.5796 (+0.1242)
0.4676	2300	0.5771	0.5680	0.6343 (+0.0939)	0.4238 (+0.0987)	0.6611 (+0.1604)	0.5731 (+0.1177)
0.4879	2400	0.5771	0.5906	0.6213 (+0.0809)	0.4216 (+0.0965)	0.6604 (+0.1598)	0.5678 (+0.1124)
0.5082	2500	0.5631	0.5652	0.6237 (+0.0833)	0.4295 (+0.1045)	0.6806 (+0.1799)	0.5779 (+0.1226)
0.5286	2600	0.5823	0.5615	0.6398 (+0.0993)	0.4187 (+0.0936)	0.6962 (+0.1956)	0.5849 (+0.1295)
0.5489	2700	0.5689	0.5645	0.6224 (+0.0820)	0.4328 (+0.1078)	0.6984 (+0.1977)	0.5845 (+0.1292)
0.5692	2800	0.5473	0.5621	0.6000 (+0.0596)	0.4353 (+0.1102)	0.6748 (+0.1742)	0.5700 (+0.1147)
0.5896	2900	0.5682	0.5556	0.5995 (+0.0591)	0.4324 (+0.1073)	0.7073 (+0.2066)	0.5797 (+0.1243)
0.6099	3000	0.5713	0.5513	0.6080 (+0.0675)	0.4479 (+0.1229)	0.7065 (+0.2058)	0.5875 (+0.1321)
0.6302	3100	0.5633	0.5507	0.6109 (+0.0704)	0.4374 (+0.1124)	0.6865 (+0.1858)	0.5783 (+0.1229)
0.6505	3200	0.5784	0.5487	0.6192 (+0.0788)	0.4327 (+0.1076)	0.6817 (+0.1810)	0.5779 (+0.1225)
0.6709	3300	0.574	0.5454	0.6247 (+0.0843)	0.4235 (+0.0984)	0.7050 (+0.2044)	0.5844 (+0.1290)
0.6912	3400	0.57	0.5453	0.6416 (+0.1012)	0.4444 (+0.1194)	0.7099 (+0.2092)	0.5986 (+0.1433)
0.7115	3500	0.5519	0.5449	0.6312 (+0.0908)	0.4421 (+0.1170)	0.7040 (+0.2034)	0.5924 (+0.1371)
0.7319	3600	0.5563	0.5487	0.6228 (+0.0824)	0.4326 (+0.1075)	0.6887 (+0.1881)	0.5814 (+0.1260)
0.7522	3700	0.5485	0.5464	0.6179 (+0.0775)	0.4407 (+0.1157)	0.6953 (+0.1947)	0.5846 (+0.1293)
0.7725	3800	0.5859	0.5485	0.6253 (+0.0849)	0.4364 (+0.1113)	0.7080 (+0.2074)	0.5899 (+0.1345)
0.7928	3900	0.5636	0.5488	0.6207 (+0.0803)	0.4469 (+0.1218)	0.7120 (+0.2113)	0.5932 (+0.1378)
0.8132	4000	0.5534	0.5466	0.6182 (+0.0778)	0.4495 (+0.1245)	0.7015 (+0.2008)	0.5897 (+0.1344)
0.8335	4100	0.5404	0.5450	0.6184 (+0.0779)	0.4468 (+0.1217)	0.6603 (+0.1596)	0.5751 (+0.1198)
0.8538	4200	0.5584	0.5454	0.6505 (+0.1101)	0.4476 (+0.1226)	0.6859 (+0.1852)	0.5947 (+0.1393)
0.8742	4300	0.5413	0.5414	0.6533 (+0.1129)	0.4451 (+0.1201)	0.6966 (+0.1959)	0.5983 (+0.1430)
0.8945	4400	0.5355	0.5407	0.6397 (+0.0993)	0.4422 (+0.1172)	0.6985 (+0.1978)	0.5935 (+0.1381)
0.9148	4500	0.5616	0.5393	0.6482 (+0.1078)	0.4360 (+0.1110)	0.6944 (+0.1937)	0.5929 (+0.1375)
0.9351	4600	0.5261	0.5381	0.6593 (+0.1189)	0.4383 (+0.1132)	0.6893 (+0.1886)	0.5956 (+0.1403)
0.9555	4700	0.5738	0.5376	0.6530 (+0.1126)	0.4368 (+0.1117)	0.6877 (+0.1871)	0.5925 (+0.1371)
0.9758	4800	0.5479	0.5371	0.6570 (+0.1166)	0.4383 (+0.1132)	0.6917 (+0.1911)	0.5957 (+0.1403)
0.9961	4900	0.5432	0.5370	0.6570 (+0.1166)	0.4370 (+0.1120)	0.6909 (+0.1902)	0.5950 (+0.1396)
-1	-1	-	-	0.6416 (+0.1012)	0.4444 (+0.1194)	0.7099 (+0.2092)	0.5986 (+0.1433)

• The bold row denotes the saved checkpoint.

Framework Versions

• Python: 3.9.18
• Sentence Transformers: 5.1.1
• Transformers: 4.56.2
• PyTorch: 2.8.0+cu128
• Accelerate: 1.10.1
• Datasets: 4.1.1
• Tokenizers: 0.22.1

Citation

BibTeX

Sentence Transformers

@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",
}

RankNetLoss

@inproceedings{burges2005learning,
  title={Learning to Rank using Gradient Descent},
  author={Burges, Chris and Shaked, Tal and Renshaw, Erin and Lazier, Ari and Deeds, Matt and Hamilton, Nicole and Hullender, Greg},
  booktitle={Proceedings of the 22nd international conference on Machine learning},
  pages={89--96},
  year={2005}
}