PL EN
Superpixel-graph self-supervised pretraining with joint-embedding and contrastive objectives for wound image segmentation
 
Więcej
Ukryj
1
Computer Science Department, Ukrainian National Forestry University, 103 Gen. Chuprynky St. room 45, Lviv, Ukraine, 79057
 
2
PhD, Associate Professor, Computer Design Systems Department, Lviv Polytechnic National University, 3 Metropolitan Andrey St. IV educational building room 324, Lviv, Ukraine, 79013
 
 
Autor do korespondencji
Bohdan Lukashchuk   

Computer Science Department, Ukrainian National Forestry University, 103 Gen. Chuprynky St. room 45, Lviv, Ukraine, 79057
 
 
 
SŁOWA KLUCZOWE
DZIEDZINY
STRESZCZENIE
Annotated data remain a major problem and bottleneck for supervised learning approaches, especially in wound imaging. In this work, we propose a self-supervised learning pretraining method that represents a wound image as a superpixel adjacency graph, generated using the simple linear iterative clustering algorithm and uses graph structure to define building blocks for calculating joint-embedding and contrastive learning objectives. We evaluate three objective variants - contrastive, joint-embedding and their combination as a pretraining stage for further finetuning for the wound segmentation task. After self-supervised pretraining on 100 unlabelled images and supervised fine-tuning on only 30 labelled images using the contrastive objective, we received a Dice score of 0.266 on a 400-image test set, compared to 0.354 for a fully supervised U-Net trained on all 100 annotated images, thus having gap of 0.088 in Dice score with more than three times less annotated data. The results suggest that superpixel-graph-based self-supervised learning is a promising pretraining strategy for wound analysis in settings with limited annotated data.
Journals System - logo
Scroll to top