DRDM Icon VasTSD: Learning 3D Vascular Tree-state Space Diffusion Model for Angiography Synthesis

1National University of Defense Technology
2Xiangjiang Laboratory
CVPR 2025

*Co-first authors    Corresponding Authors

Abstract

Angiography imaging is a medical imaging technique that enhances the visibility of blood vessels within the body by using contrast agents. Angiographic images can effectively assist in the diagnosis of vascular diseases. However, contrast agents may bring extra radiation exposure which is harmful to patients with health risks. To mitigate these concerns, in this paper, we aim to automatically generate angiography from non-angiographic inputs, by leveraging and enhancing the inherent physical properties of vascular structures. Previous methods relying on 2D slice-based angiography synthesis struggle with maintaining continuity in 3D vascular structures and exhibit limited effectiveness across different imaging modalities. We propose VasTSD, a 3D vascular tree-state space diffusion model to synthesize angiography from 3D non-angiographic volumes, with a novel state space serialization approach that dynamically constructs vascular tree topologies, integrating these with a diffusion-based generative model to ensure the generation of anatomically continuous vasculature in 3D volumes. A pre-trained vision embedder is employed to construct vascular state space representations, enabling consistent modeling of vascular structures across multiple modalities. Extensive experiments on various angiographic datasets demonstrate the superiority of VasTSD over prior works, achieving enhanced continuity of blood vessels in synthesized angiographic synthesis for multiple modalities and anatomical regions.

Method Overview

The overall framework of VasTSD. VasTSD contains a pre-trained vision embedder and a 3D vascular tree-state space diffusion module. The vision embedder encodes 3D medical data and generates embeddings for the diffusion process. The 3D vascular state space diffusion module consists of a forward diffusion and a 3D denoising process based on tree-state space.

Results

Vascular Tree Scanning Algorithm
Require:
\( X = \{x_i\}_{i=1}^L \), \( Y \)
Ensure:
\( output \), gradients
Initialize:
\( G \gets \text{initializeGraph}(V, E) \)
\( G_T \gets \text{kruskalMST}(G) \)
\( h[i] = 0, \forall i \in G_T \)
for \( i = 1 \) to \( L \):
foreach \( j \in \text{getNeighbors}(i, 4) \):
\( \text{addEdge}(G, i, j, \text{cosineDist}(X[i], X[j])) \)
Update:
\( h[i] = \sum_{j \in \text{Neigh}(i)} S(E[i,j]) \cdot B[j] \cdot X[j], \forall i \in G_T \)
Compute:
\( output = C \cdot (A \cdot h + D \cdot X) \)
\( \eta_i \gets \frac{\partial \mathcal{L}}{\partial h_i} \)
for \( i \in G_T \) (Bottom-up & Top-down):
Aggregate: \( \eta_i \gets \bar{B}_i \eta_i + \sum_{j \in \text{Children}(i)} \eta_j \bar{A}_j \)
Propagate: \( \eta_j \gets \bar{B}_j \eta_i, \quad \forall j \in \text{Children}(i) \)
Calculate:
\( \mathcal{L}_{\text{scan}} = \sum_{i} \frac{1}{2} \| \eta_i \|^2 \)
Return: \( output \), gradients

Results

BibTeX


      @inproceedings{wang2025vastsd,
          title={VasTSD: Learning 3D Vascular Tree-state Space Diffusion Model for Angiography Synthesis},
          author={Wang, Zhifeng and Yi, Renjiao and Wen, Xin and Zhu, Chenyang and Xu, Kai},
          booktitle={Proceedings of the Computer Vision and Pattern Recognition Conference},
          pages={15693--15702},
          year={2025}
        }