Medical image analysis using improved SAM-Med2D: segmentation and classification perspectives
Recently emerged SAM-Med2D represents a state-of-the-art advancement in medical image segmentation. Through fine-tuning the Large Visual Model, Segment Anything Model (SAM), on extensive medical datasets, it has achieved impressive results in cross-modal medical image segmentation. However, its reliance on interactive prompts may restrict its applicability under specific conditions. To address this limitation, we introduce SAM-AutoMed, which achieves automatic segmentation of medical images by replacing the original prompt encoder with an improved MobileNet v3 backbone. The performance on multiple datasets surpasses both SAM and SAM-Med2D. Current enhancements on the Large Visual Model SAM lack applications in the field of medical image classification. Therefore, we introduce SAM-MedCls, which combines the encoder of SAM-Med2D with our designed attention modules to construct an end-to-end medical image classification model. It performs well on datasets of various modalities, even achieving state-of-the-art results, indicating its potential to become a universal model for medical image classification.
Authors
- Zhiyi He (ORCID: https://orcid.org/0009-0007-2969-9061)
- Jiakang Sun (ORCID: https://orcid.org/0000-0002-5771-4712)
- Siyuan Ren (ORCID: https://orcid.org/0009-0008-2555-2713)
- Ke Chen (ORCID: https://orcid.org/0000-0003-2538-7793)
- Xinyang He (ORCID: https://orcid.org/0009-0006-0324-1899)
- Cheng Peng (ORCID: https://orcid.org/0000-0001-6050-1294)
- Xu Liu (ORCID: https://orcid.org/0009-0007-8947-1448)
Institutions
- Chinese Academy of Sciences (CN)
- Chengdu Institute of Biology (CN)
- University of Chinese Academy of Sciences (CN)
Publication Details
- Journal
- BMC Medical Imaging
- Published
- 2024-09-16
- DOI
- https://doi.org/10.1186/s12880-024-01401-6
- Citations
- 19
- Primary Topic
- COVID-19 diagnosis using AI
- Type
- article
- Field-Weighted Citation Impact
- 7.68