Optimization of Sleep Disorder Classification Using ANN with Multi-Method Feature Selection

Authors

  • Devi Nova Kharisma Nahdlatul Ulama Sunan Giri University, Indonesia
  • Ifnu Wisma Dwi Prastya Nahdlatul Ulama Sunan Giri University, Indonesia
  • Ita Aristia Saida Nahdlatul Ulama Sunan Giri University, Indonesia
Pages Icon

DOI:

https://doi.org/10.63158/journalisi.v8i2.1473

Keywords:

Artificial Neural Network, Sleep Disorder Classification, Multiclass Classification, Sleep Health Dataset, Feature Selection

Abstract

Sleep disorders are health problems that can affect quality of life and have the potential to increase the risk of various chronic diseases. Therefore, a computational approach is needed to accurately and efficiently classify sleep disorders. The ANN model used has a two-layer hidden architecture with 128 and 64 neurons, respectively, and uses the ReLU activation function, equipped with a dropout layer to reduce overfitting. Three neurons with a softmax activation function make up the output layer, which produces probabilities for every class. To improve model performance, three feature selection methods were compared, namely Chi-Square, Information Gain, and Pearson Correlation. The test results showed that the ANN model without feature selection produced an accuracy of 89.3%. After feature selection, the model's performance improved significantly. The Chi-Square method produced 8 selected features with the highest accuracy of 97.3%, followed by Information Gain with 5 features and an accuracy of 97.3%, and Pearson Correlation with 3 features and an accuracy of 88.0%. The results of this study demonstrate that selecting appropriate features can significantly enhance an ANN's ability to categorize sleep problems. The proposed approach is expected to be a reference in the development of a more accurate sleep disorder diagnostic aid system.

Downloads

Download data is not yet available.

References

[1] A. Rahman et al., “Improving Sleep Disorder Diagnosis Through Optimized Machine Learning Approaches,” IEEE Access, vol. 13, no. January, pp. 20989–21004, 2025, doi: 10.1109/ACCESS.2025.3535535.

[2] D. Fitriyani, M. Amelia, and S. S. Yuliana, “Penerapan Algoritma Random Forest Untuk Klasifikasi Gangguan Tidur Berdasarkan Pola Kehidupan Sehari-hari,” vol. 1, no. 1, pp. 21–26, 2025.

[3] Ramadani, A. R. N., “Pengaruh kualitas tidur terhadap hasil belajar mata kuliah statistika mahasiswa semester 1 Pendidikan Fisika Universitas Negeri Yogyakarta,” Jurnal Sosial Humaniora dan Pendidikan, vol. 2, no. 3, pp. 106–112, 2023.

[4] W. Widyastuty and M. A. Azis, “Classification and Evaluation of Sleep Disorders Using Random Forest Algorithm in Health and Lifestyle Dataset,” vol. 13, no. 1, pp. 11–18, 2024, doi: 10.28989/compiler.v13i1.2184.

[5] M. Ali Ridla and A. A. Bawazin, “Klasifikasi Gangguan Tidur Berdasarkan Gaya Hidup Menggunakan Algoritma Decision Tree,” vol. 3, no. 2, pp. 119–126, 2025, doi: 10.26905/jisad.v3i2.16038.

[6] M. F. Alie and R. Rahmanda, “Model Prediksi Gangguan Tidur berdasarkan Beberapa Faktor menggunakan Machine Learning Prediction of Sleep Disorders based on Several Factors using Machine Learning,” vol. 9, no. 2, pp. 504–517, 2024.

[7] T. S. Alshammari, “Applying Machine Learning Algorithms for the Classification of Sleep Disorders,” IEEE Access, vol. 12, no. January, pp. 36110–36121, 2024, doi: 10.1109/ACCESS.2024.3374408.

[8] D. Sari, “Prediksi Gangguan Tidur pada Sleep Health and Lifestyle Menggunakan Support Vector Machine dan Neural Network,” pp. 36–42, 2024.

[9] M. Maulidah and N. Hidayati, “Prediksi Kesehatan Tidur Dan Gaya Hidup Menggunakan Machine Learning,” vol. 4, no. 1, pp. 81–86, 2024, doi: 10.31294/conten.v4i1.4918.

[10] N. Khansa and Z. Fatah, “Analisis Perbandingan Algoritma Machine Learning Dalam Klasifikasi Gangguan Tidur,” Gudang J. Multidisiplin Ilmu, vol. 2, no. November, pp. 76–81, 2024.

