Domain Generalization with Fourier Transform and Soft Thresholding
- URL: http://arxiv.org/abs/2309.09866v3
- Date: Tue, 12 Dec 2023 22:37:06 GMT
- Title: Domain Generalization with Fourier Transform and Soft Thresholding
- Authors: Hongyi Pan, Bin Wang, Zheyuan Zhang, Xin Zhu, Debesh Jha, Ahmet Enis
Cetin, Concetto Spampinato, Ulas Bagci
- Abstract summary: Domain generalization aims to train models on multiple source domains so that they can generalize well to unseen target domains.
To overcome this limitation, we introduce a soft-thresholding function in the Fourier domain.
The innovative nature of the soft thresholding fused with Fourier-transform-based domain generalization improves neural network models' performance.
- Score: 10.50210846364862
- License: http://creativecommons.org/licenses/by/4.0/
- Abstract: Domain generalization aims to train models on multiple source domains so that
they can generalize well to unseen target domains. Among many domain
generalization methods, Fourier-transform-based domain generalization methods
have gained popularity primarily because they exploit the power of Fourier
transformation to capture essential patterns and regularities in the data,
making the model more robust to domain shifts. The mainstream
Fourier-transform-based domain generalization swaps the Fourier amplitude
spectrum while preserving the phase spectrum between the source and the target
images. However, it neglects background interference in the amplitude spectrum.
To overcome this limitation, we introduce a soft-thresholding function in the
Fourier domain. We apply this newly designed algorithm to retinal fundus image
segmentation, which is important for diagnosing ocular diseases but the neural
network's performance can degrade across different sources due to domain
shifts. The proposed technique basically enhances fundus image augmentation by
eliminating small values in the Fourier domain and providing better
generalization. The innovative nature of the soft thresholding fused with
Fourier-transform-based domain generalization improves neural network models'
performance by reducing the target images' background interference
significantly. Experiments on public data validate our approach's effectiveness
over conventional and state-of-the-art methods with superior segmentation
metrics.
Related papers
- A Fourier Transform Framework for Domain Adaptation [8.997055928719515]
unsupervised domain adaptation (UDA) can transfer knowledge from a label-rich source domain to a target domain that lacks labels.
Many existing UDA algorithms suffer from directly using raw images as input.
We employ the Fourier method (FTF) to incorporate low-level information from the target domain into the source domain.
arXiv Detail & Related papers (2024-03-12T16:35:32Z) - FAN-Net: Fourier-Based Adaptive Normalization For Cross-Domain Stroke
Lesion Segmentation [17.150527504559594]
We propose a novel FAN-Net, a U-Net-based segmentation network incorporated with a Fourier-based adaptive normalization (FAN)
The experimental results on the ATLAS dataset, which consists of MR images from 9 sites, show the superior performance of the proposed FAN-Net compared with baseline methods.
arXiv Detail & Related papers (2023-04-23T06:58:21Z) - Unsupervised Domain Transfer with Conditional Invertible Neural Networks [83.90291882730925]
We propose a domain transfer approach based on conditional invertible neural networks (cINNs)
Our method inherently guarantees cycle consistency through its invertible architecture, and network training can efficiently be conducted with maximum likelihood.
Our method enables the generation of realistic spectral data and outperforms the state of the art on two downstream classification tasks.
arXiv Detail & Related papers (2023-03-17T18:00:27Z) - Domain Generalisation via Domain Adaptation: An Adversarial Fourier
Amplitude Approach [13.642506915023871]
We adversarially synthesise the worst-case target domain and adapt a model to that worst-case domain.
On the DomainBedNet dataset, the proposed approach yields significantly improved domain generalisation performance.
arXiv Detail & Related papers (2023-02-23T14:19:07Z) - Deep Fourier Up-Sampling [100.59885545206744]
Up-sampling in the Fourier domain is more challenging as it does not follow such a local property.
We propose a theoretically sound Deep Fourier Up-Sampling (FourierUp) to solve these issues.
arXiv Detail & Related papers (2022-10-11T06:17:31Z) - Deep Frequency Filtering for Domain Generalization [55.66498461438285]
Deep Neural Networks (DNNs) have preferences for some frequency components in the learning process.
We propose Deep Frequency Filtering (DFF) for learning domain-generalizable features.
We show that applying our proposed DFF on a plain baseline outperforms the state-of-the-art methods on different domain generalization tasks.
arXiv Detail & Related papers (2022-03-23T05:19:06Z) - Functional Regularization for Reinforcement Learning via Learned Fourier
Features [98.90474131452588]
We propose a simple architecture for deep reinforcement learning by embedding inputs into a learned Fourier basis.
We show that it improves the sample efficiency of both state-based and image-based RL.
arXiv Detail & Related papers (2021-12-06T18:59:52Z) - A Fourier-based Framework for Domain Generalization [82.54650565298418]
Domain generalization aims at tackling this problem by learning transferable knowledge from multiple source domains in order to generalize to unseen target domains.
This paper introduces a novel Fourier-based perspective for domain generalization.
Experiments on three benchmarks have demonstrated that the proposed method is able to achieve state-of-the-arts performance for domain generalization.
arXiv Detail & Related papers (2021-05-24T06:50:30Z) - FDA: Fourier Domain Adaptation for Semantic Segmentation [82.4963423086097]
We describe a simple method for unsupervised domain adaptation, whereby the discrepancy between the source and target distributions is reduced by swapping the low-frequency spectrum of one with the other.
We illustrate the method in semantic segmentation, where densely annotated images are aplenty in one domain, but difficult to obtain in another.
Our results indicate that even simple procedures can discount nuisance variability in the data that more sophisticated methods struggle to learn away.
arXiv Detail & Related papers (2020-04-11T22:20:48Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.