Views
No views yet

Disease, Disability, ClinicalEntity, Chemical... Medical MT5-xl-multitask has been finetuned for English, Spanish, French and Italian, although it may work with a wide range of languages.
| HiTZ/Medical-mT5-large | HiTZ/Medical-MT5-xl | HiTZ/Medical-mT5-large-multitask | HiTZ/Medical-MT5-xl-multitask | |
|---|---|---|---|---|
| Param. no. | 738M | 3B | 738M | 3B |
| Task | Language Modeling | Language Modeling | Multitask Sequence Labeling | Multitask Sequence Labeling |
<Disease> Torsade de pointes ventricular tachycardia during low dose intermittent dobutamine treatment in a patient with dilated cardiomyopathy and congestive heart failure .1import torch
2from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
3
4model = AutoModelForSeq2SeqLM.from_pretrained("Medical-MT5-xl-multitask",torch_dtype=torch.bfloat16, device_map="auto")
5tokenizer = AutoTokenizer.from_pretrained("Medical-MT5-xl-multitask")
6
7input_example = "<Disease> Torsade de pointes ventricular tachycardia during low dose intermittent dobutamine treatment in a patient with dilated cardiomyopathy and congestive heart failure ."
8
9model_input = tokenizer(input_example, return_tensors="pt")
10
11output = model.generate(**model_input.to(model.device),max_new_tokens=128,num_beams=1,num_return_sequences=1,do_sample=False)
12
13print(tokenizer.decode(output[0], skip_special_tokens=True))
1@misc{garcíaferrero2024medical,
2 title={Medical mT5: An Open-Source Multilingual Text-to-Text LLM for The Medical Domain},
3 author={Iker García-Ferrero and Rodrigo Agerri and Aitziber Atutxa Salazar and Elena Cabrio and Iker de la Iglesia and Alberto Lavelli and Bernardo Magnini and Benjamin Molinet and Johana Ramirez-Romero and German Rigau and Jose Maria Villa-Gonzalez and Serena Villata and Andrea Zaninello},
4 year={2024},
5 eprint={2404.07613},
6 archivePrefix={arXiv},
7 primaryClass={cs.CL}
8}