Semantic Data Augmentation Enhanced Invariant Risk Minimization for Medical Image Domain Generalization
Fuente:
arXiv
Enregistré dans:
| Auteurs principaux: | , , , , , |
|---|---|
| Format: | Preprint |
| Publié: |
2025
|
| Sujets: | |
| Accès en ligne: | |
| Tags: |
Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
|
| _version_ | 1866908704027181056 |
|---|---|
| author | Zhu, Yaoyao Cai, Xiuding Wang, Yingkai Yao, Yu Luo, Xu Fu, Zhongliang |
| author_facet | Zhu, Yaoyao Cai, Xiuding Wang, Yingkai Yao, Yu Luo, Xu Fu, Zhongliang |
| contents | Deep learning has achieved remarkable success in medical image classification. However, its clinical application is often hindered by data heterogeneity caused by variations in scanner vendors, imaging protocols, and operators. Approaches such as invariant risk minimization (IRM) aim to address this challenge of out-of-distribution generalization. For instance, VIRM improves upon IRM by tackling the issue of insufficient feature support overlap, demonstrating promising potential. Nonetheless, these methods face limitations in medical imaging due to the scarcity of annotated data and the inefficiency of augmentation strategies. To address these issues, we propose a novel domain-oriented direction selector to replace the random augmentation strategy used in VIRM. Our method leverages inter-domain covariance as a guider for augmentation direction, guiding data augmentation towards the target domain. This approach effectively reduces domain discrepancies and enhances generalization performance. Experiments on a multi-center diabetic retinopathy dataset demonstrate that our method outperforms state-of-the-art approaches, particularly under limited data conditions and significant domain heterogeneity. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2502_05593 |
| institution | arXiv |
| publishDate | 2025 |
| record_format | arxiv |
| spellingShingle | Semantic Data Augmentation Enhanced Invariant Risk Minimization for Medical Image Domain Generalization Zhu, Yaoyao Cai, Xiuding Wang, Yingkai Yao, Yu Luo, Xu Fu, Zhongliang Computer Vision and Pattern Recognition Deep learning has achieved remarkable success in medical image classification. However, its clinical application is often hindered by data heterogeneity caused by variations in scanner vendors, imaging protocols, and operators. Approaches such as invariant risk minimization (IRM) aim to address this challenge of out-of-distribution generalization. For instance, VIRM improves upon IRM by tackling the issue of insufficient feature support overlap, demonstrating promising potential. Nonetheless, these methods face limitations in medical imaging due to the scarcity of annotated data and the inefficiency of augmentation strategies. To address these issues, we propose a novel domain-oriented direction selector to replace the random augmentation strategy used in VIRM. Our method leverages inter-domain covariance as a guider for augmentation direction, guiding data augmentation towards the target domain. This approach effectively reduces domain discrepancies and enhances generalization performance. Experiments on a multi-center diabetic retinopathy dataset demonstrate that our method outperforms state-of-the-art approaches, particularly under limited data conditions and significant domain heterogeneity. |
| title | Semantic Data Augmentation Enhanced Invariant Risk Minimization for Medical Image Domain Generalization |
| topic | Computer Vision and Pattern Recognition |
| url | https://arxiv.org/abs/2502.05593 |