Junaid199f / evo_nas_med_2d_3d

Evolutionary Neural Architecture Search for 2D and 3D Medical Image Classification

Evolutionary Neural Architecture Search for 2D and 3D Medical Image Classification

Overview

• This repository contains the Python implementation of Evolutionary Framework for 2D and 3D Medical Image Classification
• The framework extended the DARTS search space for 2D and 3D Medical Image Classification and MealPy library for the implementation of metaheurstics.
• For more details, please refer to our paper Evolutionary Neural Architecture Search for 2D and 3D Medical Image Classification by Muhammad Junaid Ali, Laurent Moalic, Mokhtar Essaid and Lhassane Idoumghar. If you find this implementation helpful, please consider citing our work:

BibTex @inproceedings{ali2024evolutionary,
  title={Evolutionary Neural Architecture Search for 2D and 3D Medical Image Classification},
  author={Ali, Muhammad Junaid and Moalic, Laurent and Essaid, Mokhtar and Idoumghar, Lhassane},
  booktitle={International Conference on Computational Science},
  pages={131--146},
  year={2024},
  organization={Springer}
}

Datasets used

• [MedMNIST dataset] (https://medmnist.com/)
• [BreakHIS dataset] (https://web.inf.ufpr.br/vri/databases/breast-cancer-histopathological-database-breakhis/)