FAMNet: Frequency-aware Matching Network for Cross-domain Few-shot Medical Image Segmentation
Journal:
arXiv
Published Date:
Dec 12, 2024
Abstract
Existing few-shot medical image segmentation (FSMIS) models fail to address a
practical issue in medical imaging: the domain shift caused by different
imaging techniques, which limits the applicability to current FSMIS tasks. To
overcome this limitation, we focus on the cross-domain few-shot medical image
segmentation (CD-FSMIS) task, aiming to develop a generalized model capable of
adapting to a broader range of medical image segmentation scenarios with
limited labeled data from the novel target domain. Inspired by the
characteristics of frequency domain similarity across different domains, we
propose a Frequency-aware Matching Network (FAMNet), which includes two key
components: a Frequency-aware Matching (FAM) module and a Multi-Spectral Fusion
(MSF) module. The FAM module tackles two problems during the meta-learning
phase: 1) intra-domain variance caused by the inherent support-query bias, due
to the different appearances of organs and lesions, and 2) inter-domain
variance caused by different medical imaging techniques. Additionally, we
design an MSF module to integrate the different frequency features decoupled by
the FAM module, and further mitigate the impact of inter-domain variance on the
model's segmentation performance. Combining these two modules, our FAMNet
surpasses existing FSMIS models and Cross-domain Few-shot Semantic Segmentation
models on three cross-domain datasets, achieving state-of-the-art performance
in the CD-FSMIS task.