Domain Generalization With Amplitude-Based Data Generation and Feature Random Suppression.

Abstract

Segmenting unknown domains using a model trained in the source domain still faces challenges. Although some approaches tried to resolve the problem through various data generation and network architecture designs, they cannot achieve satisfactory segmentation results compared with single domain segmentation of consistent data distribution. Therefore, we propose a data augmentation method based on amplitude perturbation to expand the distribution of data types, thereby covering target data. A feature suppression strategy is proposed to reduce the network's over-reliance on important features of the source domain data to improve generalization performance. In addition, we design a luminance contrast consistency (LCC) learning module to harmonize the data styles between different domains and a multiscale convolutional attention (MSCA) module to enhance the network's perception of small target objects and improve the segmentation performance of the model, which further improves segmentation performance. Our method achieves the state-of-the-art (SOTA) results on two public datasets of ATLAS2.0 and Prostate. The code is available at https://github.com/butterflyGN/DGSFTAFS.