FR²Seg: Continual Segmentation Across Multiple Sites via Fourier Style Replay and Adaptive Consistency Regularization

Authors

  • Cheng Xu South China University of Technology, China The Hong Kong Polytechnic University, Hong Kong SAR, China
  • Weiwen Zhang South China University of Technology, China
  • Hongrui Zhang South China University of Technology, China
  • Xuemiao Xu South China University of Technology, China Guangdong Engineering Center for Large Model and GenAI Technology State Key Laboratory of Subtropical Building Science Ministry of Education Key Laboratory of Big Data and Intelligent Robot Guangdong Provincial Key Lab of Computational Intelligence and Cyberspace Information
  • Huaidong Zhang South China University of Technology, China
  • Jing Zou The Hong Kong Polytechnic University, Hong Kong SAR, China
  • Jing Qin The Hong Kong Polytechnic University, Hong Kong SAR, China

DOI:

https://doi.org/10.1609/aaai.v39i8.32953

Abstract

In clinical imaging, medical segmentation networks typically require continually adapting to new data from multiple sites over time, as aggregating all data for learning at once can be impractical due to storage limitations and privacy concerns. However, existing methods basically overlook domain-specific characteristics and fall short of adequately capturing domain-invariant knowledge during continual learning, leading to undesired catastrophic forgetting of previous sites and inferior generalization to new sites. To tackle this issue, this paper introduces FR2Seg, to sufficiently exploit both domain-specific and domain-invariant knowledge for efficient continual learning with the aid of low-frequency cues. For the former aspect, we propose a Fourier style replay module to synthesize pseudo images with old-site styles for data augmentation during new-site training, effectively preventing catastrophic forgetting without sacrificing data privacy. For the latter, we present a Fourier adaptive consistency regularization to identify and constrain the optimization of domain-invariant parameters with explicit awareness of knowledge transferability across sites, ensuring excellent generalizability to new sites. Experimental results on two public datasets confirm our method's superiority over existing state-of-the-art continual learning methods.

Downloads

Published

2025-04-11

How to Cite

Xu, C., Zhang, W., Zhang, H., Xu, X., Zhang, H., Zou, J., & Qin, J. (2025). FR²Seg: Continual Segmentation Across Multiple Sites via Fourier Style Replay and Adaptive Consistency Regularization. Proceedings of the AAAI Conference on Artificial Intelligence, 39(8), 8815–8823. https://doi.org/10.1609/aaai.v39i8.32953

Issue

Section

AAAI Technical Track on Computer Vision VII