aL3x-O-o-Hung / GLCSA_ECLoss

Cross-Slice Attention and Evidential Critical Loss for Uncertainty-Aware Prostate Cancer Detection

This is the official code implementation of our MICCAI paper Cross-Slice Attention and Evidential Critical Loss for Uncertainty-Aware Prostate Cancer Detection

This is a followup work of our previous works, CSAM: A 2.5D Cross-Slice Attention Module for Anisotropic Volumetric Medical Image Segmentation and CAT-Net: A Cross-Slice Attention Transformer Model for Prostate Zonal Segmentation in MRI

Credits

If you use this code or the paper in your work, please cite our paper:

@article{hung2024cross,
  title={Cross-Slice Attention and Evidential Critical Loss for Uncertainty-Aware Prostate Cancer Detection},
  author={Hung, Alex Ling Yu and Zheng, Haoxin and Zhao, Kai and Pang, Kaifeng and Terzopoulos, Demetri and Sung, Kyunghyun},
  journal={arXiv preprint arXiv:2407.01146},
  year={2024}
}