Doctor-OPT-350M

This model is a fine-tuned version of facebook/opt-350m on the sid6i7/patient-doctor.

Model description

This is an LLM trained on patient-doctor dialogues, mostly comprised of doctors giving medical advice to patients.

Intended uses & limitations

This model is intended for research on the capabilities of LLMs. Any output from this LLM is not to be treated as legitimate medical advice.

The model works best when given input in the format of the dataset it was trained on. Here is an example of the format:

Patient: Please can you help me with thoughts about its. Where my gut gives out spasms and. Uncontrol movements like pulsating 24,7. It seems. Worst when stressed I would like something that regards relaxing the bowel to get some more pleasures in life. Many thanks. XXXX

Doctor: <start> Hello, Continuous abdominal spasms can be due to different reasons. Is there any intense pain present? Are you on any medication?It might be related to food habits. Try consuming food that is rich in fibers.

Sample Use

from transformers import pipeline

generator = pipeline('text-generation', model='linkanjarad/Doctor-OPT-350M')

# Format the prompt to match the dataset formatting
prompt = '''Patient: Hi, I'm 30 years old. Recently, I've been having abdominal pain and discomfort after meals. It feels like a burning sensation. What could be the possible causes?

Doctor: <start>'''

output = generator(prompt, do_sample=True, min_length=10, max_length=50, temperature=0.5)
print(output[0]['generated_text'])

Sample Outputs

Patient: "I'm a 40-year-old male with a family history of heart disease. Lately, I've been experiencing chest discomfort and shortness of breath during physical activity. Are these symptoms indicative of a heart problem, and what tests should I undergo to evaluate my heart health?

Doctor: <start> Hello,The symptoms you have mentioned are suggestive of cardiac problems. These symptoms are also possible due to high blood pressure and cholesterol. I would recommend performing physical activity such as walking, rollerblading, swimming, etc. to increase your activity levels.I would also recommend doing an ultrasound scan of the heart to determine if there is a risk of a heart problem. I would also recommend doing a blood test to

Patient: I'm feeling this pain in my chest when I see this particular person. My breath feels shortened and I feel nervous, and I get a tingly sensation. My eyes seem to follow the person. What could this be?

Doctor: <start> Hi,
 Thanks for writing in.
 The symptoms you have described suggest to be related to some sort of pulmonary infections. I would advise you to undergo a check up by a physician of the chest and abdomen. In addition, you should go for a chest X-ray and a chest and abdomen MRI.

Patient: Hello, I am a 20 year old female and have been getting stabbing pains on the lower right side of my abdomen everyday for about 4 weeks now. I know it cant be my appendix for it was removed in 2010. I have noticed that above the scar of my appendix removal the area seems risen, and beneath the scar the area is flat. Should i be worried and seek further help?

Doctor: <start> Hi dear,Thanks for writing in.The pain in the upper right side of the abdomen may be because of some sort of appendicitis. Some times it can be due to a condition called anemia. If there is anemia and the pain is not related to anemia, then chances of appendicitis are low. I would suggest you to consult an emic specialist and get a urine and blood test done. Anemia can be treated in a few ways. For example, erythrocyte sedimentation rate (ESR), blood protein level, and a few more tests like CT scan of the abdomen and kidney are important but the actual diagnosis is not always confirmed. I hope my answer will help you.Wish you all the best.