SENNERF: Stacking Ensemble with Neural Networks and Enhanced Random Forest for improved heart disease prediction
摘要
Heart disease continues to be a prominent reason for death throughout the world, which highlights the importance of effective predictive models for better early diagnosis. Due to their limited scalability and inability to capture complex relationships, existing machine learning methods often need more accuracy when applied to larger datasets. SENNERF, a unique ensemble framework, integrates DenseNet topologies and an upgraded random forest model to improve heart disease prediction. This research proposes a scalable, accurate, and generalizable model for huge datasets. We perform feature extraction and enhanced random forest prediction after using DenseNet128, DenseNet256, and CompactNet64 architectures. SENNERF performed well on a large heart illness dataset, with 99.97% accuracy, 99% precision and recall, and 99.71% F1-score. This outperforms previous research that struggled with smaller datasets and had lower accuracy. SENNERF’s capacity to handle large datasets with high accuracy and cheap processing costs makes it a unique heart disease solution. Additionally, the proposed model was evaluated with the benchmark heart disease dataset. The proposed SENNERF model attained 100% accuracy for classifying the highly imbalanced small dataset. Click here to view