DFSel-FT: A Differentiable Feature Selection and FT-Transformer Framework for Interpretable Thyroid Disease Classification Using Tabular Data

Thyroid diseases are very common endocrine diseases that afflict millions of people around the world and need proper and timely diagnosis to ensure proper treatment. Although machine learning and hybrid metaheuristic methods have advanced, current models have high computation costs, low interpretability, and low probability calibration, which limit their use in clinical settings. In this research, a new DFSel-FT (Differentiable Feature Selection and an FT-Transformer) system is suggested, which combines DFSel-FT to allow one to diagnose thyroid disease effectively and interpretably. It employs Concrete (Gumbel-Softmax) gates to select the features end-to-end to make sure that only the most relevant clinical attributes are carried through the training. A Transformer-based architecture is then used to process the chosen features to learn intricate interdependencies. The model is trained with class-balanced focal loss and temperature scaling to better enhance calibration. Experimental evaluation on the UCI Thyroid Disease Dataset (22,632 samples) showed that the proposed model achieved 97.85% accuracy, 97.65% Macro-F1, and 98.10% AUC-OVR, showing competitive performance compared with traditional machine learning models, modern tabular deep learning baselines, and hybrid metaheuristic methods. Other indicators of robustness and reliability include MCC (0.955), Cohen Kappa (0.951), and small calibration error (ECE = 0.021). SHAP and LIME explainability analysis reveals clinically relevant features that include TSH, TT4, and T3. The proposed framework provides a balanced integration of predictive performance, interpretability, and probability calibration, making it a promising benchmark-level framework for interpretable and calibrated thyroid disease classification, requiring external clinical validation before real-world deployment.

Authors

Institutions

Publication Details

Journal
Computers
Published
2026-05-22
DOI
https://doi.org/10.3390/computers15060332
Primary Topic
Artificial Intelligence in Healthcare
Type
article
Field-Weighted Citation Impact
0.00

Funders

Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

DFSel-FT: A Differentiable Feature Selection and FT-Transformer Framework for Interpretable Thyroid Disease Classification Using Tabular Data

Pritendra Kumar Malakar, Parul Dubey, R Kanesaraj Ramasamy, Abhinav Shukla et al.
Computers
Artificial Intelligence in Healthcare
article

DFSel-FT: A Differentiable Feature Selection and FT-Transformer Framework for Interpretable Thyroid Disease Classification Using Tabular Data

Pritendra Kumar Malakar, Parul Dubey, R Kanesaraj Ramasamy, Abhinav Shukla, Ganga Sagar Soni
article en

Abstract

Thyroid diseases are very common endocrine diseases that afflict millions of people around the world and need proper and timely diagnosis to ensure proper treatment. Although machine learning and hybrid metaheuristic methods have advanced, current models have high computation costs, low interpretability, and low probability calibration, which limit their use in clinical settings. In this research, a new DFSel-FT (Differentiable Feature Selection and an FT-Transformer) system is suggested, which combines DFSel-FT to allow one to diagnose thyroid disease effectively and interpretably. It employs Concrete (Gumbel-Softmax) gates to select the features end-to-end to make sure that only the most relevant clinical attributes are carried through the training. A Transformer-based architecture is then used to process the chosen features to learn intricate interdependencies. The model is trained with class-balanced focal loss and temperature scaling to better enhance calibration. Experimental evaluation on the UCI Thyroid Disease Dataset (22,632 samples) showed that the proposed model achieved 97.85% accuracy, 97.65% Macro-F1, and 98.10% AUC-OVR, showing competitive performance compared with traditional machine learning models, modern tabular deep learning baselines, and hybrid metaheuristic methods. Other indicators of robustness and reliability include MCC (0.955), Cohen Kappa (0.951), and small calibration error (ECE = 0.021). SHAP and LIME explainability analysis reveals clinically relevant features that include TSH, TT4, and T3. The proposed framework provides a balanced integration of predictive performance, interpretability, and probability calibration, making it a promising benchmark-level framework for interpretable and calibrated thyroid disease classification, requiring external clinical validation before real-world deployment.

ComputersVol. 15(6)
Multimedia University (MY), Symbiosis International University (IN), Dr. C. V. Raman University (IN)
Multimedia University
Openalex Percentile: Top 2%
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.