Esse repositório contém meus experimentos com fine tuning do MedGemma 4B, baseada nos notebooks Colab fornecidos pelo Google. Meu objetivo é ajustar o modelo original para ser capaz de distinguir entre imagens com calcificação e massa em imagens de mamografia.
Primeiro eu mofiquei o notebook 'quick_start_with_hugging_face' para avaliar se o MedGemma conseguiria identificar achados em mamografias. Então eu utilizei o dataset CBIS-DDSM para ajustar o modelo ('fine_tune_with_hugging_face') e por fim testei ('testing_fine_tuned_model') a resposta do novo modelo em imagens da web.
Obs: O github não está renderizando corretamente os notebooks eles podem ser visualizados e executados no Google Colab nos seguintes links:
Quick Start with Hugging Face
Fine Tune with Hugging Face
Testing Fine Tuned Model
Google Colab units are provided for this project. #AISprint