用强化学习提升脑白质束追踪精度,无需真实路径标注
A Deep RL based Framework for Targeted White Matter Tractography

- 结合强化学习与大模型策略,实现束特异性追踪优化
- 在多个公开数据集上准确重建关键白质通路,泛化性强
- 适合神经影像研究者和临床医生,降低对标注数据依赖
纤维束追踪能非侵入性地描绘大脑结构连接,是现代神经影像的核心技术。然而,其仍面临白质结构复杂、假阳性率高等挑战。本文提出一种融合强化学习与监督学习的混合框架,专用于束特异性追踪,无需真实纤维作为训练标签。该框架摒弃显式分割步骤,简化流程。第一,采用基于GPT的策略学习方法,优化强化学习策略;第二,构建可扩展的数据驱动多策略融合机制,利用多个强化学习策略的互补优势,提升追踪性能与鲁棒性。在TractoInferno、HCP和ISMRM-2015等基准数据集上验证,结果表明该框架具备强泛化能力,能准确重建脑白质束。这些贡献显著提升了追踪的可靠性与准确性,减少对真实标注的依赖。
原文摘要 · Abstract (English)
Fiber tractography's ability to reconstruct the brain's structural pathways, has made it a crucial component of modern neuroimaging, enabling detailed, non-invasive mapping of structural connectivity and supporting a wide range of neurological research and clinical applications. However, despite its importance, tractography remains a challenging task due to the inherent complexity of white matter structure and its susceptibility to false positives, which can lead to the misrepresentation of critical pathways. To overcome these limitations, in this thesis, we propose a hybrid framework that integrates reinforcement learning with supervised learning for refining RL policies, specifically tailored for tract-specific tractography. Notably, our framework does not rely on ground-truth fibers for training. Moreover, the tract-specific formulation bypasses the need for an explicit segmentation process, simplifying the overall pipeline. Our work includes two main contributions, each building upon the previous. First, we introduce a hybrid approach that combines reinforcement learning with supervised learning (specifically, GPT-based policy learning) to refine policies in a tract-specific context. Second, we propose a scalable framework for data-driven multi-policy fusion, which leverages the complementary strengths of multiple RL policies to improve tractography performance and robustness. We demonstrate the effectiveness of our framework through extensive validation on benchmark public datasets including TractoInferno, HCP, and ISMRM-2015, highlighting its ability to generalize across data sources and accurately reconstruct brain white matter tracts. We believe that these contributions represent significant advancements in the field of tractography, improving robustness, reliability, and accuracy while reducing dependence on ground-truth annotations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。