SYNTOLOGY HomeExplorerAtlasCodeMethodologyAboutDevelopersFeedPricing
Paper · 2304.13785 · 2023

Customized Segment Anything Model for Medical Image Segmentation

arXiv · PDF · Open in the Atlas

Code that ran

We lifted 9 functions out of this paper's own repositories and ran 6 of them in a sandbox. "Ran" means the function executed on a synthesized input and returned a value. It is not a reproduction of the paper's results.

RepositoryRoleRan
hitachinsk/samed canonical 6 of 9
FunctionStatusWhere it lives
calc_loss Ran hitachinsk/samed/SAMed_h/trainer.py
code served (permissive licence) · get_code("ad5471db7ab93cbf")
get_rel_pos Ran hitachinsk/samed/SAMed_h/segment_anything/modeling/image_encoder.py
code served (permissive licence) · get_code("733d7f0bedcb74c2")
random_rot_flip Ran hitachinsk/samed/SAMed_h/datasets/dataset_synapse.py
code served (permissive licence) · get_code("960b785e65a00358")
random_rotate Ran hitachinsk/samed/SAMed_h/datasets/dataset_synapse.py
code served (permissive licence) · get_code("43fa66325ce3cafd")
window_partition Ran hitachinsk/samed/SAMed_h/segment_anything/modeling/image_encoder.py
code served (permissive licence) · get_code("105fa08885dc36cc")
window_unpartition Ran hitachinsk/samed/SAMed_h/segment_anything/modeling/image_encoder.py
code served (permissive licence) · get_code("50f37d517e2be27f")
build_sam_vit_b Not yet run hitachinsk/samed/SAMed_h/segment_anything/build_sam.py
code served (permissive licence) · get_code("4182f118ae5496ec")
build_sam_vit_h Not yet run hitachinsk/samed/SAMed_h/segment_anything/build_sam.py
code served (permissive licence) · get_code("35ae09d11793c334")
build_sam_vit_l Not yet run hitachinsk/samed/SAMed_h/segment_anything/build_sam.py
code served (permissive licence) · get_code("6c6888a7271fbc94")

Repositories linked to this paper

Some links come from the archived Papers with Code dataset (CC BY-SA 4.0): attribution and licence.

Abstract

We propose SAMed, a general solution for medical image segmentation. Different from the previous methods, SAMed is built upon the large-scale image segmentation model, Segment Anything Model (SAM), to explore the new research paradigm of customizing large-scale models for medical image segmentation. SAMed applies the low-rank-based (LoRA) finetuning strategy to the SAM image encoder and finetunes it together with the prompt encoder and the mask decoder on labeled medical image segmentation datasets. We also observe the warmup finetuning strategy and the AdamW optimizer lead SAMed to successful convergence and lower loss. Different from SAM, SAMed could perform semantic segmentation on medical images. Our trained SAMed model achieves 81.88 DSC and 20.64 HD on the Synapse multi-organ segmentation dataset, which is on par with the state-of-the-art methods. We conduct extensive experiments to validate the effectiveness of our design. Since SAMed only updates a small fraction of the SAM parameters, its deployment cost and storage cost are quite marginal in practical usage. The code of SAMed is available at https://github.com/hitachinsk/SAMed.

For agents

The same record, over MCP at https://syntology.ai/mcp:

get_harvested_code_for_paper("2304.13785")
get_code_for_paper("2304.13785")
have("2304.13785")

Connect an agent — have() is free.