Supervised threshold-based heart sound classification algorithm

Deep classification networks have been one of the predominant methods for classifying heart sound recordings. To satisfy their demand for sample size, the most commonly used method for data augmentation is that which divides each heart sound instance into a number of segments, with each segment labelled as the same category as its origin and used as a new sample for training or forecasting. However, performing this poses a crucial issue as to how to determine the category of a predicted heart sound instance from its segments' prediction results.

To solve this issue, this paper establishes a mathematical formula to connect the classification performance of these heart sound instances with the prediction results of their segments via a threshold which is supervised by the training set. The optimal value of the proposed threshold is calculated by maximizing the prediction accuracy of the training instances. Seeking the optimal threshold by a gradient-based method, we prove that a continuous function can closely approximate a part of the function of accuracy which transforms the discrete function of accuracy into a continuous function. The optimal threshold is used to recognize the undetermined heart sound recording.

Experimental results show the classification performance from a 10-fold cross-validation, measured by the commonly used scales of sensitivity, specificity and mean accuracy (MAcc). The proposed algorithm improves the MAcc by about 4% by modifying the baseline. In addition, the MAcc surpasses the champion of the PhysioNet/Computing in Cardiology Challenge 2016.

Our study develops a methodology to determine the category of a predicted heart sound instance from its segments' prediction results, thus assisting in the data augmentation exercise which is necessary to provide sufficient data for deep classification networks. Our method significantly improves the classification performance.