arXiv:2603.10418cs.CV2026-03

统一优化脑白质纤维束配准与聚类,提升分析精度

TractoRC: A Unified Probabilistic Learning Framework for Joint Tractography Registration and Clustering

  • 构建共享嵌入空间,联合优化配准与聚类
  • 在多模态数据上显著提升配准与聚类准确率
  • 适合神经影像分析、脑连接组研究者使用

扩散MRI纤维束追踪可实现白质路径的活体重建。轨迹分析中两个关键任务为:1)轨迹配准,即对齐不同个体的纤维束;2)流线聚类,即将流线分组为紧凑的纤维束。尽管两者均旨在捕捉几何相似结构以刻画一致的白质组织,但通常独立进行。本文提出TractoRC,一种统一的概率学习框架,将轨迹配准与聚类整合于单一优化过程中,使两任务能相互借鉴。TractoRC学习流线点的潜在嵌入空间,作为两任务的共享表示。在此空间中,配准通过概率关键点建模解剖地标分布以对齐不同受试者轨迹,聚类则学习捕捉几何相似性的结构原型以形成连贯簇。为有效学习该共享空间,引入变换等变的自监督策略,获得几何感知且变换不变的嵌入。实验表明,联合优化显著优于独立处理的最先进方法。代码将在https://github.com/yishengpoxiao/TractoRC公开。

原文摘要 · Abstract (English)

Diffusion MRI tractography enables in vivo reconstruction of white matter (WM) pathways. Two key tasks in tractography analysis include: 1) tractogram registration that aligns streamlines across individuals, and 2) streamline clustering that groups streamlines into compact fiber bundles. Although both tasks share the goal of capturing geometrically similar structures to characterize consistent WM organization, they are typically performed independently. In this work, we propose TractoRC, a unified probabilistic framework that jointly performs tractogram registration and streamline clustering within a single optimization scheme, enabling the two tasks to leverage complementary information. TractoRC learns a latent embedding space for streamline points, which serves as a shared representation for both tasks. Within this space, both tasks are formulated as probabilistic inference over structural representations: registration learns the distribution of anatomical landmarks as probabilistic keypoints to align tractograms across subjects, and clustering learns streamline structural prototypes that capture geometric similarity to form coherent streamline clusters. To support effective learning of this shared space, we introduce a transformation-equivariant self-supervised strategy to learn geometry-aware and transformation-invariant embeddings. Experiments demonstrate that jointly optimizing registration and clustering significantly improves performance in both tasks over state-of-the-art methods that treat them independently. Code will be made publicly available at https://github.com/yishengpoxiao/TractoRC .

纤维束追踪配准聚类深度学习

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