用统一模型同时完成白质纤维束的分类、检索、描述和问答。
TractoGraphVLM: A Unified Vision-Language Framework for White Matter Tractography

- 将纤维束建模为带3D位置和方向信息的图结构,用图变压器统一处理。
- 在HCP数据上达到91.8%分类准确率,文本到纤维检索R@1达84.7%。
- 仅靠语言监督就能学习神经解剖结构,适合医学图像理解研究者。
视觉语言模型已改变2D医学影像分析,但将其扩展至3D白质纤维束追踪仍具挑战性,因纤维束拓扑复杂。我们提出TractoGraphVLM,一个统一框架,支持四类任务:纤维束分类、文本到纤维检索、解剖描述生成与视觉问答,基于共享的GPS架构、训练流程与输出设计。纤维束表示为流线图,节点编码3D位置与切向方向。通用、强大、可扩展(GPS)图变换器通过对比学习与冻结的BiomedBERT文本编码器对齐,生成束嵌入;带有视觉前缀令牌的BioGPT解码器生成描述与回答。单一共享编码器与解码器联合训练,从同一检查点评估。在HCP年轻成人受试者上训练,该模型在留出测试集上取得91.8%分类准确率,84.7%检索R@1,BLEU-4=20.1,ROUGE-L=66.8,VQA准确率66.4%。相同检查点零样本迁移至HCP老年受试者,判别任务略有下降,生成任务下降更明显,显示对年龄与采集差异的鲁棒性。语言监督比仅标签训练生成更丰富的表征,恢复了半球、纤维族等结构,这些未作为标签提供但体现在描述中。仅更换视觉编码器时,保留纤维方向的图结构优于体素基线,且GPS表现最佳。生成指标衡量与结构化知识库的一致性,而非独立临床文本;即便如此,TractoGraphVLM证明仅靠语言监督即可训练一个模型,实现纤维束的分类、检索、描述与问答,学习可迁移的神经解剖知识。
原文摘要 · Abstract (English)
Vision language models have transformed 2D medical imaging, yet extending them to 3D white matter tractography remains challenging due to the complex topology of fiber bundles. We introduce TractoGraphVLM, a unified framework for four tasks, bundle classification, text-to-tract retrieval, anatomical captioning, and visual question answering, built on a shared GPS architecture, training procedure, and read-out design. Fiber bundles are represented as streamline graphs whose nodes encode 3D position and tangent orientation. A General, Powerful, Scalable (GPS) graph transformer produces bundle embeddings aligned with a frozen BiomedBERT text encoder via contrastive learning, while a BioGPT decoder with visual prefix tokens generates captions and answers. A single shared encoder and decoder is trained jointly across all four tasks and evaluated from one checkpoint. Trained on HCP Young Adult subjects, TractoGraphVLM achieves 91.8% bundle classification accuracy, 84.7% retrieval R@1, BLEU-4=20.1, ROUGE-L=66.8, and 66.4% VQA accuracy on a held-out test set. The same checkpoints transfer zero-shot to HCP Aging subjects, with a modest drop on discriminative tasks and a larger drop on generative tasks, showing robustness to age and acquisition shift. Language supervision yields richer representations than label-only training, recovering structure like hemisphere and fiber family, carried by captions but never given as a label. Swapping only the visual encoder, graphs preserving fiber orientation outperform volumetric baselines, with GPS giving the best balance. Generative metrics measure consistency with a structured knowledge base rather than independent clinical text; even so, TractoGraphVLM shows that classifying, retrieving, describing, and answering questions about a white matter bundle can be served by one jointly trained model that learns transferable neuroanatomy from language alone.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。