Automated assessment of balance: A neural network approach based on large-scale balance function data

Abstract

Balance impairment (BI) is an important cause of falls in the elderly. However, the existing balance estimation system needs to measure a large number of items to obtain the balance score and balance level, which is less efficient and redundant. In this context, we aim at building a model to automatically predict the balance ability, so that the early screening of large-scale physical examination data can be carried out quickly and accurately. We collected and sorted out 17,541 samples, each with 61-dimensional features and two labels. Moreover, using this data a lightweight artificial neural network model was trained to accurately predict the balance score and balance level. On the premise of ensuring high prediction accuracy, we reduced the input feature dimension of the model from 61 to 13 dimensions through the recursive feature elimination (RFE) algorithm, which makes the evaluation process more streamlined with fewer measurement items. The proposed balance prediction method was evaluated on the test set, in which the determination coefficient (R2) of balance score reaches 92.2%. In the classification task of balance level, the metrics of accuracy, area under the curve (AUC), and F1 score reached 90.5, 97.0, and 90.6%, respectively. Compared with other competitive machine learning models, our method performed best in predicting balance capabilities, which is especially suitable for large-scale physical examination.