@inproceedings{
ma2024addg,
title={{ADDG}: An Adaptive Domain Generalization Framework for Cross-Plane {MRI} Segmentation},
author={Zibo Ma and Bo Zhang and Zheng Zhang and Wu Liu and Wufan Wang and Hui Gao and Wendong Wang},
booktitle={ACM Multimedia 2024},
year={2024},
url={https://openreview.net/forum?id=P1n5GLMxde}
}
Multi-planar and multi-slice magnetic resonance imaging (MRI) can provide more comprehensive 3D structural information for disease diagnosis. However, compared to multi-source MRI, multi-planar MRI uses almost the same scanning parameters but scans different internal structures. This atypical domain difference may lead to poor performance of traditional domain generalization methods in handling multi-planar MRI, especially when MRI from different planes also comes from different sources. In this paper, we introduce ADDG, an Adaptive Domain Generalization Framework tailored for accurate cross-plane MRI segmentation. ADDG significantly mitigates the impact of information loss caused by slice spacing by incorporating 3D shape constraints of the segmentation target, and better clarifies the feature differences between different planes of data through adaptive data partitioning strategy. Specifically, we propose a mesh deformation-based organ segmentation network to simultaneously delineate the 2D boundary and 3D mask of the prostate, as well as to guide more accurate mesh deformation. We also develop an organ-specific mesh template and employ Loop subdivision for unpooling new vertices to a triangular mesh to guide the mesh deformation task, resulting in smoother organ shapes. Furthermore, we design a flexible meta-learning paradigm that adaptively partitions data domains based on invariant learning, which can learn domain invariant features from multi-source training sets to further enhance the generalization ability of the model. Experimental results show that our approach outperforms several medical image segmentation, single-planar-based 3D shape reconstruction, and domain generalization methods.
ADDG: An Adaptive Domain Generalization Framework for Cross-Plane MRI Segmentation
Zibo Ma Bo Zhang Zheng Zhang Wu Liu Wufan Wang Hui Gao Wendong Wang