Semi-supervised Medical Image Classification with Global Latent Mixing

05/22/2020
by   Prashnna Kumar Gyawali, et al.
0

Computer-aided diagnosis via deep learning relies on large-scale annotated data sets, which can be costly when involving expert knowledge. Semi-supervised learning (SSL) mitigates this challenge by leveraging unlabeled data. One effective SSL approach is to regularize the local smoothness of neural functions via perturbations around single data points. In this work, we argue that regularizing the global smoothness of neural functions by filling the void in between data points can further improve SSL. We present a novel SSL approach that trains the neural network on linear mixing of labeled and unlabeled data, at both the input and latent space in order to regularize different portions of the network. We evaluated the presented model on two distinct medical image data sets for semi-supervised classification of thoracic disease and skin lesion, demonstrating its improved performance over SSL with local perturbations and SSL with global mixing but at the input space only. Our code is available at https://github.com/Prasanna1991/LatentMixing.

READ FULL TEXT
research
05/15/2020

Semi-supervised Medical Image Classification with Relation-driven Self-ensembling Model

Training deep neural networks usually requires a large amount of labeled...
research
09/23/2020

Enhancing Mixup-based Semi-Supervised Learning with Explicit Lipschitz Regularization

The success of deep learning relies on the availability of large-scale a...
research
03/20/2020

ROAM: Random Layer Mixup for Semi-Supervised Learning in Medical Imaging

Medical image segmentation is one of the major challenges addressed by m...
research
01/31/2023

NP-Match: Towards a New Probabilistic Model for Semi-Supervised Learning

Semi-supervised learning (SSL) has been widely explored in recent years,...
research
12/20/2021

Transformers Can Do Bayesian Inference

Currently, it is hard to reap the benefits of deep learning for Bayesian...
research
05/24/2021

Semi-supervised learning of images with strong rotational disorder: assembling nanoparticle libraries

The proliferation of optical, electron, and scanning probe microscopies ...
research
05/16/2023

Multi-Level Global Context Cross Consistency Model for Semi-Supervised Ultrasound Image Segmentation with Diffusion Model

Medical image segmentation is a critical step in computer-aided diagnosi...

Please sign up or login with your details

Forgot password? Click here to reset