Edit this page

NA-MIC Project Weeks

Back to Projects List

Quality Control Model for Brain Surfaces

Key Investigators

Presenter location: In-person

Project Description

ShapeAXI is a shape analysis package that regroups many AI networks which use analysis via transformer networks or 2D convolutional neural networks. This package is available on Pypi and has been developed using Python and MONAI framework. The objective of ShapeAXI is to provide different architectures that can be used by anyone using his own data.

One of this network, called SaxiRing, has been used on the Adolescent Brain Cognitive Development (ABCD) data as a quality control (QC) model. One of the outputs of this architecture is a visual explanation from the regions of an input image that are most influential for the model’s decision.

The project would be to create the extension of this QC model and the visualization on 3D Slicer.

Objective

  1. Build and deploy the extension on 3D-slicer for the QC model and the visualization (GRAD-CAM)
  2. The end result would be to have a new 3D Slicer extension ready to be used for anyone who wants to use the QC model on his own data

Approach and Plan

  1. Create the extension into 3D-Slicer
  2. Implement the Extension Logic (organise the code, develop the Logic Module, develop the User Interface (UI))
  3. Integrate the QC model
  4. Integrate the GRAD-CAM
  5. Distribute the extension

Progress and Next Steps

  1. We are able to load the model (on Linux)
  2. We are able to run the prediction over a direcotry of subjects (on Linux)

Video Demo

Next steps :

  1. Make sure that all preliminary steps have no issue
  2. Start creating the extension
  3. Thinking about the best UI to improve the accessibility

Illustrations

QC Model Results

QC_DATA_1_TO_1_test_prediction_norm_confusion

Example of GRAD-CAM in 3D-Slicer

Screenshot 2024-06-14 at 10 12 44

Background and References