---
language:
- en
license: mit
tags:
- sentence-similarity
base_model: BAAI/bge-small-en-v1.5
widget:
- source_sentence: >-
    Infections affecting the acromioclavicular joint can present with symptoms
    such as shoulder pain, swelling, and potential fever. Patients may
    experience systemic signs of infection, including chills and malaise,
    alongside localized symptoms related to joint movement.
  sentences:
  - >-
    Gastric stromal tumors with uncertain behavior can present with symptoms
    such as abdominal pain, nausea, and changes in appetite. Patients may
    experience psychological distress due to concerns about malignancy and the
    need for further evaluation. The presence of these tumors can lead to
    complications depending on their size and location.
  - 'Disease Name : Complication, umbrella device, vascular, embolism'
  - 'Disease Name : Infection, infected, infective, acromioclavicular'
- source_sentence: >-
    cyanosis: a bluish discoloration of the skin; fatigue: extreme tiredness;
    shortness of breath: difficulty in breathing; headache: pain in the head;
    confusion: difficulty in thinking clearly.
  sentences:
  - 'Disease Name : Anomaly, anomalous, spine, spinal NEC, column NEC, kyphosis'
  - 'Disease Name : Methemoglobinemia, Hb M disease'
  - >-
    Superficial insect bites on the finger can lead to localized swelling,
    redness, and itching. Patients may experience discomfort and potential
    allergic reactions, requiring symptomatic treatment.
- source_sentence: 'Disease Name : Excess, excessive, excessively, crying, in infant'
  sentences:
  - >-
    Excessive crying in infants can manifest as prolonged periods of distress,
    often accompanied by signs of discomfort such as clenching fists, arching of
    the back, and difficulty in soothing. The infant may exhibit a high-pitched
    cry and may seem inconsolable, leading to parental concern and anxiety about
    the child's well-being.
  - 'Disease Name : Syphilis, syphilitic, meningitis'
  - 'Disease Name : Prolapse, prolapsed, ileostomy bud'
- source_sentence: >-
    Cysts affecting the lacrimal passages or sac can lead to symptoms such as
    tearing, swelling, and discomfort in the inner corner of the eye. Patients
    may experience recurrent infections, redness, and a sensation of fullness,
    which can significantly impact tear drainage and overall eye health.
  sentences:
  - >-
    A condition characterized by the fusion of a single suture can lead to
    various physical deformities and developmental challenges. Symptoms may
    include cognitive impairment and various physical challenges that can impact
    daily life.
  - 'Disease Name : Cyst, Lacrimal, Passages or Sac'
  - >-
    This condition presents with swelling and pain in the area of the placenta,
    often accompanied by tenderness and discomfort. Patients may experience
    complications during pregnancy, including bleeding and changes in fetal
    movement. In some cases, there may be associated risks to both the mother
    and the fetus.
- source_sentence: 'Disease Name : Tuberculosis, tubercular, tuberculous, brain'
  sentences:
  - >-
    hoarseness: abnormal voice quality due to vocal cord dysfunction; difficulty
    breathing: shortness of breath or stridor from airway compression; loss of
    voice: inability to speak due to vocal cord paralysis; throat discomfort:
    pain or irritation in the throat area.
  - >-
    Symptoms include cough, wheezing, and difficulty breathing, often
    exacerbated by exposure to specific environmental triggers such as air
    conditioning systems. Patients may also experience fever and malaise, with a
    history of allergic reactions or asthma. The condition can lead to
    significant respiratory distress if not managed appropriately.
  - >-
    Symptoms often include headaches, confusion, and potential neurological
    deficits, along with systemic signs such as fever and fatigue. Patients may
    also experience seizures or altered consciousness, indicating a serious
    underlying condition.
pipeline_tag: sentence-similarity
datasets:
- SalmanFaroz/DisEmbed-Symptom-Disease-v1
---

# DisEmbed (Disease Embedding)
DisEmbed-v1 is a disease-focused embedding model designed for the medical domain, trained on a synthetic dataset comprising disease descriptions, symptoms, and Q&A pairs. It outperforms general medical models in disease-specific tasks, particularly in distinguishing similar diseases. DisEmbed excels in retrieval task and disease-context identification.


## Model Details

### Model Description
- **Dataset : [DisEmbed-Symptom-Disease-v1](https://huggingface.co/datasets/SalmanFaroz/DisEmbed-Symptom-Disease-v1)**
- **Paper : [DisEmbed: Transforming Disease Understanding through Embeddings](https://arxiv.org/abs/2412.15258)**
- **Maximum Sequence Length:** 512 tokens
- **Output Dimensionality:** 384 dimensions
- **Similarity Function:** Cosine Similarity
- **Language:** en
- **License:** mit

![image/png](https://cdn-uploads.huggingface.co/production/uploads/631772607690c5b55e5b5edd/xqnT8KqYPmDIPTn08bksU.png)

## Usage

### Direct Usage (Sentence Transformers)

First install the Sentence Transformers library:

```bash
pip install -U sentence-transformers
```

Then you can load this model and run inference.
```python
from sentence_transformers import SentenceTransformer

model = SentenceTransformer("SalmanFaroz/DisEmbed-v1")
# Run inference
sentences = [
    'Chronic cough with blood-streaked sputum, severe night sweats, and unintentional weight loss.Painful breathing or chest pain, often worsened by coughing.Swelling in the neck or lymph nodes, and frequent fatigue.',
    'Asthma',
    'Tuberculosis'
]
embeddings = model.encode(sentences)
print(embeddings.shape)
# [3, 384]

# Get the similarity scores for the embeddings
similarities = model.similarity(embeddings, embeddings)
print(similarities)

```


Citation
```
@article{faroz2024disembed,
  title={DisEmbed: Transforming Disease Understanding through Embeddings},
  author={Faroz, Salman},
  journal={arXiv preprint arXiv:2412.15258},
  year={2024},
  doi={10.48550/arXiv.2412.15258},
  url={https://arxiv.org/abs/2412.15258}
}
```