Skip to content

Official Tensorflow implementation of "Choquet Integral and Coalition Game-based Ensemble of Deep Learning Models for COVID-19 Screening from Chest X-ray Images"

License

Notifications You must be signed in to change notification settings

CMATER-JUCS/Covid-Chestxray-lambda-fuzzy

 
 

Repository files navigation

Covid-Chestxray-lambda-fuzzy

Our solution for Novel COVID-19 Chestxray Repository

In this project, we have applied Choquet integral for ensemble of deep CNN models and propose a novel method for the evaluation of fuzzy measures using Coalition Game Theory, Information Theory and Lambda fuzzy approximation. Three different sets of Fuzzy Measures are calculated using three different weighting schemes along with information theory and coalition game theory. Using these three sets of fuzzy measures three Choquet Integrals are calculated and their decisions are finally combined.To the best of our knowledge,our experimental results outperform many state-of-the-art methods.

Table of Contents

Team Members

Journal Paper

If you find this work useful for your publications, please consider citing:

Installation

  1. Make sure you have python3 setup on your system
  2. Clone the repo
git clone https://github.com/subhankar01/Covid-Chestxray-lambda-fuzzy
  1. Install requirements
pip install -r requirements.txt

Dependencies

Our project is built using Python 3.8.6 and the following packages

numpy==1.19.5
pandas==1.1.5
matplotlib==3.2.2
seaborn==2.5.0
opencv-python==4.1.2
tensorflow==2.5.0

Method Overview

Fig 1:Flowchart of the proposed method

Dataset

We have used the Novel COVID-19 Chestxray Repository for evaluation of our proposed methodology. We have also used our code to show our method performance over the popular COVIDx dataset. Information about the Novel COVID-19 Chestxray Database and its parent image repositories is provided in Table 1

Table 1: Dataset Description

Dataset COVID-19 Pneumonia Normal
COVID Chestxray set 521 239 218
COVID-19 Radiography Database 219 1345 1341
Actualmed COVID chestxray dataset 12 0 80
Total 752 1584 1639

Results

Table 2: Results of 3-class classification

Classifier/Ensemble Validation Accuracy(in %) Test Accuracy(in %) Precision(Avg) Recall(Avg) AUC
VGG16 96.71 91.22 0.92 0.92 0.92
Xception 97.02 92.98 0.93 0.93 0.92
InceptionV3 97.49 93.48 0.94 0.94 0.94
Choquet Integral (Weight 1) 97.74 94.23 0.94 0.94 -
Choquet Integral (Weight 2) 98.24 94.23 0.94 0.94 -
Choquet Integral (Weight 3) 97.49 93.73 0.95 0.95 -
Ensemble 98.99 95.49 0.96 0.96 0.97

Fig 2:ROC of the 3 DCNN models and proposed ensemble method

Fig 3:Multi-labelled ROC curve of the proposed ensemble method

Fig 4:Confusion Matrix of the proposed method

Contact

In case of doubt or further collaboration, feel free to email us ! 😊

About

Official Tensorflow implementation of "Choquet Integral and Coalition Game-based Ensemble of Deep Learning Models for COVID-19 Screening from Chest X-ray Images"

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 81.8%
  • Jupyter Notebook 18.2%