18 May 2023 Deep clustering for abdominal organ classification in ultrasound imaging
Hind Dadoun, Hervé Delingette, Anne-Laure Rousseau, Eric de Kerviler, Nicholas Ayache
Author Affiliations +
Abstract

Purpose

The purpose of this study is to examine the utilization of unlabeled data for abdominal organ classification in multi-label (non-mutually exclusive classes) ultrasound images, as an alternative to the conventional transfer learning approach.

Approach

We present a new method for classifying abdominal organs in ultrasound images. Unlike previous approaches that only relied on labeled data, we consider the use of both labeled and unlabeled data. To explore this approach, we first examine the application of deep clustering for pretraining a classification model. We then compare two training methods, fine-tuning with labeled data through supervised learning and fine-tuning with both labeled and unlabeled data using semisupervised learning. All experiments were conducted on a large dataset of unlabeled images (nu = 84967) and a small set of labeled images (ns = 2742) comprising progressively 10%, 20%, 50%, and 100% of the images.

Results

We show that for supervised fine-tuning, deep clustering is an effective pre-training method, with performance matching that of ImageNet pre-training using five times less labeled data. For semi-supervised learning, deep clustering pre-training also yields higher performance when the amount of labeled data is limited. Best performance is obtained with deep clustering pre-training combined with semi-supervised learning and 2742 labeled example images with an F1-score weighted average of 84.1%.

Conclusions

This method can be used as a tool to preprocess large unprocessed databases, thus reducing the need for prior annotations of abdominal ultrasound studies for the training of image classification algorithms, which in turn could improve the clinical use of ultrasound images.

© 2023 Society of Photo-Optical Instrumentation Engineers (SPIE)
Hind Dadoun, Hervé Delingette, Anne-Laure Rousseau, Eric de Kerviler, and Nicholas Ayache "Deep clustering for abdominal organ classification in ultrasound imaging," Journal of Medical Imaging 10(3), 034502 (18 May 2023). https://doi.org/10.1117/1.JMI.10.3.034502
Received: 19 September 2022; Accepted: 27 April 2023; Published: 18 May 2023
Lens.org Logo
CITATIONS
Cited by 1 scholarly publication.
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Machine learning

Ultrasonography

Education and training

Image classification

Data modeling

Liver

Databases

Back to Top