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

Lightweight Frequency Masker for Cross-Domain Few-Shot Semantic Segmentation

About

Cross-domain few-shot segmentation (CD-FSS) is proposed to first pre-train the model on a large-scale source-domain dataset, and then transfer the model to data-scarce target-domain datasets for pixel-level segmentation. The significant domain gap between the source and target datasets leads to a sharp decline in the performance of existing few-shot segmentation (FSS) methods in cross-domain scenarios. In this work, we discover an intriguing phenomenon: simply filtering different frequency components for target domains can lead to a significant performance improvement, sometimes even as high as 14% mIoU. Then, we delve into this phenomenon for an interpretation, and find such improvements stem from the reduced inter-channel correlation in feature maps, which benefits CD-FSS with enhanced robustness against domain gaps and larger activated regions for segmentation. Based on this, we propose a lightweight frequency masker, which further reduces channel correlations by an Amplitude-Phase Masker (APM) module and an Adaptive Channel Phase Attention (ACPA) module. Notably, APM introduces only 0.01% additional parameters but improves the average performance by over 10%, and ACPA imports only 2.5% parameters but further improves the performance by over 1.5%, which significantly surpasses the state-of-the-art CD-FSS methods.

Jintao Tong, Yixiong Zou, Yuhua Li, Ruixuan Li• 2024

Related benchmarks

TaskDatasetResultRank
Few-shot SegmentationDeepGlobe
mIoU44.9
83
Few-shot SegmentationChest X-ray
mIoU82.8
82
Few-shot Semantic SegmentationAverage Deepglobe, ISIC, Chest X-Ray, FSS-1000
mIoU65.2
54
Few-shot SegmentationFSS-1000
mIoU81.9
22
Few-shot SegmentationISIC
mIoU51.2
22
Medical Image SegmentationAbdominal MRI-CT
Dice61.11
20
Medical Image SegmentationAbdominal CT-MRI
Dice Score0.5951
20
Lesion SegmentationSkin Dermoscopy
DSC39.85
12
Lung SegmentationChest X-ray
Lung DSC71.49
12
Medical Image SegmentationCardiac LGE-bSSFP
Dice Score (LV-BP)68.91
12
Showing 10 of 12 rows

Other info

Code

Follow for update