HCDG: A Hierarchical Consistency Framework for Domain Generalization on Medical Image Segmentation
Not on the map. Why?
Abstract (arXiv)
Modern deep neural networks struggle to transfer knowledge and generalize across diverse domains when deployed to real-world applications. Currently, domain generalization (DG) is introduced to learn a universal representation from multiple domains to improve the network generalization ability on unseen domains. However, previous DG methods only focus on the data-level consistency scheme without considering the synergistic regularization among different consistency schemes. In this paper, we present a novel Hierarchical Consistency framework for Domain Generalization (HCDG) by integrating Extrinsic Consistency and Intrinsic Consistency synergistically. Particularly, for the Extrinsic Consistency, we leverage the knowledge across multiple source domains to enforce data-level consistency. To better enhance such consistency, we design a novel Amplitude Gaussian-mixing strategy into Fourier-based data augmentation called DomainUp. For the Intrinsic Consistency, we perform task-level consistency for the same instance under the dual-task scenario. We evaluate the proposed HCDG framework on two medical image segmentation tasks, i.e., optic cup/disc segmentation on fundus images and prostate MRI segmentation. Extensive experimental results manifest the effectiveness and versatility of our HCDG framework.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 37 of its 59 references that are in the map. First posted on arXiv 13 Sep 2021.
Builds on
37 of 59 references.
References · 37
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2017mixup: Beyond Empirical Risk Minimizationcited by 5,155
- 2018Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentationcited by 3,210
- 2019CutMix: Regularization Strategy to Train Strong Classifiers with Localizable Featurescited by 2,875
- 2016V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentationcited by 2,284
- 2018The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networkscited by 2,250
- 2014Unsupervised Domain Adaptation by Backpropagationcited by 1,906
- 2019Invariant Risk Minimizationcited by 1,832
- 2017Domain Randomization for Transferring Deep Neural Networks from Simulation to the Real Worldcited by 1,401
- 2017Learning to Generalize: Meta-Learning for Domain Generalizationcited by 762
- 2013Domain Generalization via Invariant Feature Representationcited by 622
- 2019Domain Generalization by Solving Jigsaw Puzzlescited by 449
25 more
- 2016Regularization With Stochastic Transformations and Perturbations for Deep Semi-Supervised Learningcited by 441
- 2019Domain Generalization via Model-Agnostic Learning of Semantic Featurescited by 405
- 2020Semi-Supervised Semantic Segmentation with Cross-Consistency Trainingcited by 375
- 2020Learning to Learn Single Domain Generalizationcited by 295
- 2021A Fourier-based Framework for Domain Generalizationcited by 287
- 2019Domain Randomization and Pyramid Consistency: Simulation-to-Real Generalization without Accessing Target Domain Datacited by 245
- 2020Deep Domain-Adversarial Image Generation for Domain Generalisationcited by 231
- 2019REFUGE Challenge: A Unified Framework for Evaluating Automated Methods for Glaucoma Assessment from Fundus Photographscited by 226
- 2020Semi-supervised Medical Image Segmentation through Dual-task Consistencycited by 215
- 2018Domain Generalization via Conditional Invariant Representationcited by 146
- 2020Shape-aware Meta-learning for Generalizing Prostate MRI Segmentation to Unseen Domainscited by 124
- 2020MS-Net: Multi-Site Network for Improving Prostate Segmentation with Heterogeneous MRI Datacited by 121
- 2018Joint Optic Disc and Cup Segmentation Based on Multi-label Deep Network and Polar Transformationcited by 73
- 2020DoFE: Domain-oriented Feature Embedding for Generalizable Fundus Image Segmentation on Unseen Datasetscited by 70
- 2019Boundary and Entropy-driven Adversarial Learning for Fundus Image Segmentationcited by 46
- 2019Patch-based Output Space Adversarial Learning for Joint Optic Disc and Cup Segmentationcited by 44
- 2021Semi-Supervised Domain Generalization with Stochastic StyleMatchcited by 43
- 2019Shape-Aware Organ Segmentation by Predicting Signed Distance Mapscited by 41
- 2017Automatic 3D Cardiovascular MR Segmentation with Densely-Connected Volumetric ConvNetscited by 32
- 2019Psi-Net: Shape and boundary aware joint multi-task deep network for medical image segmentationcited by 30
- 2020MaxUp: A Simple Way to Improve Generalization of Neural Network Trainingcited by 27
- 2020Explainable Deep Classification Models for Domain Generalizationcited by 27
- 2021Out-of-distribution Prediction with Invariant Risk Minimization: The Limitation and An Effective Fixcited by 27
- 2017Improving Dermoscopic Image Segmentation with Enhanced Convolutional-Deconvolutional Networkscited by 18
- 2021Generalizing Nucleus Recognition Model in Multi-source Images via Pruningcited by 7
Built on by
Citing papers per year: 2024: 1.
1 paper.