Vessel Segmentation With U-Net

Victoria D'Agostino     

vwd@duke.edu    

Paper PDF
The mission of the Center for Global Womens Health Technology (GWHT) is to use optical technologies produce widely accessible healthcare technologies, in particular for womens cancers. Current work involves in vivo models of cancer. Angiogenesis, an important hallmark of cancer, is a desirable endpoint to add. This motivated me to develop a convolutional neural network to segment vasculature. To achieve this I implemented a U-Net architecture with a trainable mask physical layer. I have achieved metrics of 0.1975 loss and 0.9244 accuracy after 30 epochs training with 134 images. I have shown promising results with the addition of a trainable mask. Training with only 20 images, I achieved training loss of 0.392 and validation loss of 0.393, which was an improvement over the same model trained with 20 images without the trainable mask, which yielded training loss of 0.567 and the validation loss was 0.495.
Paper:
Code:
  • My code without the physical layer: Link to CoLab Notebook
  • My code with the physical layer: Link to CoLab Notebook