Git Product home page Git Product logo

deep-learning-explainability-for-breast-cancer-detection-in-mammography's Introduction

Deep-Learning-Explainability-for-Breast-Cancer-Detection-in-Mammography

This code was developed as a master thesis project for the Erasmus Mundus Joint Master's Degree in MedicAl Imaging and Applications (MAIA) by the University of Girona, the University of Cassino and Souther Lazio & the University of Burgundy. The work was developed in collaboration with the ViCOROB research group in the University of Girona. Highest TP part 1

Main contributors

  • Karla Sam Millan
  • Robert Martí Marly

Description

This repository offers the code for a qualitative and quantitative analysis of different XAI methods trained for breast cancer detection in mammography. The chosen explainability methods are:

  • Saliency maps
  • Occlusion
  • Integrated Gradients
  • Guided GradCAM
  • DeepLIFT
  • SHAP
  • LIME

Repository contents

  • main: contains the main code for training the network and subsequently obtaining the attribution maps for each of the methods, as well as the IOU scores.
  • aux_functions: contains the auxiliary functions used in main, mainly for the visualization of the model's performance, as well as the visualization of the attribution maps and the generation of the bounding boxes.
  • models: contains the code for loading models from pytorch.
  • best_models: contain the trained EfficientNetB0 and ResNet-50 models for whole-mammogram classification and patch-based classification, respectively.
  • iou_scores: a csv file containing the IOU scores computed with the ground truth bounding boxes and the bounding boxes from the attribution maps for every image in the validation set for each of the explainability methods.
  • documentation: contains the thesis paper, presentation and poster.

deep-learning-explainability-for-breast-cancer-detection-in-mammography's People

Contributors

karla-sam avatar

Stargazers

 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.