Saved in:
| Main Authors: | , , , , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2024
|
| Subjects: | |
| Online Access: | https://arxiv.org/abs/2411.02372 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866917941961818112 |
|---|---|
| author | Dey, Neel Billot, Benjamin Wong, Hallee E. Wang, Clinton J. Ren, Mengwei Grant, P. Ellen Dalca, Adrian V. Golland, Polina |
| author_facet | Dey, Neel Billot, Benjamin Wong, Hallee E. Wang, Clinton J. Ren, Mengwei Grant, P. Ellen Dalca, Adrian V. Golland, Polina |
| contents | Current volumetric biomedical foundation models struggle to generalize as public 3D datasets are small and do not cover the broad diversity of medical procedures, conditions, anatomical regions, and imaging protocols. We address this by creating a representation learning method that instead anticipates strong domain shifts at training time itself. We first propose a data engine that synthesizes highly variable training samples that would enable generalization to new biomedical contexts. To then train a single 3D network for any voxel-level task, we develop a contrastive learning method that pretrains the network to be stable against nuisance imaging variation simulated by the data engine, a key inductive bias for generalization. This network's features can be used as robust representations of input images for downstream tasks and its weights provide a strong, dataset-agnostic initialization for finetuning on new datasets. As a result, we set new standards across both multimodality registration and few-shot segmentation, a first for any 3D biomedical vision model, all without (pre-)training on any existing dataset of real images. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2411_02372 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Learning General-Purpose Biomedical Volume Representations using Randomized Synthesis Dey, Neel Billot, Benjamin Wong, Hallee E. Wang, Clinton J. Ren, Mengwei Grant, P. Ellen Dalca, Adrian V. Golland, Polina Computer Vision and Pattern Recognition Machine Learning Current volumetric biomedical foundation models struggle to generalize as public 3D datasets are small and do not cover the broad diversity of medical procedures, conditions, anatomical regions, and imaging protocols. We address this by creating a representation learning method that instead anticipates strong domain shifts at training time itself. We first propose a data engine that synthesizes highly variable training samples that would enable generalization to new biomedical contexts. To then train a single 3D network for any voxel-level task, we develop a contrastive learning method that pretrains the network to be stable against nuisance imaging variation simulated by the data engine, a key inductive bias for generalization. This network's features can be used as robust representations of input images for downstream tasks and its weights provide a strong, dataset-agnostic initialization for finetuning on new datasets. As a result, we set new standards across both multimodality registration and few-shot segmentation, a first for any 3D biomedical vision model, all without (pre-)training on any existing dataset of real images. |
| title | Learning General-Purpose Biomedical Volume Representations using Randomized Synthesis |
| topic | Computer Vision and Pattern Recognition Machine Learning |
| url | https://arxiv.org/abs/2411.02372 |