Search papers, labs, and topics across Lattice.
This paper introduces a prototype parts-based neural network tailored for interpretable classification of medical tabular data. The model learns prototypical parts by applying trainable patching over discretized features, representing patient data as binary or discretized feature subsets. Experiments on medical benchmark datasets demonstrate competitive classification performance compared to baselines, while providing interpretable concept-based predictions through comparison with learned prototypes.
Achieve human-readable interpretability in medical tabular data classification without sacrificing accuracy by learning and comparing against prototypical patient feature subsets.
The ability to interpret machine learning model decisions is critical in such domains as healthcare, where trust in model predictions is as important as their accuracy. Inspired by the development of prototype parts-based deep neural networks in computer vision, we propose a new model for tabular data, specifically tailored to medical records, that requires discretization of diagnostic result norms. Unlike the original vision models that rely on the spatial structure, our method employs trainable patching over features describing a patient, to learn meaningful prototypical parts from structured data. These parts are represented as binary or discretized feature subsets. This allows the model to express prototypes in human-readable terms, enabling alignment with clinical language and case-based reasoning. Our proposed neural network is inherently interpretable and offers interpretable concept-based predictions by comparing the patient's description to learned prototypes in the latent space of the network. In experiments, we demonstrate that the model achieves classification performance competitive to widely used baseline models on medical benchmark datasets, while also offering transparency, bridging the gap between predictive performance and interpretability in clinical decision support.
