Learning Dynamic Collaborative Network for Semi-supervised 3D Vessel Segmentation

TL;DR

This paper introduces DiCo, a dynamic collaborative network for semi-supervised 3D vessel segmentation that dynamically switches teacher-student roles, integrates multi-view analysis, and uses adversarial supervision to improve segmentation accuracy.

Contribution

The paper proposes a novel dynamic role-switching mechanism in teacher-student models, along with multi-view integration and adversarial shape constraints for enhanced semi-supervised 3D vessel segmentation.

Findings

Achieves state-of-the-art results on three benchmarks.

Effectively handles label inconsistencies through multi-view projection.

Demonstrates improved segmentation accuracy with dynamic role switching.

Abstract

In this paper, we present a new dynamic collaborative network for semi-supervised 3D vessel segmentation, termed DiCo. Conventional mean teacher (MT) methods typically employ a static approach, where the roles of the teacher and student models are fixed. However, due to the complexity of 3D vessel data, the teacher model may not always outperform the student model, leading to cognitive biases that can limit performance. To address this issue, we propose a dynamic collaborative network that allows the two models to dynamically switch their teacher-student roles. Additionally, we introduce a multi-view integration module to capture various perspectives of the inputs, mirroring the way doctors conduct medical analysis. We also incorporate adversarial supervision to constrain the shape of the segmented vessels in unlabeled data. In this process, the 3D volume is projected into 2D views to mitigate the impact of label inconsistencies. Experiments demonstrate that our DiCo method sets new state-of-the-art performance on three 3D vessel segmentation benchmarks. The code repository address is https://github.com/xujiaommcome/DiCo