Home

Awesome

<div align="center"> <h1>MD-Dose </h1> This is the codebase for [MD-Dose: A Diffusion Model based on the Mamba for Radiotherapy Dose Prediction](https://arxiv.org/abs/2403.08479). </div>

Envs. for Pretraining

Train Your MD-Dose

python train.py(Training without a structural encoder)

python train_struc.py(Training with a structural encoder)

Acknowledgement :heart:

This project is based on Vim (paper, code), Causal-Conv1d (code). Thanks for their wonderful works.

Citation

If you find MD-Dose is useful in your research or applications, please consider giving us a star 🌟 and citing it by the following BibTeX entry.

@misc{fu2024mddose,
      title={MD-Dose: A Diffusion Model based on the Mamba for Radiotherapy Dose Prediction}, 
      author={Linjie Fu and Xia Li and Xiuding Cai and Yingkai Wang and Xueyao Wang and Yali Shen and Yu Yao},
      year={2024},
      eprint={2403.08479},
      archivePrefix={arXiv},
      primaryClass={eess.IV}
}