Source-linked AI summary

A Tour of Unsupervised Deep Learning for Medical Image Analysis

Khalid Raza, Nripendra Kumar Singh

arXiv:1812.07715v1eess.IVcs.CVcs.LG

TL;DR

Medical images are commonly interpreted by human experts, and pathology variation and expert fatigue have prompted computer-assisted interventions. The review surveys unsupervised deep-learning models and their medical-image applications, tools, benchmark datasets, opportunities, and challenges. The review identifies semantic segmentation at acceptable accuracy as a continuing challenge in medical-image analysis.

  • Problem

    Medical images are commonly interpreted by human experts, and pathology variation and expert fatigue have prompted computer-assisted interventions.

    Source: §1 · p. 1
  • Method

    The review surveys unsupervised deep-learning models and their medical-image applications, tools, benchmark datasets, opportunities, and challenges.

    Source: §1 · p. 2
  • Results

    The review identifies semantic segmentation at acceptable accuracy as a continuing challenge in medical-image analysis.

    Source: §6 · p. 22
  • Takeaways & Limitations

    Unsupervised models are also used for tasks including compression and dimensionality reduction, beyond classification.

    Source: §6 · p. 21

    Because unsupervised learning lacks labels, quantifying accuracy is nearly impossible, motivating a need for objective performance measures.

    Source: §6 · p. 21

Abstract

from arXiv · show

Interpretation of medical images for diagnosis and treatment of complex disease from high-dimensional and heterogeneous data remains a key challenge in transforming healthcare. In the last few years, both supervised and unsupervised deep learning achieved promising results in the area of medical imaging and image analysis. Unlike supervised learning which is biased towards how it is being supervised and manual efforts to create class label for the algorithm, unsupervised learning derive insights directly from the data itself, group the data and help to make data driven decisions without any external bias. This review systematically presents various unsupervised models applied to medical image analysis, including autoencoders and its several variants, Restricted Boltzmann machines, Deep belief networks, Deep Boltzmann machine and Generative adversarial network. Future research opportunities and challenges of unsupervised techniques for medical image analysis have also been discussed.

1. Introduction

Medical imaging supports diagnosis and treatment, while human interpretation and supervised approaches face challenges when supervision is unavailable, inadequate, or biased. The chapter surveys unsupervised deep learning models, their medical-image applications, tools, datasets, opportunities, and challenges.

1. Introduction

  • Medical imaging techniques are used for early detection, diagnosis, and treatment of complex diseases, with images often interpreted by human experts.The chapter describes computer-assisted medical image analysis as an automated alternative to manual interpretation.Source: §1 · p. 1
  • When human supervision is unavailable, inadequate, or biased, supervised algorithms may not be directly usable.The chapter presents unsupervised deep learning as an approach applied to medical image analysis and other medical and engineering problems.Source: §1 · p. 2
  • The chapter covers unsupervised deep learning models, medical-image applications, software tools and packages, benchmark datasets, opportunities, and future challenges.Source: §1 · p. 2

2. Why Unsupervised Learning?

Supervised workflows require labels and can reflect the way they are supervised, while unsupervised learning derives insights from data for data-driven decisions and a range of tasks.

2. Why Unsupervised Learning?

  • Supervised workflows require substantial manual labeling and can carry supervision-related biases that may limit consideration of corner cases.The passage also identifies reduced scalability of the target function as a limitation.Source: §2 · p. 2
  • Unsupervised learning derives insights from data and uses them for data-driven decisions without knowing the eventual task in advance.The passage describes it as a possible preparation step for supervised tasks, where learned representations may allow better classifier generalization (Jabeen et al., 2018).Source: §2 · p. 2
  • Unsupervised learning is presented as useful beyond classification for compression, dimensionality reduction, denoising, and super resolution.Source: §2 · p. 2

3. Taxonomy of Unsupervised Learning Tasks

Unsupervised learning groups unlabeled data according to underlying hidden features, providing information about the raw data.

