Style Content Decomposition-based Data Augmentation for Domain Generalizable Medical Image Segmentation

Code

• github.com/senyh/stycona
  OfficialIn paper★ 0

Abstract

Due to the domain shifts between training and testing medical images, learned segmentation models often experience significant performance degradation during deployment. In this paper, we first decompose an image into its style code and content map and reveal that domain shifts in medical images involve: style shifts (i.e., differences in image appearance) and content shifts (i.e., variations in anatomical structures), the latter of which has been largely overlooked. To this end, we propose StyCona, a style content decomposition-based data augmentation method that innovatively augments both image style and content within the rank-one space, for domain generalizable medical image segmentation. StyCona is a simple yet effective plug-and-play module that substantially improves model generalization without requiring additional training parameters or modifications to the segmentation model architecture. Experiments on cross-sequence, cross-center, and cross-modality medical image segmentation settings with increasingly severe domain shifts, demonstrate the effectiveness of StyCona and its superiority over state-of-the-arts. The code is available at https://github.com/Senyh/StyCona.

Tasks

Reproductions