A modular deep learning architecture for interpretable disease prediction across tabular clinical and biometric datasets

Accurate disease prediction using clinical datasets is essential for improving early diagnosis and clinical decision-support systems; however, many existing deep learning approaches are disease-specific, computationally intensive, and difficult to generalize across heterogeneous biomedical datasets. This study addresses this challenge by proposing a unified and dataset-aware deep learning framework that enables accurate and interpretable disease prediction across diverse clinical datasets. The framework adopts a modular architecture that selects appropriate models based on dataset characteristics such as feature dimensionality, sample size, and class imbalance. It integrates multiple deep learning architectures, including MLP, one-dimensional CNN, FT-Transformer, autoencoder-based classifiers, and ensemble strategies. Robust preprocessing, fold-safe feature selection, and nested cross-validation are incorporated to ensure reliable performance evaluation. The framework is evaluated on three heterogeneous benchmark datasets: the UCI Heart Disease dataset (303 samples, 13 clinical features), the PIMA Indians Diabetes dataset (768 samples, 8 metabolic features), and the Parkinson's disease voice dataset (195 recordings, 22 acoustic features). Experimental results demonstrate competitive predictive performance relative to classical baselines across the diverse tasks. The FT-Transformer + autoencoder ensemble achieved an AUC of 0.8980 (±0.0483) for heart disease prediction, while the CNN + Autoencoder ensemble obtained an AUC of 0.8451 (±0.0270) for diabetes classification. For Parkinson's disease detection, the MLP achieved an AUC of 0.7538 with perfect specificity. Overall, all models achieved AUC values comparable to ML baselines. The study contributes a scalable and interpretable deep learning framework that improves reliability, generalization, and practical applicability for multi-disease prediction in real-world healthcare environments.

Authors

Institutions

Publication Details

Journal
PLoS ONE
Published
2026-05-08
DOI
https://doi.org/10.1371/journal.pone.0348670
Primary Topic
Artificial Intelligence in Healthcare
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

A modular deep learning architecture for interpretable disease prediction across tabular clinical and biometric datasets

Sandip Bobade, Kirti A. Patil, Vijay U. Rathod, Aoudumber D. Londhe et al.
PLoS ONE
Artificial Intelligence in Healthcare
article

A modular deep learning architecture for interpretable disease prediction across tabular clinical and biometric datasets

Sandip Bobade, Kirti A. Patil, Vijay U. Rathod, Aoudumber D. Londhe, Siddhesh Sanjay Amrutkar, Melkamu Workie Kebede, Virendrakumar A. Dhotre
article en

Abstract

Accurate disease prediction using clinical datasets is essential for improving early diagnosis and clinical decision-support systems; however, many existing deep learning approaches are disease-specific, computationally intensive, and difficult to generalize across heterogeneous biomedical datasets. This study addresses this challenge by proposing a unified and dataset-aware deep learning framework that enables accurate and interpretable disease prediction across diverse clinical datasets. The framework adopts a modular architecture that selects appropriate models based on dataset characteristics such as feature dimensionality, sample size, and class imbalance. It integrates multiple deep learning architectures, including MLP, one-dimensional CNN, FT-Transformer, autoencoder-based classifiers, and ensemble strategies. Robust preprocessing, fold-safe feature selection, and nested cross-validation are incorporated to ensure reliable performance evaluation. The framework is evaluated on three heterogeneous benchmark datasets: the UCI Heart Disease dataset (303 samples, 13 clinical features), the PIMA Indians Diabetes dataset (768 samples, 8 metabolic features), and the Parkinson's disease voice dataset (195 recordings, 22 acoustic features). Experimental results demonstrate competitive predictive performance relative to classical baselines across the diverse tasks. The FT-Transformer + autoencoder ensemble achieved an AUC of 0.8980 (±0.0483) for heart disease prediction, while the CNN + Autoencoder ensemble obtained an AUC of 0.8451 (±0.0270) for diabetes classification. For Parkinson's disease detection, the MLP achieved an AUC of 0.7538 with perfect specificity. Overall, all models achieved AUC values comparable to ML baselines. The study contributes a scalable and interpretable deep learning framework that improves reliability, generalization, and practical applicability for multi-disease prediction in real-world healthcare environments.

PLoS ONEVol. 21(5)
Savitribai Phule Pune University (IN), Debre Markos University (ET)
Peace, Justice and strong institutions
Openalex Percentile: Top 5%
Artificial Intelligence in Healthcare
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.