3. Taxonomy of Unsupervised Learning Tasks

  • Unsupervised learning groups unlabeled data based on underlying hidden features.Source: §3 · p. 2
  • Grouping data through unsupervised learning yields at least some information about the raw data.Source: §3 · p. 2
  • The passage characterizes the grouped inputs as an unlabeled dataset rather than data with supplied class labels.Source: §3 · p. 2

3.1 Density estimation

Density estimation is described as a non-parametric way to discover structure in unlabeled data without strong distributional assumptions; KDE and histogram-based methods offer distinct estimation approaches.

3.1 Density estimation

  • Density estimation discovers intrinsic features and structure in complex unlabeled data through a non-parametric approach.Unlike parametric estimation, it is described as having few restrictions and distributional assumptions.Source: §3.1 · p. 2
  • The passage describes density estimation without prior functional assumptions as capable of producing a broad range of functions from data.Source: §3.1 · p. 3

3.1.1 Kernel density estimation

  • KDE produces a probabilistic distribution resembling an observed variable and is used for smoothing, exploratory analysis, and visualization.The passage lists Gaussian mixture and multivariate Gaussian mixture models among proposed kernels.Source: §3.1.1 · p. 3
  • KDE requires no model specification because the data determine the estimate.Source: §3.1.1 · p. 3
  • With sufficient samples, KDE estimation can converge to any density shape and generalize to higher dimensions.Source: §3.1.1 · p. 3
  • KDE is described as accommodating multivariate, multimodal densities with irregular cluster shapes.Source: §3.1.1 · p. 3

3.1.2 Histogram density estimation

  • Histogram-based estimation smooths the reconstructed density curve, with smoothness optimized by kernel parameters.The passage relates this technique to KNN density estimation (Bishop et al., 2006).Source: §3.1.2 · p. 3

3.2 Dimensionality reduction

Dimensionality reduction seeks a lower-dimensional representation that conveys the original data concisely; the section describes PCA and kernel PCA as approaches to this task.

3.2 Dimensionality reduction

  • Dimensionality reduction removes unwanted dimensions while retaining information concisely; correlated variables can also be grouped into fewer underlying factors.The passage motivates reduction by redundancy and noise in captured data, and identifies exploratory and confirmatory factor analysis as two approaches.Source: §3.2 · p. 3, §S.1.2 · p. 4

3.2.2 Principal component analysis

  • PCA maps data to linear combinations that maximize variance, retaining eigenvectors for the largest eigenvalues while potentially losing some data.It uses a covariance or correlation matrix; simple PCA cannot construct nonlinear mappings, though kernel techniques can implement nonlinear classification.Source: §3.2.2 · pp. 4–5

3.2.3 Kernel PCA

  • Kernel PCA extends conventional PCA to nonlinear subspaces, with the mapping depending on input magnitude and problem setup.For respiratory-signal extraction from X-ray fluoroscopy, Fischer et al. (2017) reported that Multi-Resolution PCA performed best among most parameters; PCANet and C-RBH-PCANet are also discussed as unsupervised feature-learning frameworks.Source: §3.2.3 · pp. 5–6

3.3 Clustering

Clustering groups unlabeled patterns by similarity to reveal structure in data. The section contrasts hierarchical clustering with partitional k-means, whose squared-error objective is NP-hard to minimize.

3.3 Clustering

  • Clustering groups unlabeled patterns or feature vectors by similarity to reveal structure in data.The passage notes that clustering is used across areas including image analysis, bioinformatics, and computer vision.Source: §3.3 · p. 6

3.3.1 Hierarchical clustering

  • Hierarchical clustering recursively merges the most similar clusters from singletons or divides one cluster into smaller groups, yielding a dendrogram of patterns and similarity levels.These approaches are respectively agglomerative (bottom-up) and divisive (top-down).Source: §3.3.1 · p. 7

3.3.2 Partitional (k-means) clustering

  • K-means is a widely used partitional method that minimizes squared error between cluster elements and their empirical means, but the objective is NP-hard even for k = 2.As a greedy algorithm, k-means can be expected to converge only to local minima.Source: §3.3.2 · p. 7

4. Unsupervised deep learning models

This section surveys unsupervised deep-learning architectures used in medical image analysis, with autoencoders and their variants forming a major model family. The variants differ in how they constrain, corrupt, stack, or probabilistically represent inputs and features.

4. Unsupervised deep learning models

  • The review classifies unsupervised deep-learning models and reports that autoencoders and their variants are extensively applied in medical image analysis.It identifies autoencoders as a family within its broader overview of unsupervised architectures.Source: §4 · p. 7, §4.1 · p. 8

4.1.1 Autoencoders and Stacked autoencoder

  • Basic autoencoders encode inputs into latent representations and minimize reconstruction error; stacking them adds depth and greedy layer-wise training.Stacked autoencoders can capture hierarchical groupings and have greater expressive power than a single-layer autoencoder.Source: §4.1.1 · pp. 8–9

4.1.2 Denoising autoencoder

  • Denoising autoencoders reconstruct clean inputs from corrupted versions, discouraging trivial identity solutions while learning features useful for denoising.Their training minimizes reconstruction loss between the clean input and its reconstruction from the corrupted input.Source: §4.1.2 · p. 9

4.1.3 Sparse autoencoder

  • Sparse autoencoders constrain average hidden-unit activity, allowing many hidden units while regularizing model complexity and helping prevent overfitting.Sparsity can be induced with a training loss penalty or by zeroing selected strong activations; k-sparse variants retain the k highest activations.Source: §4.1.3 · pp. 9–10

4.1.4 Convolutional autoencoder

  • Convolutional autoencoders are presented as image-oriented models that preserve local data structure and avoid feature-space distortion.They also offer an end-to-end alternative to stacked autoencoders’ fully connected layers and layer-wise pretraining.Source: §4.1.4 · p. 10, Figure 5 · p. 11

4.1.5 Variational autoencoder

  • Variational autoencoders use a probabilistic latent representation and optimize encoder and decoder parameters to maximize training-data likelihood.The section describes the encoder qϕ(z|x) as an approximation to the generative model pθ(x, z), using SGVB and AEVB estimators.Source: §4.1.5 · p. 12

4.1.6 Contractive autoencoder

  • Contractive autoencoders add an explicit regularizer based on the activation Jacobian’s squared Frobenius norm to encourage robustness to input variation.The review distinguishes their robustness of representation from denoising autoencoders’ robustness of reconstruction.Source: §4.1.6 · p. 12

4.2. Restricted Boltzmann Machines

The section introduces RBMs and surveys DBNs, DBMs, and GANs as unsupervised architectures used in medical image analysis. It describes their model structures and reports selected application examples, including cardiac-arrhythmia recognition results.

4.2. Restricted Boltzmann Machines

  • RBMs are bipartite undirected generative models with bidirectional visible-hidden connections, used in medical-image tasks including disease-variation detection, segmentation, dimensionality reduction, and feature learning.The restriction removes within-layer connections and is described as making training more efficient than for general Boltzmann machines.Source: §4.2 · p. 14
  • For cardiac arrhythmia classification, average recognition accuracy was 93.63% for ventricular ectopic beats and 95.57% for supraventricular ectopic beats, respectively.Source: §4.2 · p. 14

4.3. Deep Belief Networks

  • DBNs stack RBMs hierarchically and use greedy layer-wise unsupervised training, with lower layers learning low-level features before higher layers learn high-level features.The section also lists applications including feature extraction and autism-spectrum-disorder classification.Source: §4.3 · pp. 15–16
  • For cardiac arrhythmia classification, average recognition accuracy was 93.63% for ventricular ectopic beats and 95.57% for supraventricular ectopic beats, respectively.Source: §4.3 · p. 16

4.4. Deep Boltzmann Machine

  • DBMs stack RBMs into an undirected generative model that combines information from lower and upper layers, which the review says improves representation power compared with DBNs.The section describes DBMs as designed to handle ambiguous inputs robustly.Source: §4.4 · p. 17

