SentenceTransformer based on google/embeddinggemma-300m

This is a sentence-transformers model finetuned from google/embeddinggemma-300m. 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 Type: Sentence Transformer
• Base model: google/embeddinggemma-300m
• Maximum Sequence Length: 2048 tokens
• Output Dimensionality: 768 dimensions
• Similarity Function: Cosine Similarity

Model Sources

• Documentation: Sentence Transformers Documentation
• Repository: Sentence Transformers on GitHub
• Hugging Face: Sentence Transformers on Hugging Face

Full Model Architecture

SentenceTransformer(
  (0): Transformer({'max_seq_length': 2048, 'do_lower_case': False, 'architecture': 'Gemma3TextModel'})
  (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})
  (2): Dense({'in_features': 768, 'out_features': 3072, 'bias': False, 'activation_function': 'torch.nn.modules.linear.Identity'})
  (3): Dense({'in_features': 3072, 'out_features': 768, 'bias': False, 'activation_function': 'torch.nn.modules.linear.Identity'})
  (4): Normalize()
)

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("ayushexel/xynft-embeddinggemma-300m-5e")
# Run inference
queries = [
    "What is the critical issue with the ICICI payment gateway, and what are the potential consequences and mitigation strategies being discussed?",
]
documents = [
    'The critical issue is the sustained `BANK_TIMEOUT` errors occurring with the ICICI payment gateway, which are identified as the primary driver of a business loss. If the engineering team cannot provide an ETA for a fix, the proposed mitigation strategy is to temporarily deprioritize this payment method on the checkout page to protect the user experience and the overall payment success rate.',
    "To mitigate the impact on merchants from the ongoing UPI ICICI gateway latency spike and high 'Transaction Not Permitted' errors, two immediate actions are proposed. First, Rahul Verma is asked to explore temporarily de-prioritizing the ICICI gateway in the routing logic for high-volume merchants until a confirmation on the issue is received from the bank. Second, the team needs to investigate the severity of the issue by checking Grafana to confirm if the p99 latency for the gateway has breached the Service Level Agreement (SLA).",
    "The initial analysis indicates that the latency spike is not a failure at the SDK or app-level. The issue is believed to originate from a downstream system, pointing specifically to a problem either in Juspay's backend communication with ICICI's systems or within the ICICI gateway itself. The critical implication of this finding is that it allows the event to be classified as a 'specific partner gateway degradation' rather than a failure of Juspay's core payment system, which is a key distinction for any required regulatory reporting to the RBI.",
]
query_embeddings = model.encode_query(queries)
document_embeddings = model.encode_document(documents)
print(query_embeddings.shape, document_embeddings.shape)
# [1, 768] [3, 768]

# Get the similarity scores for the embeddings
similarities = model.similarity(query_embeddings, document_embeddings)
print(similarities)
# tensor([[0.6159, 0.3319, 0.2661]])

Evaluation

Metrics

Triplet

• Datasets: baseline_pre_training and ft-dev
• Evaluated with TripletEvaluator

Training Details

Training Dataset

Unnamed Dataset

• Size: 741 training samples
• Columns: query, answer, and negative
• Approximate statistics based on the first 741 samples:

  	query	answer	negative
  type	string	string	string
  details	min: 20 tokensmean: 31.37 tokensmax: 56 tokens	min: 61 tokensmean: 126.29 tokensmax: 228 tokens	min: 69 tokensmean: 124.72 tokensmax: 222 tokens

