Performance or Trust? Why Not Both. Deep AUC Maximization with Self-Supervised Learning for COVID-19 Chest X-ray Classifications

12/14/2021
by   Siyuan He, et al.
57

Effective representation learning is the key in improving model performance for medical image analysis. In training deep learning models, a compromise often must be made between performance and trust, both of which are essential for medical applications. Moreover, models optimized with cross-entropy loss tend to suffer from unwarranted overconfidence in the majority class and over-cautiousness in the minority class. In this work, we integrate a new surrogate loss with self-supervised learning for computer-aided screening of COVID-19 patients using radiography images. In addition, we adopt a new quantification score to measure a model's trustworthiness. Ablation study is conducted for both the performance and the trust on feature learning methods and loss functions. Comparisons show that leveraging the new surrogate loss on self-supervised models can produce label-efficient networks that are both high-performing and trustworthy.

READ FULL TEXT

page 1

page 2

page 3

research
04/07/2021

Self-Supervised Learning for Gastritis Detection with Gastric X-Ray Images

We propose a novel self-supervised learning method for medical image ana...
research
06/07/2022

Self-Knowledge Distillation based Self-Supervised Learning for Covid-19 Detection from Chest X-Ray Images

The global outbreak of the Coronavirus 2019 (COVID-19) has overloaded wo...
research
08/01/2023

DINO-CXR: A self supervised method based on vision transformer for chest X-ray classification

The limited availability of labeled chest X-ray datasets is a significan...
research
12/06/2020

Robust Deep AUC Maximization: A New Surrogate Loss and Empirical Studies on Medical Image Classification

Deep AUC Maximization (DAM) is a paradigm for learning a deep neural net...
research
12/19/2022

COVID-19 Detection Based on Self-Supervised Transfer Learning Using Chest X-Ray Images

Purpose: Considering several patients screened due to COVID-19 pandemic,...
research
07/19/2022

Towards Trustworthy Healthcare AI: Attention-Based Feature Learning for COVID-19 Screening With Chest Radiography

Building AI models with trustworthiness is important especially in regul...
research
09/13/2021

Self supervised learning improves dMMR/MSI detection from histology slides across multiple cancers

Microsatellite instability (MSI) is a tumor phenotype whose diagnosis la...

Please sign up or login with your details

Forgot password? Click here to reset