chaoyi-wu / RadFM

The official code for "Towards Generalist Foundation Model for Radiology by Leveraging Web-scale 2D&3D Medical Data".
315 stars 32 forks source link

Can the model work beyond the 5000 diseases? #14

Open hzhz2020 opened 7 months ago

hzhz2020 commented 7 months ago

Dear Author

Thanks for the exciting work! It's amazing that the model has been trained on and thus might cover over 5000 diseases. I am wondering, since your goal is building a generalist foundation model for radiology, does the model work beyond the 5000 diseases that it is trained on?

For example, i work with a heart disease that is diagnose via ultrasound (which is a modality covered in your model). Can i ask the diagnosis question about this disease to the model?

Thanks!

chaoyi-wu commented 7 months ago

It depends. Please check our latest version paper where we test the generality on PadChest to see how model will work on unseen diseases classes. For some diseases it works but some are just random guessing. But, in general, I have to acknowledge that the zero-shot performance on unseen classes is not so promising, especially compared with the specifically trained diagnosis models, I tend to recommend you to view our model as a pre-trained backbone more instead of using it as a zero-shot generative model in this unseen case.

hzhz2020 commented 7 months ago

Dear Author

Thank you for your prompt reply, what you said make sense! Looking forward to more of the exciting work on this space!