Skip to content

Official Code for "Representing Anatomical Trees by Denoising Diffusion of Implicit Neural Fields"

License

Notifications You must be signed in to change notification settings

sfu-mial/TreeDiffusion

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

25 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

$TrIND$: Representing Anatomical Trees by Denoising Diffusion of Implicit Neural Fields

This is the official code supporting our paper:

Representing Anatomical Trees by Denoising Diffusion of Implicit Neural Fields
Ashish Sinha and Ghassan Hamarneh
Simon Fraser University
[Paper] | [arxiv] | [Poster] | [Code]

🎉🎉 The Paper is now accepted to MICCAI '24. 🎉🎉

Abstract

Anatomical trees play a central role in clinical diagnosis and treatment planning. However, accurately representing anatomical trees is challenging due to their varying and complex topology and geometry. Traditional methods for representing tree structures, captured using medical imaging, while invaluable for visualizing vascular and bronchial networks, exhibit drawbacks in terms of limited resolution, flexibility, and efficiency. Recently, implicit neural representations (INRs) have emerged as a powerful tool for representing shapes accurately and efficiently. We propose a novel approach for representing anatomical trees using INR, while also capturing the distribution of a set of trees via denoising diffusion in the space of INRs. We accurately capture the intricate geometries and topologies of anatomical trees at any desired resolution. Through extensive qualitative and quantitative evaluation, we demonstrate high-fidelity tree reconstruction with arbitrary resolution yet compact storage, and versatility across anatomical sites and tree complexities.

image

How to Use

Setup Environment

python -m venv venv
source activate
pip install -r requirements.txt

Stage 1

python stage1.py --pcd_path /path/to/data --epochs <no. of iterations> --nsample <no. of points to sample>

Stage 2

python stage2.py --epochs <no. of epochs> --dset_name <dataset to use>

Cite

@inproceedings{sinha2024trind,
      title={Representing Anatomical Trees by Denoising Diffusion of Implicit Neural Fields}, 
      author={Ashish Sinha and Ghassan Hamarneh},
      booktitle={Proceedings of the International Conference on Medical Image Computing and Computer-Assisted Intervention (MICCAI)},
      year={2024},
}