Medical Knowledge-Guided Deep Learning for Imbalanced Medical Image Classification

11/20/2021
by   Long Gao, et al.
0

Deep learning models have gained remarkable performance on a variety of image classification tasks. However, many models suffer from limited performance in clinical or medical settings when data are imbalanced. To address this challenge, we propose a medical-knowledge-guided one-class classification approach that leverages domain-specific knowledge of classification tasks to boost the model's performance. The rationale behind our approach is that some existing prior medical knowledge can be incorporated into data-driven deep learning to facilitate model learning. We design a deep learning-based one-class classification pipeline for imbalanced image classification, and demonstrate in three use cases how we take advantage of medical knowledge of each specific classification task by generating additional middle classes to achieve higher classification performances. We evaluate our approach on three different clinical image classification tasks (a total of 8459 images) and show superior model performance when compared to six state-of-the-art methods. All codes of this work will be publicly available upon acceptance of the paper.

READ FULL TEXT

page 3

page 4

research
06/12/2020

HMIC: Hierarchical Medical Image Classification, A Deep Learning Approach

Image classification is central to the big data revolution in medicine. ...
research
05/31/2022

Deep learning pipeline for image classification on mobile phones

This article proposes and documents a machine-learning framework and tut...
research
09/29/2021

Does deep learning model calibration improve performance in class-imbalanced medical image classification?

In medical image classification tasks, it is common to find that the num...
research
09/24/2020

PK-GCN: Prior Knowledge Assisted Image Classification using Graph Convolution Networks

Deep learning has gained great success in various classification tasks. ...
research
01/27/2021

An Ultra-Specific Image Dataset for Automated Insect Identification

Automated identification of insects is a tough task where many challenge...
research
06/04/2022

Interpretable Models Capable of Handling Systematic Missingness in Imbalanced Classes and Heterogeneous Datasets

Application of interpretable machine learning techniques on medical data...
research
09/10/2020

MedMeshCNN – Enabling MeshCNN for Medical Surface Models

Background and objective: MeshCNN is a recently proposed Deep Learning f...

Please sign up or login with your details

Forgot password? Click here to reset