bobox's picture
train_loss = AdaptiveLayerLoss(model=model,
25d3738 verified
|
raw
history blame
126 kB
metadata
language:
  - en
library_name: sentence-transformers
tags:
  - sentence-transformers
  - sentence-similarity
  - feature-extraction
  - generated_from_trainer
  - dataset_size:78183
  - loss:AdaptiveLayerLoss
  - loss:CoSENTLoss
  - loss:GISTEmbedLoss
  - loss:OnlineContrastiveLoss
  - loss:MultipleNegativesSymmetricRankingLoss
base_model: microsoft/deberta-v3-small
datasets:
  - sentence-transformers/all-nli
  - sentence-transformers/stsb
  - tals/vitaminc
  - nyu-mll/glue
  - allenai/scitail
  - sentence-transformers/xsum
  - sentence-transformers/sentence-compression
  - allenai/sciq
  - allenai/qasc
  - allenai/openbookqa
  - sentence-transformers/msmarco-msmarco-distilbert-base-v3
  - sentence-transformers/natural-questions
  - sentence-transformers/trivia-qa
  - sentence-transformers/quora-duplicates
  - sentence-transformers/gooaq
metrics:
  - pearson_cosine
  - spearman_cosine
  - pearson_manhattan
  - spearman_manhattan
  - pearson_euclidean
  - spearman_euclidean
  - pearson_dot
  - spearman_dot
  - pearson_max
  - spearman_max
widget:
  - source_sentence: >-
      The X and Y chromosomes in human beings that determine the sex of an
      individual.
    sentences:
      - A glacier leaves behind bare rock when it retreats.
      - >-
        Prokaryotes are unicellular organisms that lack organelles surrounded by
        membranes.
      - >-
        Mammalian sex determination is determined genetically by the presence of
        chromosomes identified by the letters x and y.
  - source_sentence: Police officer with riot shield stands in front of crowd.
    sentences:
      - A police officer stands in front of a crowd.
      - A pair of people play video games together on a couch.
      - People are outside digging a hole.
  - source_sentence: >-
      A young girl sitting on a white comforter on a bed covered with clothing,
      holding a yellow stuffed duck.
    sentences:
      - A man standing in a room is pointing up.
      - A Little girl is enjoying cake outside.
      - A yellow duck being held by a girl.
  - source_sentence: A teenage girl in winter clothes slides down a decline in a red sled.
    sentences:
      - A woman preparing vegetables.
      - A girl is sliding on a red sled.
      - A person is on a beach.
  - source_sentence: How many hymns of Luther were included in the Achtliederbuch?
    sentences:
      - >-
        the ABC News building was renamed Peter Jennings Way in 2006 in honor of
        the recently deceased longtime ABC News chief anchor and anchor of World
        News Tonight.
      - >-
        In early 2009, Disney–ABC Television Group merged ABC Entertainment and
        ABC Studios into a new division, ABC Entertainment Group, which would be
        responsible for both its production and broadcasting operations.
      - >-
        Luther's hymns were included in early Lutheran hymnals and spread the
        ideas of the Reformation.
pipeline_tag: sentence-similarity
model-index:
  - name: SentenceTransformer based on microsoft/deberta-v3-small
    results:
      - task:
          type: semantic-similarity
          name: Semantic Similarity
        dataset:
          name: sts test
          type: sts-test
        metrics:
          - type: pearson_cosine
            value: 0.7643554891812735
            name: Pearson Cosine
          - type: spearman_cosine
            value: 0.7591947144735277
            name: Spearman Cosine
          - type: pearson_manhattan
            value: 0.769108031897504
            name: Pearson Manhattan
          - type: spearman_manhattan
            value: 0.7590854149926064
            name: Spearman Manhattan
          - type: pearson_euclidean
            value: 0.7608914486061109
            name: Pearson Euclidean
          - type: spearman_euclidean
            value: 0.7488315275075106
            name: Spearman Euclidean
          - type: pearson_dot
            value: 0.6257426306656716
            name: Pearson Dot
          - type: spearman_dot
            value: 0.6045082518573447
            name: Spearman Dot
          - type: pearson_max
            value: 0.769108031897504
            name: Pearson Max
          - type: spearman_max
            value: 0.7591947144735277
            name: Spearman Max

SentenceTransformer based on microsoft/deberta-v3-small

This is a sentence-transformers model finetuned from microsoft/deberta-v3-small on the nli-pairs, sts-label, vitaminc-pairs, qnli-contrastive, scitail-pairs-qa, scitail-pairs-pos, xsum-pairs, compression-pairs, sciq_pairs, qasc_pairs, openbookqa_pairs, msmarco_pairs, nq_pairs, trivia_pairs, quora_pairs and gooaq_pairs datasets. It maps sentences & paragraphs to a 768-dimensional dense vector space and can be used for semantic textual similarity, semantic search, paraphrase mining, text classification, clustering, and more.

Model Details

Model Description

Model Sources

Full Model Architecture

SentenceTransformer(
  (0): Transformer({'max_seq_length': 512, 'do_lower_case': False}) with Transformer model: DebertaV2Model 
  (1): Pooling({'word_embedding_dimension': 768, 'pooling_mode_cls_token': False, 'pooling_mode_mean_tokens': True, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False, 'pooling_mode_weightedmean_tokens': False, 'pooling_mode_lasttoken': False, 'include_prompt': True})
)

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 SentenceTransformer

# Download from the 🤗 Hub
model = SentenceTransformer("bobox/DeBERTaV3-small-SenTra-AdaptiveLayers-AllSoft-HighTemp")
# Run inference
sentences = [
    'How many hymns of Luther were included in the Achtliederbuch?',
    "Luther's hymns were included in early Lutheran hymnals and spread the ideas of the Reformation.",
    'the ABC News building was renamed Peter Jennings Way in 2006 in honor of the recently deceased longtime ABC News chief anchor and anchor of World News Tonight.',
]
embeddings = model.encode(sentences)
print(embeddings.shape)
# [3, 768]

# Get the similarity scores for the embeddings
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [3, 3]

Evaluation

Metrics

Semantic Similarity

Metric Value
pearson_cosine 0.7644
spearman_cosine 0.7592
pearson_manhattan 0.7691
spearman_manhattan 0.7591
pearson_euclidean 0.7609
spearman_euclidean 0.7488
pearson_dot 0.6257
spearman_dot 0.6045
pearson_max 0.7691
spearman_max 0.7592

Training Details

Training Datasets

nli-pairs

  • Dataset: nli-pairs at d482672
  • Size: 6,500 training samples
  • Columns: sentence1 and sentence2
  • Approximate statistics based on the first 1000 samples:
    sentence1 sentence2
    type string string
    details
    • min: 5 tokens
    • mean: 16.62 tokens
    • max: 62 tokens
    • min: 4 tokens
    • mean: 9.46 tokens
    • max: 29 tokens
  • Samples:
    sentence1 sentence2
    A person on a horse jumps over a broken down airplane. A person is outdoors, on a horse.
    Children smiling and waving at camera There are children present
    A boy is jumping on skateboard in the middle of a red bridge. The boy does a skateboarding trick.
  • Loss: AdaptiveLayerLoss with these parameters:
    {
        "loss": "GISTEmbedLoss",
        "n_layers_per_step": -1,
        "last_layer_weight": 1.5,
        "prior_layers_weight": 0.15,
        "kl_div_weight": 2,
        "kl_temperature": 2
    }
    

sts-label

  • Dataset: sts-label at ab7a5ac
  • Size: 5,749 training samples
  • Columns: sentence1, sentence2, and score
  • Approximate statistics based on the first 1000 samples:
    sentence1 sentence2 score
    type string string float
    details
    • min: 6 tokens
    • mean: 9.81 tokens
    • max: 27 tokens
    • min: 5 tokens
    • mean: 9.74 tokens
    • max: 25 tokens
    • min: 0.0
    • mean: 0.54
    • max: 1.0
  • Samples:
    sentence1 sentence2 score
    A plane is taking off. An air plane is taking off. 1.0
    A man is playing a large flute. A man is playing a flute. 0.76
    A man is spreading shreded cheese on a pizza. A man is spreading shredded cheese on an uncooked pizza. 0.76
  • Loss: CoSENTLoss with these parameters:
    {
        "scale": 20.0,
        "similarity_fct": "pairwise_cos_sim"
    }
    

vitaminc-pairs

  • Dataset: vitaminc-pairs at be6febb
  • Size: 3,194 training samples
  • Columns: label, sentence1, and sentence2
  • Approximate statistics based on the first 1000 samples:
    label sentence1 sentence2
    type int string string
    details
    • 1: 100.00%
    • min: 6 tokens
    • mean: 15.76 tokens
    • max: 75 tokens
    • min: 8 tokens
    • mean: 37.3 tokens
    • max: 502 tokens
  • Samples:
    label sentence1 sentence2
    1 The film will be screened in 2200 theaters . In the United States and Canada , pre-release tracking suggest the film will gross $ 7�8 million from 2,200 theaters in its opening weekend , trailing fellow newcomer 10 Cloverfield Lane ( $ 25�30 million projection ) , but similar t
    1 Neighbors 2 : Sorority Rising ( film ) scored over 65 % on Rotten Tomatoes . On Rotten Tomatoes , the film has a rating of 67 % , based on 105 reviews , with an average rating of 5.9/10 .
    1 Averaged on more than 65 reviews , The Handmaiden scored 94 % . On Rotten Tomatoes , the film has a rating of 94 % , based on 67 reviews , with an average rating of 8/10 .
  • Loss: AdaptiveLayerLoss with these parameters:
    {
        "loss": "GISTEmbedLoss",
        "n_layers_per_step": -1,
        "last_layer_weight": 1.5,
        "prior_layers_weight": 0.15,
        "kl_div_weight": 2,
        "kl_temperature": 2
    }
    

qnli-contrastive

  • Dataset: qnli-contrastive at bcdcba7
  • Size: 4,000 training samples
  • Columns: sentence1, sentence2, and label
  • Approximate statistics based on the first 1000 samples:
    sentence1 sentence2 label
    type string string int
    details
    • min: 6 tokens
    • mean: 13.64 tokens
    • max: 30 tokens
    • min: 6 tokens
    • mean: 34.57 tokens
    • max: 149 tokens
    • 0: 100.00%
  • Samples:
    sentence1 sentence2 label
    What professors established the importance of Whitehead's work? Professors such as Wieman, Charles Hartshorne, Bernard Loomer, Bernard Meland, and Daniel Day Williams made Whitehead's philosophy arguably the most important intellectual thread running through the Divinity School. 0
    When did people start living on the edge of the desert? It was long believed that the region had been this way since about 1600 BCE, after shifts in the Earth's axis increased temperatures and decreased precipitation. 0
    What was the title of Gertrude Stein's 1906-1908 book? Picasso in turn was an important influence on Stein's writing. 0
  • Loss: AdaptiveLayerLoss with these parameters:
    {
        "loss": "OnlineContrastiveLoss",
        "n_layers_per_step": -1,
        "last_layer_weight": 1.5,
        "prior_layers_weight": 0.15,
        "kl_div_weight": 2,
        "kl_temperature": 2
    }
    

scitail-pairs-qa

  • Dataset: scitail-pairs-qa at 0cc4353
  • Size: 4,300 training samples
  • Columns: sentence2 and sentence1
  • Approximate statistics based on the first 1000 samples:
    sentence2 sentence1
    type string string
    details
    • min: 7 tokens
    • mean: 16.2 tokens
    • max: 41 tokens
    • min: 7 tokens
    • mean: 14.65 tokens
    • max: 33 tokens
  • Samples:
    sentence2 sentence1
    Ash that enters the air naturally as a result of a volcano eruption is classified as a primary pollutant. Ash that enters the air naturally as a result of a volcano eruption is classified as what kind of pollutant?
    Exposure to ultraviolet radiation can increase the amount of pigment in the skin and make it appear darker. Exposure to what can increase the amount of pigment in the skin and make it appear darker?
    A lysozyme destroys bacteria by digesting their cell walls. How does lysozyme destroy bacteria?
  • Loss: AdaptiveLayerLoss with these parameters:
    {
        "loss": "GISTEmbedLoss",
        "n_layers_per_step": -1,
        "last_layer_weight": 1.5,
        "prior_layers_weight": 0.15,
        "kl_div_weight": 2,
        "kl_temperature": 2
    }
    

scitail-pairs-pos

  • Dataset: scitail-pairs-pos at 0cc4353
  • Size: 2,200 training samples
  • Columns: sentence1 and sentence2
  • Approximate statistics based on the first 1000 samples:
    sentence1 sentence2
    type string string
    details
    • min: 7 tokens
    • mean: 23.6 tokens
    • max: 74 tokens
    • min: 7 tokens
    • mean: 15.23 tokens
    • max: 41 tokens
  • Samples:
    sentence1 sentence2
    An atom that gains electrons would be a negative ion. Atoms that have gained electrons and become negatively charged are called negative ions.
    Scientists will use data collected during the collisions to explore the particles known as quarks and gluons that make up protons and neutrons. Protons and neutrons are made of quarks, which are fundamental particles of matter.
    Watersheds and divides All of the land area whose water drains into a stream system is called the system's watershed. All of the land drained by a river system is called its basin, or the "wet" term watershed
  • Loss: AdaptiveLayerLoss with these parameters:
    {
        "loss": "GISTEmbedLoss",
        "n_layers_per_step": -1,
        "last_layer_weight": 1.5,
        "prior_layers_weight": 0.15,
        "kl_div_weight": 2,
        "kl_temperature": 2
    }
    

xsum-pairs

  • Dataset: xsum-pairs at 788ddaf
  • Size: 2,500 training samples
  • Columns: sentence1 and sentence2
  • Approximate statistics based on the first 1000 samples:
    sentence1 sentence2
    type string string
    details
    • min: 2 tokens
    • mean: 350.46 tokens
    • max: 512 tokens
    • min: 6 tokens
    • mean: 27.13 tokens
    • max: 70 tokens
  • Samples:
    sentence1 sentence2
    An eyewitness told BBC Persian that the crowds were sharply divided between hardliners and moderates, but it was clear many people had responded to a call from former President Mohammad Khatami to attend the funeral as a show of support for the opposition reform movement.
    Some were chanting opposition slogans, and others carried placards emphasising Mr Rafsanjani's links to the moderate and reformist camps.
    "Long live Khatami, Long Live Rouhani. Hashemi, your soul is at peace!" said one banner.
    "The circle became too closed for the centre," said another, using a quotation from Persian poetry to underline the growing distance in recent years between Mr Rafsanjani and Iran's hardline political establishment.
    At one stage state television played loud music over its live broadcast of the event in order to drown out opposition slogans being chanted by the crowd.
    As the official funeral eulogies were relayed to the crowds on the streets, they responded with calls of support for former President Khatami, and opposition leader Mir Hossein Mousavi, and shouts of: "You have the loudspeakers, we have the voice! Shame on you, Shame on State TV!"
    On Iranian social media the funeral has been the number one topic with many opposition supporters using the hashtag #weallgathered to indicate their support and sympathy.
    People have been posting photos and videos emphasising the number of opposition supporters out on the streets and showing the opposition slogans which state TV has been trying to obscure.
    But government supporters have also taken to Twitter to play down the opposition showing at the funeral, accusing them of political opportunism.
    "A huge army came out of love of the Supreme Leader," wrote a cleric called Sheikh Reza. "While a few foot soldiers came with their cameras to show off."
    Another conversation engaging many on Twitter involved the wording of the prayers used at the funeral.
    Did the Supreme Leader Ayatollah Ali Khamenei deliberately leave out a section praising the goodness of the deceased, some opposition supporters asked. And was this a comment on the political tensions between the two?
    "No," responded another Twitter user, cleric Abbas Zolghadri. "The words of the prayer can be changed. There are no strict rules."
    He followed this with a poignant photo of an empty grave - "Hashemi's final resting place" was the caption, summing up the sense of loss felt by Iranians of many different political persuasions despite the deep and bitter divisions.
    Tehran has seen some of the biggest crowds on the streets since the 2009 "Green Movement" opposition demonstrations, as an estimated 2.5 million people gathered to bid farewell to Akbar Hashemi Rafsanjani, the man universally known as "Hashemi".
    Mark Evans is retracing the same route across the Rub Al Khali, also known as the "Empty Quarter", taken by Bristol pioneer Bertram Thomas in 1930.
    The 54-year-old Shropshire-born explorer is leading a three-man team to walk the 800 mile (1,300 km) journey from Salalah, Oman to Doha, Qatar.
    The trek is expected to take 60 days.
    The Rub Al Khali desert is considered one of the hottest, driest and most inhospitable places on earth.
    Nearly two decades after Thomas completed his trek, British explorer and writer Sir Wilfred Thesiger crossed the Empty Quarter - mapping it in detail along the way.
    60 days
    To cross the Rub' Al Khali desert
    * From Salalah in Oman to Doha, Qatar
    * Walking with camels for 1,300km
    * Area nearly three times the size of the UK
    Completed by explorer Bertram Thomas in 1930
    Bertram Thomas, who hailed from Pill, near Bristol, received telegrams of congratulation from both King George V and Sultan Taimur, then ruler of Oman.
    He went on to lecture all over the world about the journey and to write a book called Arabia Felix.
    Unlike Mr Evans, Thomas did not obtain permission for his expedition.
    He said: "The biggest challenges for Thomas were warring tribes, lack of water in the waterholes and his total dependence on his Omani companion Sheikh Saleh to negotiate their way through the desert.
    "The biggest challenge for those who wanted to make the crossing in recent decades has been obtaining government permissions to walk through this desolate and unknown territory."
    An explorer has embarked on a challenge to become only the third British person in history to cross the largest sand desert in the world.
    An Olympic gold medallist, he was also three-time world heavyweight champion and took part in some of the most memorable fights in boxing history.
    He had a professional career spanning 21 years and BBC Sport takes a look at his 61 fights in more detail.
    Boxing legend Muhammad Ali, who died at the age of 74, became a sporting icon during his career.
  • Loss: AdaptiveLayerLoss with these parameters:
    {
        "loss": "GISTEmbedLoss",
        "n_layers_per_step": -1,
        "last_layer_weight": 1.5,
        "prior_layers_weight": 0.15,
        "kl_div_weight": 2,
        "kl_temperature": 2
    }
    

compression-pairs

  • Dataset: compression-pairs at 605bc91
  • Size: 4,000 training samples
  • Columns: sentence1 and sentence2
  • Approximate statistics based on the first 1000 samples:
    sentence1 sentence2
    type string string
    details
    • min: 10 tokens
    • mean: 31.89 tokens
    • max: 125 tokens
    • min: 5 tokens
    • mean: 10.21 tokens
    • max: 28 tokens
  • Samples:
    sentence1 sentence2
    The USHL completed an expansion draft on Monday as 10 players who were on the rosters of USHL teams during the 2009-10 season were selected by the League's two newest entries, the Muskegon Lumberjacks and Dubuque Fighting Saints. USHL completes expansion draft
    Major League Baseball Commissioner Bud Selig will be speaking at St. Norbert College next month. Bud Selig to speak at St. Norbert College
    It's fresh cherry time in Michigan and the best time to enjoy this delicious and nutritious fruit. It's cherry time
  • Loss: AdaptiveLayerLoss with these parameters:
    {
        "loss": "MultipleNegativesSymmetricRankingLoss",
        "n_layers_per_step": -1,
        "last_layer_weight": 1.5,
        "prior_layers_weight": 0.15,
        "kl_div_weight": 2,
        "kl_temperature": 2
    }
    

sciq_pairs

  • Dataset: sciq_pairs at 2c94ad3
  • Size: 6,500 training samples
  • Columns: sentence1 and sentence2
  • Approximate statistics based on the first 1000 samples:
    sentence1 sentence2
    type string string
    details
    • min: 7 tokens
    • mean: 17.26 tokens
    • max: 60 tokens
    • min: 2 tokens
    • mean: 84.37 tokens
    • max: 512 tokens
  • Samples:
    sentence1 sentence2
    What type of organism is commonly used in preparation of foods such as cheese and yogurt? Mesophiles grow best in moderate temperature, typically between 25°C and 40°C (77°F and 104°F). Mesophiles are often found living in or on the bodies of humans or other animals. The optimal growth temperature of many pathogenic mesophiles is 37°C (98°F), the normal human body temperature. Mesophilic organisms have important uses in food preparation, including cheese, yogurt, beer and wine.
    What phenomenon makes global winds blow northeast to southwest or the reverse in the northern hemisphere and northwest to southeast or the reverse in the southern hemisphere? Without Coriolis Effect the global winds would blow north to south or south to north. But Coriolis makes them blow northeast to southwest or the reverse in the Northern Hemisphere. The winds blow northwest to southeast or the reverse in the southern hemisphere.
    Changes from a less-ordered state to a more-ordered state (such as a liquid to a solid) are always what? Summary Changes of state are examples of phase changes, or phase transitions. All phase changes are accompanied by changes in the energy of a system. Changes from a more-ordered state to a less-ordered state (such as a liquid to a gas) areendothermic. Changes from a less-ordered state to a more-ordered state (such as a liquid to a solid) are always exothermic. The conversion of a solid to a liquid is called fusion (or melting). The energy required to melt 1 mol of a substance is its enthalpy of fusion (ΔHfus). The energy change required to vaporize 1 mol of a substance is the enthalpy of vaporization (ΔHvap). The direct conversion of a solid to a gas is sublimation. The amount of energy needed to sublime 1 mol of a substance is its enthalpy of sublimation (ΔHsub) and is the sum of the enthalpies of fusion and vaporization. Plots of the temperature of a substance versus heat added or versus heating time at a constant rate of heating are calledheating curves. Heating curves relate temperature changes to phase transitions. A superheated liquid, a liquid at a temperature and pressure at which it should be a gas, is not stable. A cooling curve is not exactly the reverse of the heating curve because many liquids do not freeze at the expected temperature. Instead, they form a supercooled liquid, a metastable liquid phase that exists below the normal melting point. Supercooled liquids usually crystallize on standing, or adding a seed crystal of the same or another substance can induce crystallization.
  • Loss: AdaptiveLayerLoss with these parameters:
    {
        "loss": "GISTEmbedLoss",
        "n_layers_per_step": -1,
        "last_layer_weight": 1.5,
        "prior_layers_weight": 0.15,
        "kl_div_weight": 2,
        "kl_temperature": 2
    }
    

qasc_pairs

  • Dataset: qasc_pairs at a34ba20
  • Size: 6,500 training samples
  • Columns: id, sentence1, and sentence2
  • Approximate statistics based on the first 1000 samples:
    id sentence1 sentence2
    type string string string
    details
    • min: 17 tokens
    • mean: 21.35 tokens
    • max: 27 tokens
    • min: 5 tokens
    • mean: 11.47 tokens
    • max: 25 tokens
    • min: 14 tokens
    • mean: 35.55 tokens
    • max: 66 tokens
  • Samples:
    id sentence1 sentence2
    3E7TUJ2EGCLQNOV1WEAJ2NN9ROPD9K What type of water formation is formed by clouds? beads of water are formed by water vapor condensing. Clouds are made of water vapor.. Beads of water can be formed by clouds.
    3LS2AMNW5FPNJK3C3PZLZCPX562OQO Where do beads of water come from? beads of water are formed by water vapor condensing. Condensation is the change of water vapor to a liquid.. Vapor turning into a liquid leaves behind beads of water
    3TMFV4NEP8DPIPCI8H9VUFHJG8V8W3 What forms beads of water? beads of water are formed by water vapor condensing. An example of water vapor is steam.. Steam forms beads of water.
  • Loss: AdaptiveLayerLoss with these parameters:
    {
        "loss": "GISTEmbedLoss",
        "n_layers_per_step": -1,
        "last_layer_weight": 1.5,
        "prior_layers_weight": 0.15,
        "kl_div_weight": 2,
        "kl_temperature": 2
    }
    

openbookqa_pairs

  • Dataset: openbookqa_pairs at 388097e
  • Size: 2,740 training samples
  • Columns: sentence1 and sentence2
  • Approximate statistics based on the first 1000 samples:
    sentence1 sentence2
    type string string
    details
    • min: 3 tokens
    • mean: 13.83 tokens
    • max: 78 tokens
    • min: 4 tokens
    • mean: 11.37 tokens
    • max: 30 tokens
  • Samples:
    sentence1 sentence2
    The sun is responsible for the sun is the source of energy for physical cycles on Earth
    When food is reduced in the stomach digestion is when stomach acid breaks down food
    Stars are a star is made of gases
  • Loss: AdaptiveLayerLoss with these parameters:
    {
        "loss": "GISTEmbedLoss",
        "n_layers_per_step": -1,
        "last_layer_weight": 1.5,
        "prior_layers_weight": 0.15,
        "kl_div_weight": 2,
        "kl_temperature": 2
    }
    

msmarco_pairs

  • Dataset: msmarco_pairs at 28ff31e
  • Size: 6,500 training samples
  • Columns: sentence1 and sentence2
  • Approximate statistics based on the first 1000 samples:
    sentence1 sentence2
    type string string
    details
    • min: 4 tokens
    • mean: 8.61 tokens
    • max: 27 tokens
    • min: 18 tokens
    • mean: 75.09 tokens
    • max: 206 tokens
  • Samples:
    sentence1 sentence2
    what are the liberal arts? liberal arts. 1. the academic course of instruction at a college intended to provide general knowledge and comprising the arts, humanities, natural sciences, and social sciences, as opposed to professional or technical subjects.
    what is the mechanism of action of fibrinolytic or thrombolytic drugs? Baillière's Clinical Haematology. 6 Mechanism of action of the thrombolytic agents. 6 Mechanism of action of the thrombolytic agents JEFFREY I. WEITZ Fibrin formed during the haemostatic, inflammatory or tissue repair process serves a temporary role, and must be degraded to restore normal tissue function and structure.
    what is normal plat count 78 Followers. A. Platelets are the tiny blood cells that help stop bleeding by binding together to form a clump or plug at sites of injury inside blood vessels. A normal platelet count is between 150,000 and 450,000 platelets per microliter (one-millionth of a liter, abbreviated mcL).The average platelet count is 237,000 per mcL in men and 266,000 per mcL in women.8 Followers. A. Platelets are the tiny blood cells that help stop bleeding by binding together to form a clump or plug at sites of injury inside blood vessels. A normal platelet count is between 150,000 and 450,000 platelets per microliter (one-millionth of a liter, abbreviated mcL).
  • Loss: AdaptiveLayerLoss with these parameters:
    {
        "loss": "GISTEmbedLoss",
        "n_layers_per_step": -1,
        "last_layer_weight": 1.5,
        "prior_layers_weight": 0.15,
        "kl_div_weight": 2,
        "kl_temperature": 2
    }
    

nq_pairs

  • Dataset: nq_pairs at f9e894e
  • Size: 6,500 training samples
  • Columns: sentence1 and sentence2
  • Approximate statistics based on the first 1000 samples:
    sentence1 sentence2
    type string string
    details
    • min: 10 tokens
    • mean: 11.77 tokens
    • max: 21 tokens
    • min: 16 tokens
    • mean: 131.57 tokens
    • max: 512 tokens
  • Samples:
    sentence1 sentence2
    when did richmond last play in a preliminary final Richmond Football Club Richmond began 2017 with 5 straight wins, a feat it had not achieved since 1995. A series of close losses hampered the Tigers throughout the middle of the season, including a 5-point loss to the Western Bulldogs, 2-point loss to Fremantle, and a 3-point loss to the Giants. Richmond ended the season strongly with convincing victories over Fremantle and St Kilda in the final two rounds, elevating the club to 3rd on the ladder. Richmond's first final of the season against the Cats at the MCG attracted a record qualifying final crowd of 95,028; the Tigers won by 51 points. Having advanced to the first preliminary finals for the first time since 2001, Richmond defeated Greater Western Sydney by 36 points in front of a crowd of 94,258 to progress to the Grand Final against Adelaide, their first Grand Final appearance since 1982. The attendance was 100,021, the largest crowd to a grand final since 1986. The Crows led at quarter time and led by as many as 13, but the Tigers took over the game as it progressed and scored seven straight goals at one point. They eventually would win by 48 points – 16.12 (108) to Adelaide's 8.12 (60) – to end their 37-year flag drought.[22] Dustin Martin also became the first player to win a Premiership medal, the Brownlow Medal and the Norm Smith Medal in the same season, while Damien Hardwick was named AFL Coaches Association Coach of the Year. Richmond's jump from 13th to premiers also marked the biggest jump from one AFL season to the next.
    who sang what in the world's come over you Jack Scott (singer) At the beginning of 1960, Scott again changed record labels, this time to Top Rank Records.[1] He then recorded four Billboard Hot 100 hits – "What in the World's Come Over You" (#5), "Burning Bridges" (#3) b/w "Oh Little One" (#34), and "It Only Happened Yesterday" (#38).[1] "What in the World's Come Over You" was Scott's second gold disc winner.[6] Scott continued to record and perform during the 1960s and 1970s.[1] His song "You're Just Gettin' Better" reached the country charts in 1974.[1] In May 1977, Scott recorded a Peel session for BBC Radio 1 disc jockey, John Peel.
    who produces the most wool in the world Wool Global wool production is about 2 million tonnes per year, of which 60% goes into apparel. Wool comprises ca 3% of the global textile market, but its value is higher owing to dying and other modifications of the material.[1] Australia is a leading producer of wool which is mostly from Merino sheep but has been eclipsed by China in terms of total weight.[30] New Zealand (2016) is the third-largest producer of wool, and the largest producer of crossbred wool. Breeds such as Lincoln, Romney, Drysdale, and Elliotdale produce coarser fibers, and wool from these sheep is usually used for making carpets.
  • Loss: AdaptiveLayerLoss with these parameters:
    {
        "loss": "GISTEmbedLoss",
        "n_layers_per_step": -1,
        "last_layer_weight": 1.5,
        "prior_layers_weight": 0.15,
        "kl_div_weight": 2,
        "kl_temperature": 2
    }
    

trivia_pairs

  • Dataset: trivia_pairs at a7c36e3
  • Size: 6,500 training samples
  • Columns: sentence1 and sentence2
  • Approximate statistics based on the first 1000 samples:
    sentence1 sentence2
    type string string
    details
    • min: 8 tokens
    • mean: 15.16 tokens
    • max: 48 tokens
    • min: 19 tokens
    • mean: 456.87 tokens
    • max: 512 tokens
  • Samples:
    sentence1 sentence2
    Which American-born Sinclair won the Nobel Prize for Literature in 1930? The Nobel Prize in Literature 1930 The Nobel Prize in Literature 1930 Sinclair Lewis The Nobel Prize in Literature 1930 Sinclair Lewis Prize share: 1/1 The Nobel Prize in Literature 1930 was awarded to Sinclair Lewis "for his vigorous and graphic art of description and his ability to create, with wit and humour, new types of characters". Photos: Copyright © The Nobel Foundation Share this: To cite this page MLA style: "The Nobel Prize in Literature 1930". Nobelprize.org. Nobel Media AB 2014. Web. 18 Jan 2017. http://www.nobelprize.org/nobel_prizes/literature/laureates/1930/
    Where in England was Dame Judi Dench born? Judi Dench - IMDb IMDb Actress
    In which decade did Billboard magazine first publish and American hit chart? The US Billboard song chart The US Billboard song chart Search this site with Google Song chart US Billboard The Billboard magazine has published various music charts starting (with sheet music) in 1894, the first "Music Hit Parade" was published in 1936 , the first "Music Popularity Chart" was calculated in 1940 . These charts became less irregular until the weekly "Hot 100" was started in 1958 . The current chart combines sales, airplay and downloads. A music collector that calls himself Bullfrog has been consolidating the complete chart from 1894 to the present day. he has published this information in a comprehenive spreadsheet (which can be obtained at bullfrogspond.com/ ). The Bullfrog data assigns each song a unique identifier, something like "1968_076" (which just happens to be the Bee Gees song "I've Gotta Get A Message To You"). This "Whitburn Number" is provided to match with the books of Joel Whitburn and consists of the year and a ranking within the year. A song that first entered the charts in December and has a long run is listed the following year. This numbering scheme means that songs which are still in the charts cannot be assigned a final id, because their ranking might change. So the definitive listing for a year cannot be final until about April. In our listing we only use songs with finalised IDs, this means that every year we have to wait until last year's entries are finalised before using them. (Source bullfrogspond.com/ , the original version used here was 20090808 with extra data from: the 2009 data from 20091219 the 2010 data from 20110305 the 2011 data from 20120929 the 2012 data from 20130330 the 2013 data from 20150328 The 20150328 data was the last one produced before the Billboard company forced the data to be withdrawn. As far as we know there are no more recent data sets available. This pattern of obtaining the data for a particular year in the middle of the following one comes from the way that the Bullfrog project generates the identifier for a song (what they call the "Prefix" in the spreadsheet). Recent entries are identified with keys like "2015-008" while older ones have keys like "2013_177". In the second case the underscore is significant, it indicates that this was the 177th biggest song released in 2013. Now, of course, during the year no one knows where a particular song will rank, so the underscore names can't be assigned until every song from a particular year has dropped out of the charts, so recent records are temporarily assigned a name with a dash. In about May of the following year the rankings are calculated and the final identifiers are assigned. That is why we at the Turret can only grab this data retrospectively. Attributes The original spreadsheet has a number of attributes, we have limited our attention to just a few of them: 134 9 The songs with the most entries on the chart were White Christmas (with 33 versions and a total of 110 weeks) and Stardust (with 19 and a total of 106 weeks). position The peak position that songs reached in the charts should show an smooth curve from number one down to the lowest position. This chart has more songs in the lower peak positions than one would expect. Before 1991 the profile of peak positions was exactly as you would expect, that year Billboard introduced the concept of "Recurrent" tracks, that is they removed any track from the chart which had spent more than twenty weeks in the chart and had fallen to the lower positions. weeks The effect of the "Recurrent" process, by which tracks are removed if they have spent at least twenty weeks in the chart and have fallen to the lower reaches, can clearly be seen in the strange spike in this attribute. This "adjustment" was intended to promote newer songs and ensure the chart does not become "stale". In fact since it was introduced in 1991 the length of long chart runs has increased, this might reflect the more conscious efforts of record companies to "game" the charts by controlling release times and promotions, or it coul
  • Loss: AdaptiveLayerLoss with these parameters:
    {
        "loss": "GISTEmbedLoss",
        "n_layers_per_step": -1,
        "last_layer_weight": 1.5,
        "prior_layers_weight": 0.15,
        "kl_div_weight": 2,
        "kl_temperature": 2
    }
    

quora_pairs

  • Dataset: quora_pairs at 451a485
  • Size: 4,000 training samples
  • Columns: sentence1 and sentence2
  • Approximate statistics based on the first 1000 samples:
    sentence1 sentence2
    type string string
    details
    • min: 6 tokens
    • mean: 13.53 tokens
    • max: 42 tokens
    • min: 6 tokens
    • mean: 13.68 tokens
    • max: 43 tokens
  • Samples:
    sentence1 sentence2
    Astrology: I am a Capricorn Sun Cap moon and cap rising...what does that say about me? I'm a triple Capricorn (Sun, Moon and ascendant in Capricorn) What does this say about me?
    How can I be a good geologist? What should I do to be a great geologist?
    How do I read and find my YouTube comments? How can I see all my Youtube comments?
  • Loss: AdaptiveLayerLoss with these parameters:
    {
        "loss": "GISTEmbedLoss",
        "n_layers_per_step": -1,
        "last_layer_weight": 1.5,
        "prior_layers_weight": 0.15,
        "kl_div_weight": 2,
        "kl_temperature": 2
    }
    

gooaq_pairs

  • Dataset: gooaq_pairs at b089f72
  • Size: 6,500 training samples
  • Columns: sentence1 and sentence2
  • Approximate statistics based on the first 1000 samples:
    sentence1 sentence2
    type string string
    details
    • min: 8 tokens
    • mean: 11.6 tokens
    • max: 21 tokens
    • min: 13 tokens
    • mean: 57.74 tokens
    • max: 127 tokens
  • Samples:
    sentence1 sentence2
    is toprol xl the same as metoprolol? Metoprolol succinate is also known by the brand name Toprol XL. It is the extended-release form of metoprolol. Metoprolol succinate is approved to treat high blood pressure, chronic chest pain, and congestive heart failure.
    are you experienced cd steve hoffman? The Are You Experienced album was apparently mastered from the original stereo UK master tapes (according to Steve Hoffman - one of the very few who has heard both the master tapes and the CDs produced over the years). ... The CD booklets were a little sparse, but at least they stayed true to the album's original design.
    how are babushka dolls made? Matryoshka dolls are made of wood from lime, balsa, alder, aspen, and birch trees; lime is probably the most common wood type. ... After cutting, the trees are stripped of most of their bark, although a few inner rings of bark are left to bind the wood and keep it from splitting.
  • Loss: AdaptiveLayerLoss with these parameters:
    {
        "loss": "GISTEmbedLoss",
        "n_layers_per_step": -1,
        "last_layer_weight": 1.5,
        "prior_layers_weight": 0.15,
        "kl_div_weight": 2,
        "kl_temperature": 2
    }
    

Evaluation Datasets

nli-pairs

  • Dataset: nli-pairs at d482672
  • Size: 750 evaluation samples
  • Columns: anchor and positive
  • Approximate statistics based on the first 1000 samples:
    anchor positive
    type string string
    details
    • min: 5 tokens
    • mean: 17.61 tokens
    • max: 51 tokens
    • min: 4 tokens
    • mean: 9.71 tokens
    • max: 29 tokens
  • Samples:
    anchor positive
    Two women are embracing while holding to go packages. Two woman are holding packages.
    Two young children in blue jerseys, one with the number 9 and one with the number 2 are standing on wooden steps in a bathroom and washing their hands in a sink. Two kids in numbered jerseys wash their hands.
    A man selling donuts to a customer during a world exhibition event held in the city of Angeles A man selling donuts to a customer.
  • Loss: AdaptiveLayerLoss with these parameters:
    {
        "loss": "GISTEmbedLoss",
        "n_layers_per_step": -1,
        "last_layer_weight": 1.5,
        "prior_layers_weight": 0.15,
        "kl_div_weight": 2,
        "kl_temperature": 2
    }
    

scitail-pairs-pos

  • Dataset: scitail-pairs-pos at 0cc4353
  • Size: 750 evaluation samples
  • Columns: sentence1, sentence2, and label
  • Approximate statistics based on the first 1000 samples:
    sentence1 sentence2 label
    type string string int
    details
    • min: 5 tokens
    • mean: 22.43 tokens
    • max: 61 tokens
    • min: 8 tokens
    • mean: 15.3 tokens
    • max: 36 tokens
    • 0: ~50.00%
    • 1: ~50.00%
  • Samples:
    sentence1 sentence2 label
    An introduction to atoms and elements, compounds, atomic structure and bonding, the molecule and chemical reactions. Replace another in a molecule happens to atoms during a substitution reaction. 0
    Wavelength The distance between two consecutive points on a sinusoidal wave that are in phase; Wavelength is the distance between two corresponding points of adjacent waves called. 1
    humans normally have 23 pairs of chromosomes. Humans typically have 23 pairs pairs of chromosomes. 1
  • Loss: AdaptiveLayerLoss with these parameters:
    {
        "loss": "GISTEmbedLoss",
        "n_layers_per_step": -1,
        "last_layer_weight": 1.5,
        "prior_layers_weight": 0.15,
        "kl_div_weight": 2,
        "kl_temperature": 2
    }
    

qnli-contrastive

  • Dataset: qnli-contrastive at bcdcba7
  • Size: 750 evaluation samples
  • Columns: sentence1, sentence2, and label
  • Approximate statistics based on the first 1000 samples:
    sentence1 sentence2 label
    type string string int
    details
    • min: 6 tokens
    • mean: 14.15 tokens
    • max: 36 tokens
    • min: 4 tokens
    • mean: 36.98 tokens
    • max: 225 tokens
    • 0: 100.00%
  • Samples:
    sentence1 sentence2 label
    What came into force after the new constitution was herald? As of that day, the new constitution heralding the Second Republic came into force. 0
    What is the first major city in the stream of the Rhine? The most important tributaries in this area are the Ill below of Strasbourg, the Neckar in Mannheim and the Main across from Mainz. 0
    What is the minimum required if you want to teach in Canada? In most provinces a second Bachelor's Degree such as a Bachelor of Education is required to become a qualified teacher. 0
  • Loss: AdaptiveLayerLoss with these parameters:
    {
        "loss": "OnlineContrastiveLoss",
        "n_layers_per_step": -1,
        "last_layer_weight": 1.5,
        "prior_layers_weight": 0.15,
        "kl_div_weight": 2,
        "kl_temperature": 2
    }
    

Training Hyperparameters

Non-Default Hyperparameters

  • eval_strategy: steps
  • per_device_train_batch_size: 28
  • per_device_eval_batch_size: 18
  • learning_rate: 2e-05
  • weight_decay: 5e-07
  • num_train_epochs: 2
  • lr_scheduler_type: cosine_with_restarts
  • lr_scheduler_kwargs: {'num_cycles': 3}
  • warmup_ratio: 0.25
  • save_safetensors: False
  • fp16: True
  • push_to_hub: True
  • hub_model_id: bobox/DeBERTaV3-small-SenTra-AdaptiveLayers-AllSoft-HighTemp-n
  • hub_strategy: checkpoint
  • batch_sampler: no_duplicates

All Hyperparameters

Click to expand
  • overwrite_output_dir: False
  • do_predict: False
  • eval_strategy: steps
  • prediction_loss_only: True
  • per_device_train_batch_size: 28
  • per_device_eval_batch_size: 18
  • per_gpu_train_batch_size: None
  • per_gpu_eval_batch_size: None
  • gradient_accumulation_steps: 1
  • eval_accumulation_steps: None
  • learning_rate: 2e-05
  • weight_decay: 5e-07
  • adam_beta1: 0.9
  • adam_beta2: 0.999
  • adam_epsilon: 1e-08
  • max_grad_norm: 1.0
  • num_train_epochs: 2
  • max_steps: -1
  • lr_scheduler_type: cosine_with_restarts
  • lr_scheduler_kwargs: {'num_cycles': 3}
  • warmup_ratio: 0.25
  • warmup_steps: 0
  • log_level: passive
  • log_level_replica: warning
  • log_on_each_node: True
  • logging_nan_inf_filter: True
  • save_safetensors: False
  • 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: 42
  • data_seed: None
  • jit_mode_eval: False
  • use_ipex: False
  • bf16: False
  • fp16: True
  • 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: False
  • 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}
  • deepspeed: None
  • label_smoothing_factor: 0.0
  • optim: adamw_torch
  • 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: True
  • resume_from_checkpoint: None
  • hub_model_id: bobox/DeBERTaV3-small-SenTra-AdaptiveLayers-AllSoft-HighTemp-n
  • hub_strategy: checkpoint
  • hub_private_repo: False
  • hub_always_push: False
  • gradient_checkpointing: False
  • gradient_checkpointing_kwargs: None
  • include_inputs_for_metrics: False
  • 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
  • dispatch_batches: None
  • split_batches: None
  • include_tokens_per_second: False
  • include_num_input_tokens_seen: False
  • neftune_noise_alpha: None
  • optim_target_modules: None
  • batch_eval_metrics: False
  • batch_sampler: no_duplicates
  • multi_dataset_batch_sampler: proportional

Training Logs

Epoch Step Training Loss nli-pairs loss qnli-contrastive loss scitail-pairs-pos loss sts-test_spearman_cosine
0.0503 141 8.9676 - - - -
0.1006 282 7.1505 - - - -
0.1510 423 6.5458 - - - -
0.2002 561 - 5.2347 5.0924 3.0123 -
0.2013 564 5.2862 - - - -
0.2516 705 4.31 - - - -
0.3019 846 3.904 - - - -
0.3522 987 3.3312 - - - -
0.4004 1122 - 2.8461 3.3017 1.8818 -
0.4026 1128 3.4991 - - - -
0.4529 1269 3.305 - - - -
0.5032 1410 3.0787 - - - -
0.5535 1551 3.0456 - - - -
0.6006 1683 - 2.2152 2.6721 1.5917 -
0.6039 1692 2.886 - - - -
0.6542 1833 2.9191 - - - -
0.7045 1974 2.7596 - - - -
0.7548 2115 3.0015 - - - -
0.8009 2244 - 1.9579 2.1218 1.4780 -
0.8051 2256 2.6781 - - - -
0.8555 2397 2.6899 - - - -
0.9058 2538 2.5374 - - - -
0.9561 2679 2.9215 - - - -
1.0011 2805 - 1.8911 2.2687 1.4384 -
1.0064 2820 2.9894 - - - -
1.0567 2961 2.67 - - - -
1.1071 3102 2.7954 - - - -
1.1574 3243 2.6645 - - - -
1.2013 3366 - 1.8474 2.0989 1.3498 -
1.2077 3384 2.5357 - - - -
1.2580 3525 2.099 - - - -
1.3084 3666 2.2678 - - - -
1.3587 3807 2.0013 - - - -
1.4015 3927 - 1.7168 1.8503 1.2985 -
1.4090 3948 2.2268 - - - -
1.4593 4089 2.2645 - - - -
1.5096 4230 1.8598 - - - -
1.5600 4371 2.1624 - - - -
1.6017 4488 - 1.7209 1.6492 1.2805 -
1.6103 4512 2.0678 - - - -
1.6606 4653 2.1483 - - - -
1.7109 4794 2.2059 - - - -
1.7612 4935 2.3824 - - - -
1.8019 5049 - 1.6013 1.6620 1.2365 -
1.8116 5076 2.1792 - - - -
1.8619 5217 2.1 - - - -
1.9122 5358 2.1818 - - - -
1.9625 5499 2.6552 - - - -
2.0 5604 - - - - 0.7592

Framework Versions

  • Python: 3.10.13
  • Sentence Transformers: 3.0.1
  • Transformers: 4.41.2
  • PyTorch: 2.1.2
  • Accelerate: 0.30.1
  • Datasets: 2.19.2
  • Tokenizers: 0.19.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",
}

AdaptiveLayerLoss

@misc{li20242d,
    title={2D Matryoshka Sentence Embeddings}, 
    author={Xianming Li and Zongxi Li and Jing Li and Haoran Xie and Qing Li},
    year={2024},
    eprint={2402.14776},
    archivePrefix={arXiv},
    primaryClass={cs.CL}
}

CoSENTLoss

@online{kexuefm-8847,
    title={CoSENT: A more efficient sentence vector scheme than Sentence-BERT},
    author={Su Jianlin},
    year={2022},
    month={Jan},
    url={https://kexue.fm/archives/8847},
}

GISTEmbedLoss

@misc{solatorio2024gistembed,
    title={GISTEmbed: Guided In-sample Selection of Training Negatives for Text Embedding Fine-tuning}, 
    author={Aivin V. Solatorio},
    year={2024},
    eprint={2402.16829},
    archivePrefix={arXiv},
    primaryClass={cs.LG}
}