Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Spider: A Unified Framework for Context-dependent Concept Segmentation

About

Different from the context-independent (CI) concepts such as human, car, and airplane, context-dependent (CD) concepts require higher visual understanding ability, such as camouflaged object and medical lesion. Despite the rapid advance of many CD understanding tasks in respective branches, the isolated evolution leads to their limited cross-domain generalisation and repetitive technique innovation. Since there is a strong coupling relationship between foreground and background context in CD tasks, existing methods require to train separate models in their focused domains. This restricts their real-world CD concept understanding towards artificial general intelligence (AGI). We propose a unified model with a single set of parameters, Spider, which only needs to be trained once. With the help of the proposed concept filter driven by the image-mask group prompt, Spider is able to understand and distinguish diverse strong context-dependent concepts to accurately capture the Prompter's intention. Without bells and whistles, Spider significantly outperforms the state-of-the-art specialized models in 8 different context-dependent segmentation tasks, including 4 natural scenes (salient, camouflaged, and transparent objects and shadow) and 4 medical lesions (COVID-19, polyp, breast, and skin lesion with color colonoscopy, CT, ultrasound, and dermoscopy modalities). Besides, Spider shows obvious advantages in continuous learning. It can easily complete the training of new tasks by fine-tuning parameters less than 1\% and bring a tolerable performance degradation of less than 5\% for all old tasks. The source code will be publicly available at \href{https://github.com/Xiaoqi-Zhao-DLUT/Spider-UniCDSeg}{Spider-UniCDSeg}.

Xiaoqi Zhao, Youwei Pang, Wei Ji, Baicheng Sheng, Jiaming Zuo, Lihe Zhang, Huchuan Lu• 2024

Related benchmarks

TaskDatasetResultRank
Medical Image SegmentationISIC 2018
Dice Score90.67
139
Medical Image SegmentationGLAS
Dice56.42
60
Medical Image SegmentationCOVID-CT
Dice (%)83.14
45
Medical Image SegmentationBreast Ultrasound
DSC (%)81.11
26
Medical Image SegmentationBTMRI (Source)
DSC84.66
24
Medical Image SegmentationPH2
DICE Score90.47
23
SegmentationBrain Tumor
mIoU80.17
22
Medical Image SegmentationACDC
DSC63.26
22
Medical Lesion SegmentationBreast Lesion
Dice83.8
21
Medical Lesion SegmentationLung Infection
Dice Score75.21
21
Showing 10 of 25 rows

Other info

Follow for update