Skip to content

A repository for the 02-740 Bioimage Informatics course project on GNN-based cancer detection and classification.

Notifications You must be signed in to change notification settings

simonlevine/GNN-based-diag

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

96 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

GNN-based-diag

A repository for the 02-740 Bioimage Informatics course project on GNN-based cancer detection and classification.

Initial Results:

  • We use RGB values on truncated portions of this dataset (https://www.kaggle.com/andrewmvd/malignant-lymphoma-classification) as node labels.

  • Euclidean pairwise distances are used as edge labels.

  • Graph is fully connected.

  • All of the above are good targets for improvement.

  • 120 epochs with a baseline GCN: Run 1

  • 120 epochs with a resnet-style skip connection GCN: Run 2

Future Work:

  • Use a dataset with more heterogeneous classes with respect to nodes.
  • Tweak the model archicture.

About

A repository for the 02-740 Bioimage Informatics course project on GNN-based cancer detection and classification.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages