SHAP-Integrated Convolutional Diagnostic Networks for Feature-Selective Medical Analysis

Code

• github.com/aipmlab/sicdn
  OfficialIn paperpytorch★ 0

Abstract

This study introduces the SHAP-integrated convolutional diagnostic network (SICDN), an interpretable feature selection method designed for limited datasets, to address the challenge posed by data privacy regulations that restrict access to medical datasets. The SICDN model was tested on classification tasks using pneumonia and breast cancer datasets, demonstrating over 97% accuracy and surpassing four popular CNN models. We also integrated a historical weighted moving average technique to enhance feature selection. The SICDN shows potential in medical image prediction, with the code available on https://github.com/AIPMLab/SICDN.

Tasks

Reproductions