[11] M. Albarka, Z. Chen, K. Shuaib, and Y. Liu, “Effects of feature selection and normalization on network intrusion detection,” Data Sci. Manag., vol. 8, no. 1, pp. 23–39, 2025, doi: 10.1016/j.dsm.2024.08.001.

[12] M. J. Jiménez-Navarro, M. Martínez-Ballesteros, I. S. Brito, F. Martínez-Álvarez, And G. Asencio-Cortés, “Embedded feature selection for neural networks via learnable drop layer,” vol. 00, no. 00, 2024.

[13] J. Pardede and R. Dwianto, “The Effect of Feature Selection on Machine Learning Classification,” vol. 9, no. July, 2025.

[14] H. O. Braganc, “Feature Selection for Stock Market Prediction : A Comparison of Relief and Information Gain Methods,” vol. 1, no. Iceis, pp. 996–1003, 2025, doi: 10.5220/0013481300003929.

[15] K. Mei, M. Tan, Z. Yang, and S. Shi, “Modeling of Feature Selection Based on Random Forest Algorithm and Pearson Correlation Coefficient Modeling of Feature Selection Based on Random Forest Algorithm and Pearson Correlation Coefficient,” J. Phys. Conf. Ser., pp. 1–9, 2022, doi: 10.1088/1742-6596/2219/1/012046.

[16] I. Putu, A. Jayadinanta, I. K. Gede, A. Agung, I. Ngurah, and E. Karyawati, “Klasifikasi Penyakit Jantung Dengan Neural Network dan Seleksi Fitur Chi-Square,” vol. 13, no. 3, pp. 535–542, 2025.

[17] C. Yücelbaş, “A new approach: information gain algorithm‑based k‑nearest neighbors hybrid diagnostic system for Parkinson’s disease,” Phys. Eng. Sci. Med., no. 0123456789, 2022, doi: 10.1007/s13246-021-01001-6.

[18] C. Apriza Yanti and I. Julian Akhri, “Perbedaan uji korelasi pearson, spearman dan kendall tau dalam menganalisis kejadian diare,” vol. 6, no. 1, pp. 51–58, 2022.

[19] K. V. Yadav and D. R. Kumar, “Data Preprocessing Techniques,” vol. 1, pp. 1–6, 2022.

[20] I. A. Hidayat, “Classification of Sleep Disorders Using Random Forest on Sleep Health and Lifestyle Dataset,” J. Dinda Data, vol. 3, no. 2, pp. 71–76, 2023.

[21] Alshammari, T. S., “Applying machine learning algorithms for the classification of sleep disorders,” IEEE Access, vol. 12, pp. 36110–36121, 2024.

[22] N. Pudjihartono, T. Fadason, A. W. Kempa-Liehr, and J. M. O’Sullivan, “A Review of Feature Selection Methods for Machine Learning-Based Disease Risk Prediction,” Front. Bioinforma., vol. 2, no. June, pp. 1–17, 2022, doi: 10.3389/fbinf.2022.927312.

[23] M. Aslam and F. Smarandache, “Chi-square test for imprecise data in consistency table,” Front. Appl. Math. Stat., vol. 9, 2023, doi: 10.3389/fams.2023.1279638.

[24] R. M. Kronberg, D. Meskelevicius, M. Sabel, M. Kollmann, C. Rubbert, and I. Fischer, “Optimal acquisition sequence for AI-assisted brain tumor segmentation under the constraint of largest information gain per additional MRI sequence,” Smart Agric. Technol., vol. 2, no. 4, p. 100053, 2022, doi: 10.1016/j.neuri.2022.100053.

[25] M. Ilmiyah, M. A. Barata, and P. E. Yuwita, “Implementation of ANN Optimization with SMOTE and Backward Elimination for PCOS Prediction,” vol. 12, no. 1, pp. 133–144, 2025, doi: 10.15294/sji.v12i1.22886.

[26] Z. Zhang et al., “Evaluation methods of inhibition to microorganisms in biotreatment processes: A review,” Water Cycle, vol. 4, no. December 2022, pp. 70–78, 2023, doi: 10.1016/j.watcyc.2023.02.004.

Downloads

Published

2026-04-26

Issue

Section

Articles

How to Cite

[1]
D. N. Kharisma, I. W. D. Prastya, and I. A. Saida, “Optimization of Sleep Disorder Classification Using ANN with Multi-Method Feature Selection”, journalisi, vol. 8, no. 2, pp. 2363–2381, Apr. 2026, doi: 10.63158/journalisi.v8i2.1473.