arXiv:2503.07104eess.IVcs.CV2025-03

用全局上下文替代局部细节,让脑白质束分割快100倍且更准

Global Context Is All You Need for Parallel Efficient Tractography Parcellation

  • 用Transformer并行处理分块纤维束,全脑信息相互作为全局上下文
  • 速度提升达100倍,临床工作站无GPU也能运行,准确率反而更高
  • 创新性地通过翻转不变嵌入或数据增强解决方向缺失问题,适合医学影像研究者

全脑扩散MRI纤维追踪后常需对每条纤维进行白质束分类或剔除误报。高效分割对大规模研究和临床应用均至关重要。现有先进方法TractCloud依赖局部-全局表示以追求高精度,但我们发现局部上下文不仅无助于准确率,反而在病理情况下有害。基于此,我们提出PETParc:一种基于Transformer的并行高效纤维追踪分割新方法。该方法将全脑纤维束随机划分为子束组,各组在并行分类时共享全局上下文信息。相比TractCloud,PETParc速度提升最高达两个数量级,甚至可在无GPU的临床工作站上完成推理。通过新颖的翻转不变嵌入或直接将翻转作为数据增强,有效应对纤维方向缺失问题。尽管速度飞跃,性能通常优于先前方法。代码与预训练模型将在论文接收后公开。

原文摘要 · Abstract (English)

Whole-brain tractography in diffusion MRI is often followed by a parcellation in which each streamline is classified as belonging to a specific white matter bundle, or discarded as a false positive. Efficient parcellation is important both in large-scale studies, which have to process huge amounts of data, and in the clinic, where computational resources are often limited. TractCloud is a state-of-the-art approach that aims to maximize accuracy with a local-global representation. We demonstrate that the local context does not contribute to the accuracy of that approach, and is even detrimental when dealing with pathological cases. Based on this observation, we propose PETParc, a new method for Parallel Efficient Tractography Parcellation. PETParc is a transformer-based architecture in which the whole-brain tractogram is randomly partitioned into sub-tractograms whose streamlines are classified in parallel, while serving as global context for each other. This leads to a speedup of up to two orders of magnitude relative to TractCloud, and permits inference even on clinical workstations without a GPU. PETParc accounts for the lack of streamline orientation either via a novel flip-invariant embedding, or by simply using flips as part of data augmentation. Despite the speedup, results are often even better than those of prior methods. The code and pretrained model will be made public upon acceptance.

脑连接组扩散成像Transformer加速算法

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