GraphEcho: Graph-Driven Unsupervised Domain Adaptation for Echocardiogram Video Segmentation

Jiewen Yang,Xinpeng Ding,Ziyang Zheng,Xiaowei Xu,Xiaomeng Li
2023-09-20
Abstract:Echocardiogram video segmentation plays an important role in cardiac disease diagnosis. This paper studies the unsupervised domain adaption (UDA) for echocardiogram video segmentation, where the goal is to generalize the model trained on the source domain to other unlabelled target domains. Existing UDA segmentation methods are not suitable for this task because they do not model local information and the cyclical consistency of heartbeat. In this paper, we introduce a newly collected CardiacUDA dataset and a novel GraphEcho method for cardiac structure segmentation. Our GraphEcho comprises two innovative modules, the Spatial-wise Cross-domain Graph Matching (SCGM) and the Temporal Cycle Consistency (TCC) module, which utilize prior knowledge of echocardiogram videos, i.e., consistent cardiac structure across patients and centers and the heartbeat cyclical consistency, respectively. These two modules can better align global and local features from source and target domains, improving UDA segmentation results. Experimental results showed that our GraphEcho outperforms existing state-of-the-art UDA segmentation methods. Our collected dataset and code will be publicly released upon acceptance. This work will lay a new and solid cornerstone for cardiac structure segmentation from echocardiogram videos. Code and dataset are available at: <a class="link-external link-https" href="https://github.com/xmed-lab/GraphEcho" rel="external noopener nofollow">this https URL</a>
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The problem that this paper attempts to solve is the domain adaptation problem between echocardiogram videos collected at different medical centers. Specifically, the research objective is to use the Unsupervised Domain Adaptation (UDA) method to enable the model trained on the source domain to be better generalized to the unlabeled target domain, so as to achieve the cardiac structure segmentation of echocardiogram videos. The existing UDA segmentation methods are not suitable for this task because they do not model local information and the cycle consistency of heartbeats. To solve these problems, the paper introduces a new dataset CardiacUDA and a new method GraphEcho, which includes two innovative modules: the Spatial - wise Cross - domain Graph Matching (SCGM) module and the Temporal Cycle Consistency (TCC) module. These modules utilize the prior knowledge in echocardiogram videos, that is, the consistency of cardiac structures between different patients and centers and the cycle consistency of heartbeats, so as to better align the global and local features of the source domain and the target domain and improve the UDA segmentation results.