A method for breast mass segmentation using image augmentation with SAM and receptive field expansion

Binjun Zhang, Eric Rigall, Yuting Huang, Xiao Zou, Shu Zhang*, Junyu Dong, Hui Yu

*Corresponding author for this work

    Research output: Chapter in Book/Report/Conference proceedingConference contribution

    84 Downloads (Pure)

    Abstract

    With the development of deep learning methods and their wide application in medical image segmentation tasks, mammography, used for early breast cancer screening, can further assist clinicians in diagnosis to a certain extent. Due to the loss of fine-grained features in the downsampling process and the very few available mammograms, the main methods for extracting masses from the whole mammography are relatively ineffective. In this paper, we propose a novel mass segmentation model for mammograms based on two U-Net network models, and integrate the Receptive Field Block (RFB) module in-between to enhance the deep features captured by the model. We augmented the input image using Segment Anything Model (SAM) and evaluated the proposed architecture on two public datasets, namely the Curated Breast Imaging Subset of Digital Database for Screening Mammography (CBIS-DDSM) and INBreast, as well as on a private dataset. The results show that the proposed model for segmenting masses in ROI regions can achieve high Dice scores of 92.18% and 89.85%, and Intersection over Union (IoU) scores of 85.47% and 80.80% on both INBreast and the private dataset. In addition, our model for segmenting masses on whole mammographs in CBIS-DDSM dataset can achieve Dice scores of 58.10% and 41.96% IoU scores. Besides, our model performance improved to Dice score 61.78% and IoU score 43.00% on whole mammographs in private dataset using SAM-augmented input images.

    Original languageEnglish
    Title of host publicationICCPR 2023 - Proceedings of the 2023 12th International Conference on Computing and Pattern Recognition
    PublisherAssociation for Computing Machinery (ACM)
    Pages387-394
    Number of pages8
    ISBN (Electronic)9798400707988
    DOIs
    Publication statusPublished - 28 Feb 2024
    Event12th International Conference on Computing and Pattern Recognition, ICCPR 2023 - Qingdao, China
    Duration: 27 Oct 202329 Oct 2023

    Conference

    Conference12th International Conference on Computing and Pattern Recognition, ICCPR 2023
    Country/TerritoryChina
    CityQingdao
    Period27/10/2329/10/23

    Keywords

    • Mammography
    • RFB
    • SAM
    • semantic segmentation
    • U-Net

    Fingerprint

    Dive into the research topics of 'A method for breast mass segmentation using image augmentation with SAM and receptive field expansion'. Together they form a unique fingerprint.

    Cite this