Pengoptimalan Seleksi Fitur Berbasis Particle Swarm Optimization pada Prediksi Gagal Jantung dengan Random Tree
DOI:
https://doi.org/10.26905/jasiek.v8i1.16595Keywords:
Feature Selection, Heart Failure, Particle Swarm, Optimization (PSO), Random TreeAbstract
Heart failure is one of the leading causes of hospitalization and mortality, particularly among older adults. Early detection is essential to support effective clinical decision-making. This study aims to develop a heart failure prediction model using the Random Tree classification algorithm optimized with Particle Swarm Optimization (PSO) for feature selection. Random Tree was chosen for its simplicity and interpretability, while PSO was employed to identify the most relevant features and remove less important ones. The dataset was obtained from the UCI Machine Learning Repository and consists of 299 patient records with 12 clinical attributes. Model performance was evaluated using accuracy, precision, recall, and Area Under the Curve (AUC). The baseline Random Tree model achieved an accuracy of 75.58% and an AUC of 0.632. After applying PSO-based feature selection, the optimized model achieved an accuracy of 82.27% and an AUC of 0.740. These findings indicate that integrating PSO with Random Tree effectively improves heart failure prediction performance and has potential as a clinical decision-support tool
Downloads
References
[1] R. K. Sachdeva, K. D. Singh, S. Sharma, P. Bathla, and V. Solanki, “An Organized Method for Heart Failure Classification,” 2023, doi: 10.1109/ESCI56872.2023.10099809.
[2] A. Ishaq et al., “Improving the Prediction of Heart Failure Patients’ Survival Using SMOTE and Effective Data Mining Techniques,” IEEE Access, vol. 9, pp. 39707–39716, 2021, doi: 10.1109/ACCESS.2021.3064084.
[3] D. Chicco and G. Jurman, “The advantages of the Matthews correlation coefficient (MCC) over F1 score and accuracy in binary classification evaluation,” BMC Genomics, vol. 21, no. 1, pp. 1–13, 2020, doi: 10.1186/s12864-019-6413-7.
[4] M. Mamun, A. Farjana, M. Al Mamun, M. S. Ahammed, and M. M. Rahman, “Heart failure survival prediction using machine learning algorithm: am I safe from heart failure?,” 2022, doi: 10.1109/AIIoT54504.2022.9817303.
[5] D. Chicco and L. Oneto, “An Enhanced Random Forests Approach to Predict Heart Failure From Small Imbalanced Gene Expression Data,” IEEE/ACM Trans. Comput. Biol. Bioinforma., vol. 18, no. 6, 2021, doi: 10.1109/TCBB.2020.3041527.
[6] M. Pal and S. Parija, “Prediction of Heart Diseases using Random Forest,” J. Phys. Conf. Ser., vol. 1817, no. 1, 2021, doi: 10.1088/1742-6596/1817/1/012009.
[7] R. A. M. Elghalid, A. Alwirshiffani, A. A. I. Mohamed, F. H. A. Aldeeb, and A. Andiasha, “Comparison of Some Machine Learning Algorithms for Predicting Heart Failure,” 2022, doi: 10.1109/ICEMIS56295.2022.9914325.
[8] Ridwansyah, R. L. Andharsaputri, Yudhistira, Irmawati Carolina, and Suharjanti, “K-Nearest Neighbors Optimization using Particle Swarm Optimization in Selection Digital Payments,” J. Teknol. Inf. dan Terap., vol. 12, no. 1, pp. 1–8, 2025, doi: https://doi.org/10.25047/jtit.v12i1.5911.
[9] H. Nurdin, I. Carolina, R. L. Andharsaputri, A. Wuryanto, and Ridwansyah, “Forward Selection as a Feature Selection Method in the SVM Kernel for Student Graduation Data,” Sink. J. dan Penelit. Tek. Inform., vol. 8, no. October, pp. 2531–2537, 2024, doi: 10.33395/sinkron.v8i4.14172.
[10] S. Sartini, S. Sumarna, A. Hamid, A. H. Kahf, and Nicodias Palasar, “Revolusi Diagnosis : Optimasi Random Tree-Pso Untuk Penyakit Ginjal Kronis,” JIRE (Jurnal Inform. Rekayasa Elektron., vol. 8, no. 1, pp. 149–158, 2025, doi: https://doi.org/10.36595/jire.v8i1.1542.
[11] I. Ariyati, S. Rosyida, K. Ramanda, V. Riyanto, S. Faizah, and Ridwansyah, “Optimization of the Decision Tree Algorithm Used Particle Swarm Optimization in the Selection of Digital Payments,” in Journal of Physics: Conference Series, 2020, vol. 1641, no. 1, doi: 10.1088/1742-6596/1641/1/012090.
[12] W. Widiati, N. Iriadi, I. Ariyati, I. Nawawi, and Sugiono, “Pendekatan Hibrida Decision Tree-Particle Swarm Optimization untuk Deteksi Dini Penyakit Ginjal Kronis,” JASIEK (Jurnal Apl. Sains, Informasi, Elektron. dan Komputer), vol. 6, no. 1, pp. 11–21, 2024, doi: 10.26905/jasiek.v6i1.13006.
[13] H. Nurdin, Suhardjono, A. Wuryanto, D. Yuliandari, and H. Sugiarto, “Naive Bayes and Particle Swarm Optimization in Early Detection Of Chronic Kidney Disease,” J. Tek. Inform., vol. 5, no. 3, pp. 703–708, 2024, doi: https://doi.org/10.52436/1.jutif.2024.5.3.1750.
[14] T. Prihatin, Sartini, W. E. Pangesti, Yudhistira, and R. Suryadithia, “EFISIENSI PSO DALAM PENGOPTIMALAN NEURAL NETWORK UNTUK CKD,” J. Inform. Manaj. dan Komput., vol. 16, no. 2, 2024, doi: http://dx.doi.org/10.36723/juri.v16i2.707.
[15] D. Yuliandari, A. Wuryanto, F. A. Sariasih, Sidik, and F. A. Sariasih, “Improving the Accuracy of Heart Failure Prediction Using the Particle Swarm Optimization Method,” Sink. J. dan Penelit. Tek. Inform., vol. 9, no. 1, pp. 210–220, 2024, doi: https://doi.org/10.33395/sinkron.v9i1.13017 e-ISSN.
[16] I. Nawawi, “Optimisasi Pemilihan Fitur Untuk Prediksi Gagal Jantung: Fusion Random Forest Dan Particle Swarm Optimization,” INTI NUSA MANDIRI, vol. 18, no. 2, pp. 122–128, 2024, doi: DOI: https://doi.org/10.33480/inti.v18i2.5031.
[17] Sumarna, Sartini, W. E. Pangesti, R. Suryadithia, and V. Riyanto, “Decision Tree Optimization in Heart Failure Diagnostics: a Particle Swarm Optimization Approach,” J. Tek. Inform., vol. 5, no. 3, pp. 739–746, 2024, doi: https://doi.org/10.52436/1.jutif.2024.5.3.1815.
[18] A. Hamid and Ridwansyah, “Optimizing Heart Failure Detection : A Comparison between Naive Bayes and Particle Swarm Optimization,” Paradigma, vol. 26, no. 1, pp. 30–36, 2024, doi: https://doi.org/10.31294/p.v26i1.3284.
[19] V. Riyanto, H. Destiana, T. Prihatin, Sugiono, and G. Wijaya, “Mengoptimalkan Prediksi Gagal Jantung Dengan Kombinasi Svm Dan Forward Selection,” JIRE (Jurnal Inform. Rekayasa Elektron., vol. 8, no. 1, pp. 103–111, 2025, doi: https://doi.org/10.36595/jire.v8i1.1541.
[20] C. Chodidjah, S. Rukiastiandari, F. Mutia, L. Rohimah, and A. Aprillia, “Forward Feature Selection for Heart Failure Classification Using the Random Tree Algorithm,” Voteteknika (Vocational Tek. Elektron. dan Inform., vol. 13, no. 2, 2025.
[21] H. Nurdin, I. Nawawi, A. Wuryanto, D. Yuliandari, and H. Sugiarto, “Prediksi Keterlambatan Pembayaran Mahasiswa untuk Mitigasi Risiko Cuti Menggunakan SVM Optimasi PSO,” J. Apl. Sains, Inf. , Elektron. dan Komput., vol. 7, no. 1, pp. 1–11, 2025, doi: https://doi.org/10.26905/jasiek.v7i1.15483.
Downloads
Published
How to Cite
Issue
Section
License
Copyright (c) 2026 JASIEK (Jurnal Aplikasi Sains, Informasi, Elektronika dan Komputer)
This work is licensed under a Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International License.
Authors who publish with this journal agree to the following terms:
The journal allow the authors to hold the copyright without restrictions and allow the authors to retain publishing rights without restrictions.
Authors retain copyright and grant the journal right of first publication with the work simultaneously licensed under a Creative Commons Attribution-ShareAlike 4.0 International License that allows others to share the work with an acknowledgement of the work's authorship and initial publication in this journal.
Authors are able to enter into separate, additional contractual arrangements for the non-exclusive distribution of the journal's published version of the work (e.g., post it to an institutional repository or publish it in a book), with an acknowledgement of its initial publication in this journal.
Authors are permitted and encouraged to post their work online (e.g., in institutional repositories or on their website) prior to and during the submission process, as it can lead to productive exchanges, as well as earlier and greater citation of published work (See The Effect of Open Access).
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.
