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
- Pritendra Kumar Malakar
- Parul Dubey (ORCID: https://orcid.org/0000-0001-8903-6664)
- R Kanesaraj Ramasamy (ORCID: https://orcid.org/0000-0002-2780-6069)
- Abhinav Shukla (ORCID: https://orcid.org/0009-0002-2739-329X)
- Ganga Sagar Soni
Institutions
- Multimedia University (MY)
- Symbiosis International University (IN)
- Dr. C. V. Raman University (IN)
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
- Multimedia University