• Samples:

  query	answer	negative
  What is the root cause of the data loss identified in JIRA-DATA-781, and what is the proposed plan to both fix the issue and recover the missing data?	The root cause is a schema change in the xyne-core service (v2.9.1) where the metadata field in the transaction payload was changed from a flat JSON to a nested object. The current BigQuery ingestion pipeline's dataflow job cannot parse this new structure, causing a TypeError that leads to a silent 4% data loss. The proposed plan involves Rahul Verma developing a patch for the dataflow job to handle the nested structure, which is expected by EOD. Additionally, a backfill process will be planned to re-process the raw logs from the affected period to recover the lost data, with a meeting to discuss the backfill strategy scheduled for the following day.	Aishwarya is encountering an intermittent TypeError in a Python script designed for S3 data reconciliation. The issue stems from a change in the 'transaction timestamp' column's format. While she has attempted a fix by adjusting the datetime parsing function on the branch feature/BREEZE-451-recon-fix, the problem persists. Her hypotheses for the root cause of these intermittent failures include a more complex data transformation logic error she might be missing, the possibility of mixed timestamp formats within the data, or a potential connection to recent IAM role changes for S3 access.
  What are the specific documentation and compliance actions being taken in response to the UPI Juspay latency spike, and what is the potential consequence if the issue persists?	The latency spike is being formally documented as a 'major service degradation event' for the quarterly RBI filing. An incident report has been created on Confluence (doc ID: COMP-2023-Q4-PSP-01), and all technical findings and communication logs must be attached to Jira ticket #PAY-4812 to maintain a critical audit trail. If the issue is not resolved, the company may need to trigger a penalty clause.	The investigation into the UPI 'Payment Pending' latency spike is being formally tracked under JIRA ticket OPS-1724. The immediate plan involves Siddharth Nair pulling detailed transaction logs from BigQuery for the specific merchant_id affected between 10:00 and 11:00 AM today. This issue is considered a high priority as it is impacting a key merchant. The team, including Pooja Joshi and Riya Kapoor, will convene for a sync-up call after lunch to review the findings from the logs.
  What immediate actions are being taken to address the latency spike on the ICICI payment gateway, and how will the success of these actions be validated?	To address the latency spike, a critical Jira ticket (BREEZE-4812) has been raised for the SRE team to immediately update the BGP routing configuration, effectively deprioritizing the problematic ISP for traffic from the 'payment-gateway' service. The effectiveness of this fix will be validated by a new, dedicated Grafana dashboard, which Riya Kapoor will create to specifically monitor the ICICI net banking success rate for the affected 103.22.x.x IP range once the network changes are deployed via the CI/CD pipeline.	Meera Reddy has proposed a multi-step plan to address the ICICI API latency spike. The plan involves asking the engineering team to investigate if specific transaction types (like subscriptions vs. one-time payments) are disproportionately affected. Concurrently, a consolidated report with failed transaction IDs and latency graphs will be prepared. Finally, Meera will escalate the issue by raising a high-priority ticket with the ICICI support team by the end of the day.

• Loss: MultipleNegativesRankingLoss with these parameters:

  {
      "scale": 20.0,
      "similarity_fct": "cos_sim",
      "gather_across_devices": false
  }
  

Training Hyperparameters

Non-Default Hyperparameters

• eval_strategy: steps
• per_device_train_batch_size: 128
• per_device_eval_batch_size: 128
• learning_rate: 3e-05
• weight_decay: 0.01
• num_train_epochs: 5
• warmup_ratio: 0.1
• bf16: True
• batch_sampler: no_duplicates

All Hyperparameters

• overwrite_output_dir: False
• do_predict: False
• eval_strategy: steps
• prediction_loss_only: True
• per_device_train_batch_size: 128
• per_device_eval_batch_size: 128
• 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: 3e-05
• weight_decay: 0.01
• adam_beta1: 0.9
• adam_beta2: 0.999
• adam_epsilon: 1e-08
• max_grad_norm: 1.0
• num_train_epochs: 5
• 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: 42
• 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: 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}
• 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: no_duplicates
• multi_dataset_batch_sampler: proportional
• router_mapping: {}
• learning_rate_mapping: {}

Training Logs

Framework Versions

• Python: 3.12.3
• Sentence Transformers: 5.1.0
• Transformers: 4.56.1
• PyTorch: 2.8.0+cu128
• Accelerate: 1.10.1
• Datasets: 4.0.0
• Tokenizers: 0.22.0

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

MultipleNegativesRankingLoss

@misc{henderson2017efficient,
    title={Efficient Natural Language Response Suggestion for Smart Reply},
    author={Matthew Henderson and Rami Al-Rfou and Brian Strope and Yun-hsuan Sung and Laszlo Lukacs and Ruiqi Guo and Sanjiv Kumar and Balint Miklos and Ray Kurzweil},
    year={2017},
    eprint={1705.00652},
    archivePrefix={arXiv},
    primaryClass={cs.CL}
}