Weakly supervised learning in domain transfer scenario for brain lesion segmentation in MRI

Pubali Chatterjee,Kaushik Das Sharma,Amlan Chakrabarti
DOI: https://doi.org/10.1007/s11042-023-17888-0
IF: 2.577
2024-01-05
Multimedia Tools and Applications
Abstract:In recent years, deep learning has revolutionized the field of medical image analysis. However, one of the biggest obstacles of deep learning based approaches remain at the level of large labelled data set requirement for training the network. While many of the state-of-the-art networks perform well in presence of abundance of training data, the problem of limited amount of data is a real hindrance to using these networks. Due to the lack of expert annotators, labelled training data can be very scarce in many clinical sites, especially in small scale facilities. Although there has been significant advancements in transfer learning strategies to exploit the availability of annotated data from other clinical sites, the network still demands a few fully annotated examples for fine tuning. In this work we consider the case where fully annotated data is scarce, and rather a set of weakly annotated data are available (for example, obtained from a few user scribbles). We propose a technique to handle this problem by using a two-stage network in a novel fashion. The first stage encoder-decoder network learns the examples from a cross domain. In addition, the network also has an average pooling mechanism to generate strong features from input images. The second stage of the network shares the weights from the initial encoder layers and exploits per pixel probability map to perform fine tuning by using dense Conditional Random Field (CRF) loss on weakly labelled data. We demonstrate that using this type of strategy, good results can be obtained with small amount of partially annotated training data. Experiments are performed in public datasets and results are compared with several recent state-of-the-art algorithms using standard metrics for segmentation performance evaluation.
computer science, information systems, theory & methods,engineering, electrical & electronic, software engineering
What problem does this paper attempt to address?