Explainable-by-design Semi-Supervised Representation Learning for COVID-19 Diagnosis from CT Imaging

  • Abel Díaz Berenguer
  • , Hichem Sahli
  • , Boris Joukovsky
  • , Maryna Kvasnytsia
  • , Ine Dirks
  • , Mitchel Alioscha-Pérez
  • , Nikos Deligiannis
  • , Panagiotis Gonidakis
  • , Sebastián Amador Sánchez
  • , Redona Brahimetaj
  • , Evgenia Papavasileiou
  • , Jonathan Cheung-Wai Chan
  • , Fei Li
  • , Shangzhen Song
  • , Yixin Yang
  • , Sofie Tilborghs
  • , Siri Willems
  • , Tom Eelbode
  • , Jeroen Bertels
  • , Dirk Vandermeulen
  • Frederik Maes, Paul Suetens, Lucas Fidon, Tom Vercauteren, David Robben, Arne Brys, Dirk Smeets, Bart Ilsen, Nico Buls, Nina Watté, Johan de Mey, Annemiek Snoeckx, Paul M. Parizel, Julien Guiot, Louis Deprez, Paul Meunier, Stefaan Gryspeerdt, Kristof De Smet, Bart Jansen, Jef Vandemeulebroucke

Research output: Working paperPreprint

Abstract

Our motivating application is a real-world problem: COVID-19 classification from CT imaging, for which we present a explainable Deep Learning approach based on a semi-supervised classification pipeline that employs variational
autoencoders to extract efficient feature embedding. We have optimized the architecture of two different networks for CT images: (i) a novel conditional variational autoencoder (CVAE) with a specific architecture that integrates the
class labels inside the encoder layers and uses side information with shared attention layers for the encoder, which make the most of the contextual clues for representation learning, and (ii) a downstream convolutional neural network
for supervised classification using the encoder structure of the CVAE. With the explainable classification results, the proposed diagnosis system is very effective for COVID-19 classification. Based on the promising results obtained
qualitatively and quantitatively, we envisage a wide deployment of our developed technique in large-scale clinical studies. Code is available at https://git.etrovub.be/AVSP/ct-based-covid-19-diagnostic-tool.git
Original languageEnglish
PublisherarXiv
Volumeabs/2011.11719
Publication statusPublished - 2 Sept 2021

Publication series

NameCoRR

UN SDGs

This output contributes to the following UN Sustainable Development Goals (SDGs)

  1. SDG 3 - Good Health and Well-being
    SDG 3 Good Health and Well-being

Fingerprint

Dive into the research topics of 'Explainable-by-design Semi-Supervised Representation Learning for COVID-19 Diagnosis from CT Imaging'. Together they form a unique fingerprint.

Cite this