arXiv:2606.31198cs.CVcs.AI2026-06中稿 · publication at the…

将时间一致性蒸馏到2D网络,实现前列腺超声视频实时精准分割。

Distilling Temporal Coherence into 2D Networks for Transrectal Ultrasound Prostate Video Segmentation

论文配图:Distilling Temporal Coherence into 2D Networks for Transrectal Ultrasound Prostate Video Segmentation
图 1 · 摘自论文原文
  • 通过置信加权的时间一致性损失,聚焦稳定区域抑制噪声干扰。
  • 在TRUS-V数据集上达到94.3% Dice分数,推理速度超30帧/秒。
  • 无需逐帧标注,适合临床实时导航场景的部署需求。

经直肠超声(TRUS)中前列腺的实时视频分割对图像引导干预至关重要。传统2D方法因忽略时序信息导致帧间不一致,而3D架构又带来过高延迟。为此,我们提出一种时序一致性学习框架,将时间连贯性在训练中蒸馏至2D网络,保持单帧推理效率。基于临床观察:前列腺几何结构稳定,而周围声学环境受生理运动和探头压力影响波动。传统时序约束会传播不稳定区域的错误梯度,因此我们引入基于光流扭曲残差的置信加权时间一致性目标,选择性抑制不可靠区域贡献。同时,双尺度原型对齐模块通过局部边界与全局语义特征的对比优化,强化语义一致性。此外,为避免密集帧标注,采用基于几何等变性的伪标签生成,并结合预训练教师模型的知识蒸馏。在SUN-SEG和新提出的TRUS-V基准(2,679帧)上的实验表明,该方法在实时速度下达到最先进精度与时序一致性。代码与数据集见https://github.com/DYDevelop/DTC-TRUS。

原文摘要 · Abstract (English)

Real-time video segmentation of the prostate in Transrectal Ultrasound (TRUS) is essential for image-guided interventions. While conventional 2D methods suffer from inter-frame inconsistencies by disregarding temporal context, 3D architectures incur prohibitive latency. To resolve this dilemma, we present a Temporally Consistent Learning Framework that distills temporal coherence into a 2D network during training, preserving single-frame inference efficiency. Our design is driven by a key clinical observation: the prostate exhibits geometric stability, whereas the surrounding acoustic environment fluctuates due to physiological motion and transducer pressure. Because conventional temporal constraints propagate erroneous gradients from these unstable regions, we introduce a Confidence-Weighted Temporal Consistency objective derived from optical flow warping residuals, selectively attenuating contributions from unreliable regions. Complementing this pixel-wise constraint, a Dual-scale Prototype Alignment Module enforces semantic coherence through contrastive optimization of local boundary and global semantic features. Furthermore, to eliminate the need for dense per-frame video annotations, we employ geometric equivariance-based pseudo-labeling with knowledge distillation from a pretrained teacher. Extensive experiments on SUN-SEG and our newly introduced TRUS-V benchmark (2,679 frames) demonstrate state-of-the-art accuracy and temporal consistency at real-time speed. Code and dataset are available at https://github.com/DYDevelop/DTC-TRUS.

视频分割超声影像时间一致性知识蒸馏

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。