Project-MONAI / MONAI

AI Toolkit for Healthcare Imaging
https://monai.io/
Apache License 2.0
5.73k stars 1.05k forks source link

Pre-trained weights for Transchex #3729

Closed maximek3 closed 2 years ago

maximek3 commented 2 years ago

Hi, are there any pre-trained weights available for Transchex? Thanks!

ahatamiz commented 2 years ago

Hi @maximek3

Thank you for your interest in our work. We have not released the pre-trained weights for TransChex. However, we have created an intuitive tutorial, with all steps including data pre-processing detailed. In addition, the tutorial can be trained using moderately-sized GPUs (<11G VRAM) and in a very reasonable amount of time. I recommend checking it out:

https://github.com/Project-MONAI/tutorials/tree/master/multimodal/openi_multilabel_classification_transchex

Thanks

maximek3 commented 2 years ago

Thanks for the pointer. I had a look at the tutorial already and it's very useful.

I just wondered whether the TransChex model was also pre-trained on a larger dataset. If yes, do you plan to release these weights eventually? Thanks!

ahatamiz commented 2 years ago

Thanks for the comment. Yes, in fact we have trained it on larger datasets (e.g. MIMIC III) and was able to validate its effectiveness. However, the licences of these datasets prohibit us from releasing these pre-trained weights. As such, we chose to create the above tutorial using a dataset which allows the usage in the context of our tutorial and can be trained in a reasonable time.

Thanks,