|
| 1 | +--- |
| 2 | +layout: pw44-project |
| 3 | + |
| 4 | +permalink: /:path/ |
| 5 | + |
| 6 | +project_title: Exploration of foundation models and their embeddings for other tasks using the cloud |
| 7 | +category: Cloud / Web |
| 8 | +presenter_location: |
| 9 | + |
| 10 | +key_investigators: |
| 11 | + |
| 12 | +- name: Deepa Krishnaswamy |
| 13 | + affiliation: Brigham and Women's Hospital |
| 14 | + country: USA |
| 15 | + |
| 16 | +- name: Andrey Fedorov |
| 17 | + affiliation: Brigham and Women's Hospital |
| 18 | + country: USA |
| 19 | + |
| 20 | +--- |
| 21 | + |
| 22 | +# Project Description |
| 23 | + |
| 24 | +<!-- Add a short paragraph describing the project. --> |
| 25 | + |
| 26 | + |
| 27 | +The popularity and use of foundation models (FMs) have exploded in recent years. Within the medical imaging field alone, numerous models have been developed to support various downstream tasks, including classification and segmentation. |
| 28 | + |
| 29 | +However, as a user, it's hard to understand the embeddings that the models produce. Also, it's hard to figure out: |
| 30 | +1) which model to use, and |
| 31 | +2) what tasks the model supports. |
| 32 | + |
| 33 | +In this project, we plan to explore how the cloud can help us understand these embeddings from various FMs. Recently, we have extracted embeddings from lung cancer tumors in the National Lung Screening Trial (NLST) CT dataset from 9 different models. We will use the latest features in the Google Cloud Platform to help us explore and understand these embeddings. |
| 34 | + |
| 35 | +We are most interested in: |
| 36 | +1) how these embeddings can be visualized, and if clusters are visible, |
| 37 | +2) if these embeddings can be used to find similar patients, and |
| 38 | +3) if they can actually be used for other tasks |
| 39 | + |
| 40 | +Possible extensions to this work: |
| 41 | +1) We could explore the embeddings that Google has provided from their pathology foundation model [here](https://research.google/blog/health-specific-embedding-tools-for-dermatology-and-pathology/) and [here](https://github.com/Google-Health/imaging-research/tree/master/path-foundation). |
| 42 | +2) We could extend this exploration to lung ultrasound images, where visualizing the embedding space could help us choose representative and diverse images for expert annotation |
| 43 | + |
| 44 | +How could it relate to Slicer and Imaging Data Commons (IDC)? Given a sample patient image, we could retrieve the k-closest patients in IDC. |
| 45 | + |
| 46 | + |
| 47 | + |
| 48 | +## Objective |
| 49 | + |
| 50 | +<!-- Describe here WHAT you would like to achieve (what you will have as end result). --> |
| 51 | + |
| 52 | + |
| 53 | +1. We will figure out how to store these embeddings in the cloud to enable a quick search and comparison. |
| 54 | +2. Next, we will explore and visualize these embeddings. |
| 55 | +3. Then, we will use these embeddings to perform image retrieval -- finding similar patients in the NLST collection. |
| 56 | +4. Lastly, we will show how to use these embeddings for a downstream task. |
| 57 | + |
| 58 | + |
| 59 | + |
| 60 | + |
| 61 | +## Approach and Plan |
| 62 | + |
| 63 | +<!-- Describe here HOW you would like to achieve the objectives stated above. --> |
| 64 | + |
| 65 | + |
| 66 | +1. We will see if Google Cloud Platform BigQuery (BQ) can be used to store the embeddings: https://docs.cloud.google.com/bigquery/docs/vector-search-intro and https://docs.cloud.google.com/bigquery/docs/vector-index |
| 67 | +2. Next, we will create an interactive plot to explore embeddings and any clustering in a low-dimensional space. We will let the user click on points to open up an OHIF link with the original image data. |
| 68 | +3. We will investigate whether vector search or a similarity search can be performed to find similar patients. |
| 69 | + |
| 70 | + |
| 71 | + |
| 72 | +## Progress and Next Steps |
| 73 | + |
| 74 | +<!-- Update this section as you make progress, describing of what you have ACTUALLY DONE. |
| 75 | + If there are specific steps that you could not complete then you can describe them here, too. --> |
| 76 | + |
| 77 | + |
| 78 | + |
| 79 | + |
| 80 | + |
| 81 | + |
| 82 | + |
| 83 | +# Illustrations |
| 84 | + |
| 85 | +<!-- Add pictures and links to videos that demonstrate what has been accomplished. --> |
| 86 | + |
| 87 | + |
| 88 | +Initial analysis (no cloud) of using embeddings for lung cancer histology classification: |
| 89 | +<img width="412" height="326" alt="Image" src="https://github.com/user-attachments/assets/bd9434d2-80bb-4309-9f74-5d5d18b319c8" /> |
| 90 | + |
| 91 | +Initial analysis (no cloud) of using embeddings for lung cancer staging classification: |
| 92 | +<img width="412" height="326" alt="Image" src="https://github.com/user-attachments/assets/4ddbf08d-e352-4c26-9c70-51d6bbbed3b9" /> |
| 93 | + |
| 94 | + |
| 95 | + |
| 96 | +# Background and References |
| 97 | + |
| 98 | +<!-- If you developed any software, include link to the source code repository. |
| 99 | + If possible, also add links to sample data, and to any relevant publications. --> |
| 100 | + |
| 101 | + |
| 102 | +_No response_ |
| 103 | + |
0 commit comments