An Efficient Convolutional Neural Network for Coronary Heart Disease Prediction

by   Aniruddha Dutta, et al.

This study proposes an efficient neural network with convolutional layers to classify significantly class-imbalanced clinical data. The data are curated from the National Health and Nutritional Examination Survey (NHANES) with the goal of predicting the occurrence of Coronary Heart Disease (CHD). While the majority of the existing machine learning models that have been used on this class of data are vulnerable to class imbalance even after the adjustment of class-specific weights, our simple two-layer CNN exhibits resilience to the imbalance with fair harmony in class-specific performance. In order to obtain significant improvement in classification accuracy under supervised learning settings, it is a common practice to train a neural network architecture with a massive data and thereafter, test the resulting network on a comparatively smaller amount of data. However, given a highly imbalanced dataset, it is often challenging to achieve a high class 1 (true CHD prediction rate) accuracy as the testing data size increases. We adopt a two-step approach: first, we employ least absolute shrinkage and selection operator (LASSO) based feature weight assessment followed by majority-voting based identification of important features. Next, the important features are homogenized by using a fully connected layer, a crucial step before passing the output of the layer to successive convolutional stages. We also propose a training routine per epoch, akin to a simulated annealing process, to boost the classification accuracy. Despite a 35:1 (Non-CHD:CHD) ratio in the NHANES dataset, the investigation confirms that our proposed CNN architecture has the classification power of 77 to correctly classify the presence of CHD and 81.8 a testing data, which is 85.70 characters)Please check the paper for full abstract)



There are no comments yet.


page 9

page 13


Fair-Net: A Network Architecture For Reducing Performance Disparity Between Identifiable Sub-Populations

In real world datasets, particular groups are under-represented, much ra...

Credit Card Fraud Detection Using Autoencoder Neural Network

Imbalanced data classification problem has always been a popular topic i...

Deep Over-sampling Framework for Classifying Imbalanced Data

Class imbalance is a challenging issue in practical classification probl...

On the dissection of degenerate cosmologies with machine learning

Based on the DUSTGRAIN-pathfinder suite of simulations, we investigate o...

Class Imbalance Techniques for High Energy Physics

A common problem in high energy physics is extracting a signal from a mu...

A Convolutional Neural Network Approach to the Classification of Engineering Models

This paper presents a deep learning approach for the classification of E...

DeepNAT: Deep Convolutional Neural Network for Segmenting Neuroanatomy

We introduce DeepNAT, a 3D Deep convolutional neural network for the aut...
This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.