Biologically-inspired semi-supervised semantic segmentation for biomedical imaging

Area of Science:

• Computer Vision
• Machine Learning
• Biomedical Imaging

Background:

• Semantic segmentation models are crucial for analyzing biomedical images.
• Data scarcity in the biomedical domain presents a significant challenge for training deep learning models.
• Existing supervised methods struggle with limited labeled data.

Purpose of the Study:

• To develop a novel bio-inspired, two-stage semi-supervised learning approach for semantic segmentation.
• To address the challenge of data scarcity in biomedical image analysis.
• To improve the performance of semantic segmentation models using unsupervised feature learning.

Main Methods:

• A two-stage semi-supervised learning framework utilizing downsampling-upsampling architectures.
• Stage 1: Unsupervised feature discovery using the Hebbian principle ('fire together, wire together') for weight updates in convolutional and transpose-convolutional layers.
• Stage 2: Fine-tuning with standard backpropagation on a small subset of labeled data.

Main Results:

• The proposed method outperforms state-of-the-art (SOTA) approaches on biomedical datasets across various label availability levels.
• Initializing SOTA approaches with the unsupervised stage of this method leads to performance improvements.
• Demonstrated effectiveness in overcoming data scarcity challenges in medical image segmentation.

Conclusions:

• The novel bio-inspired semi-supervised approach effectively enhances semantic segmentation in data-scarce biomedical domains.
• Hebbian principle-based unsupervised learning provides a robust initialization strategy for deep learning models.
• This methodology offers a promising solution for improving medical image analysis and computer vision applications.