4.5. Generative Adversarial Network (GAN)

  • GANs train a generator and discriminator in competition as a minimax game, and the review notes their use in generating photorealistic images.The generator captures the data distribution while the discriminator estimates whether a sample comes from training data rather than the generator.Source: §4.5 · pp. 17–18

5. List of software tools/packages and benchmark datasets

This section points readers to software packages and benchmark medical-image datasets for unsupervised-learning research. It gives examples of tools and dataset collections alongside the referenced inventories.

5. List of software tools/packages and benchmark datasets

  • The review lists software tools and packages implementing the unsupervised models it discusses.The inventory is identified as Table 7.Source: §5 · p. 18, Table 7 · p. 19
  • Examples of listed resources include DCGAN.torch, conditional adversarial networks for image-to-image translation, and an energy-based GAN implementation.The passages identify DCGAN.torch as unsupervised representation-learning software and describe the other entries as GAN implementations.Source: §5 · p. 19
  • The benchmark-dataset inventory includes OASIS MRI/PET data for normal aging or Alzheimer’s disease and TCIA collections of MRI, CT, and other images.The section also lists MedPix as an online database of medical images, teaching cases, and clinical topics.Source: Table 8 · p. 20, §5 · p. 20

6. Discussion, opportunities and challenges

The discussion presents unsupervised learning as an opportunity for medical-image analysis while identifying evaluation, data-fit, and interpretability challenges. It also outlines technical and organizational needs for broader applications.

6. Discussion, opportunities and challenges

  • Unsupervised learning can support exploratory analysis and preprocessing for supervised algorithms by generating new data representations.The review says such preprocessing may improve learning accuracy and reduce memory and time overheads.Source: §6 · p. 21
  • Without labels, it is nearly impossible to quantify unsupervised accuracy, motivating a need for objective performance measures.The review illustrates the issue with uncertainty about whether K-means found the right clusters.Source: §6 · p. 21
  • Algorithm choice depends on image-data type, hardware needs vary by problem, and clustering for segmentation works only when images fit natural groups.The review frames suitability as dependent on the problem at hand.Source: §6 · p. 21
  • Unsupervised learning is not a common choice in medical-image analysis; the reviewed models are mostly used to learn feature hierarchies for classification.The review presents this as the current pattern in the literature, while noting potential for unlabeled data and combined unsupervised and supervised tasks.Source: §6 · p. 21
  • The review calls for algorithms that handle heterogeneous image data and construct patient-specific anatomical and organ models from medical images.It also identifies medical-image semantic segmentation at acceptable accuracy as a challenge.Source: §6 · p. 22
  • Real-time transmission of data-hungry 3D medical video over limited-bandwidth channels requires efficient encoding and decoding.The discussion connects 3D video to telemedicine, U-healthcare, diagnosis, and potential benefits for remote surgery.Source: §6 · p. 22
  • Deep-learning models are difficult to interpret, leaving their acceptance by health professionals questionable.The review describes the path from model output to its explanation as complicated.Source: §6 · p. 22
  • The discussion identifies unsupervised learning as an open research area for developing training methods and stronger feature representations.It also points to inter-organizational collaboration among clinicians, radiologists, image analysts, and computational data analysts as important for medical-image CAD.Source: §6 · pp. 22–23

7. Conclusion

Medical image analysis uses both supervised and unsupervised learning, but human supervision may be unavailable, inadequate, or biased. Unsupervised models derive insights directly from data and may serve as a representation-preparation step for supervised tasks.

  • Human supervision may be unavailable, inadequate, or biased, motivating the use of both supervised and unsupervised approaches in medical image analysis.The field has also shifted toward computer-assisted diagnosis with the availability of deep learning and cheap cloud computing infrastructure.Source: §7 · p. 23
  • Unsupervised learning derives insights directly from data and supports data-driven decisions, with uses including compression, dimensionality reduction, denoising, and super resolution.The passage also describes these models as supporting some degree of decision making.Source: §7 · p. 23
  • Unsupervised representation learning may serve as preprocessing for supervised tasks and allow a classifier to generalize better.Source: §7 · p. 23
Loading 1812.07715v1…