SympTextML: Leveraging Natural Language Symptom Descriptions for Accurate Multi-Disease Prediction
This research presents an AI-driven framework for multi-disease classification using natural language symptom descriptions, optimized through large language model (LLM) oriented preprocessing techniques. The proposed system integrates essential NLP steps text normalization, lemmatization, and n-gram vectorization to convert unstructured clinical symptom data into machine-readable form. A publicly available dataset comprising 8,498 samples across ten common diseases, including pneumonia, heart attack, diabetes, stroke, asthma, and depression, was used for training and evaluation. Data balancing and cleaning ensured uniform class representation with 1,200 samples per disease category. The processed dataset was subjected to supervised machine learning models, including SVM, KNN, Decision Tree, Random Forest, and Extra Trees, to identify the most effective classifier. Experimental results, conducted in Google Colab, showed that ensemble models (Random Forest and Extra Trees) significantly outperformed the others, achieving 99% accuracy, precision, recall, and F1-scores, while SVM and Decision Tree followed closely with 98% performance across metrics. Notably, the models consistently predicted pneumonia with high confidence for relevant input queries , validating the framework's robustness. This work demonstrates the efficacy of integrating LLM-compatible preprocessing with traditional ML classifiers for accurate disease detection based on symptom narratives. The proposed approach serves as a foundational step toward developing scalable, intelligent healthcare support systems capable of real-time disease prediction and decision-making assistance.
Authors
- Dhairya Vyas (ORCID: https://orcid.org/0000-0001-7636-4167)
- Harsh Kantawala (ORCID: https://orcid.org/0000-0002-7363-0770)
- Milind Shah (ORCID: https://orcid.org/0009-0001-6077-3924)
- Brijesh Patel
- T. V. Patel
- Jalaja Enamala
Institutions
- Maharaja Sayajirao University of Baroda (IN)
- Sardar Vallabhbhai National Institute of Technology Surat (IN)
- Deccan School of Hospital Management (IN)
- Sardar Patel University (IN)
Publication Details
- Journal
- Journal of Electronics Electromedical Engineering and Medical Informatics
- Published
- 2025-07-14
- DOI
- https://doi.org/10.35882/jeeemi.v7i3.946
- Citations
- 1
- Primary Topic
- Topic Modeling
- Type
- article
- Field-Weighted Citation Impact
- 2.48