FunduSAM: A Specialized Deep Learning Model for Enhanced Optic Disc and Cup Segmentation in Fundus Images
Journal:
arXiv
Published Date:
Feb 10, 2025
Abstract
The Segment Anything Model (SAM) has gained popularity as a versatile image
segmentation method, thanks to its strong generalization capabilities across
various domains. However, when applied to optic disc (OD) and optic cup (OC)
segmentation tasks, SAM encounters challenges due to the complex structures,
low contrast, and blurred boundaries typical of fundus images, leading to
suboptimal performance. To overcome these challenges, we introduce a novel
model, FunduSAM, which incorporates several Adapters into SAM to create a deep
network specifically designed for OD and OC segmentation. The FunduSAM utilizes
Adapter into each transformer block after encoder for parameter fine-tuning
(PEFT). It enhances SAM's feature extraction capabilities by designing a
Convolutional Block Attention Module (CBAM), addressing issues related to
blurred boundaries and low contrast. Given the unique requirements of OD and OC
segmentation, polar transformation is used to convert the original fundus OD
images into a format better suited for training and evaluating FunduSAM. A
joint loss is used to achieve structure preservation between the OD and OC,
while accurate segmentation. Extensive experiments on the REFUGE dataset,
comprising 1,200 fundus images, demonstrate the superior performance of
FunduSAM compared to five mainstream approaches.
