/MAEMI

3D masked autoencoder for anomaly detection

Primary LanguagePythonOtherNOASSERTION

3D Masked Autoencoders with Application to Anomaly Detection in Non-Contrast Enhanced Breast MRI

Code for your breast cancer anomaly detection paper.

MAEMI model

  • utils/ contains all the files to download the data and prepare patches
  • training/ involves the developed 3D masked autoencoder model and training scripts
  • inference/ holds files to generate anomaly maps and subtraction images

Cite as

@article{lang2023MAEMI,
  title={3D Masked Autoencoders with Application to Anomaly Detection in Non-Contrast Enhanced Breast MRI},
  author={Lang, Daniel M and Schwartz, Eli and Bercea, Cosmin I and Giryes, Raja and Schnabel, Julia A},
  journal={arXiv preprint arXiv:2303.05861},
  year={2023}
}