DDTracking: A Deep Generative Framework for Diffusion MRI Tractography with Streamline Local-Global Spatiotemporal Modeling

2508.04568v1 cs.CV 2025-08-09
Авторы:

Yijie Li, Wei Zhang, Xi Zhu, Ye Wu, Yogesh Rathi, Lauren J. O'Donnell, Fan Zhang

Резюме на русском

Метод DDTracking представляет собой уникальную глубокоупрострянная структуру для трактографии диффузионного МРТ (dMRI), которая оптимизирует синтез структур нейронных нитей (streamlines) с использованием процесса условного деноisingа (conditional denoising). Основная идея заключается в разделении моделирования на два пути: локальное пространственное моделирование (детализация точечных структур) и глобальные зависимости времени (сохранение консистентности в длинных структурах). Улучшенная модель условного диффузионного моделирования (conditional diffusion model) объединяет эти элементы для предсказания направлений продолжения структуры в процессе стринглайн-трактографии (streamline tractography). В результате опробований на различных dMRI-датасетах, включая синтетические и клинические, DDTracking показал результаты, значительно превышающие современные методы по точности и устойчивости. Особенно выдающимися являются результаты на тестовых наборах ISMRM Challenge и TractoInferno. Этот подход обеспечивает анатомически интуитивные, надежные и гибкие результаты, которые могут быть применены в различных мед. задачах dMRI. Реализация доступна по адресу: https://github.com/yishengpoxiao/DDtracking.git.

Abstract

This paper presents DDTracking, a novel deep generative framework for diffusion MRI tractography that formulates streamline propagation as a conditional denoising diffusion process. In DDTracking, we introduce a dual-pathway encoding network that jointly models local spatial encoding (capturing fine-scale structural details at each streamline point) and global temporal dependencies (ensuring long-range consistency across the entire streamline). Furthermore, we design a conditional diffusion model module, which leverages the learned local and global embeddings to predict streamline propagation orientations for tractography in an end-to-end trainable manner. We conduct a comprehensive evaluation across diverse, independently acquired dMRI datasets, including both synthetic and clinical data. Experiments on two well-established benchmarks with ground truth (ISMRM Challenge and TractoInferno) demonstrate that DDTracking largely outperforms current state-of-the-art tractography methods. Furthermore, our results highlight DDTracking's strong generalizability across heterogeneous datasets, spanning varying health conditions, age groups, imaging protocols, and scanner types. Collectively, DDTracking offers anatomically plausible and robust tractography, presenting a scalable, adaptable, and end-to-end learnable solution for broad dMRI applications. Code is available at: https://github.com/yishengpoxiao/DDtracking.git

Ссылки и действия