Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Medical MVP: VQA+Captum demo #45

Open
andreimano opened this issue Jul 24, 2020 · 3 comments
Open

Medical MVP: VQA+Captum demo #45

andreimano opened this issue Jul 24, 2020 · 3 comments

Comments

@andreimano
Copy link
Collaborator

See #44 for more details about the dataset and network architecture.

The task is:

  1. Train a baseline VQA model with decent accuracy. Try using tools from MONAI as much as possible. (related to Medical MVP: MONAI research #40)
  2. Use Captum for interpretability. (related to Medical MVP: Integrate captum in the flow #35)

Nice to have:
3. Federalize the model and do the same Captum demo. (depends on #41)
4. Find a stronger architecture (maybe based on transformers) and make a demo with it.

@andreimano
Copy link
Collaborator Author

The authors of 1 suggest that 2 gives good results. The code is publicly available at 3, but the requirements are pretty high - it requires 64GB of memory (I'm assuming video) and 4 GPUs, according to the authors.

Where are we going to train our models? @bcebere @tudorcebere

@tudorcebere
Copy link
Collaborator

I can manage to provide access to a 2080ti and when it is really critical, I can request access to a tesla v100. I don't think I can get something close to 64VRAM tho.

@andreimano
Copy link
Collaborator Author

andreimano commented Jul 24, 2020

I was thinking about the AWS free credit that was listed on the hackaton page - I think that we could also use the Google Compute Engine 300$ free credit 1 if we wanna train something that needs 64Gb of VRAM - unfortunately I've already used my free GCE credit. I've opened a ticket for this (#46).

This issue also depends on #46 .

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants