Evaluating the use of machine learning in the assessment of joint angle using a single inertial sensor

Abstract

INTRODUCTION: Joint angle measurement is an important objective marker in rehabilitation. Inertial measurement units may provide an accurate and reliable method of joint angle assessment. The objective of this study was to assess whether a single sensor with the application of machine learning algorithms could accurately measure hip and knee joint angle, and investigate the effect of inertial measurement unit orientation algorithms and person-specific variables on accuracy. METHODS: Fourteen healthy participants completed eight rehabilitation exercises with kinematic data captured by a 3D motion capture system, used as the reference standard, and a wearable inertial measurement unit. Joint angle was calculated from the single inertial measurement unit using four machine learning models, and was compared to the reference standard to evaluate accuracy. RESULTS: Average root-mean-squared error for the best performing algorithms across all exercises was 4.81° (SD = 1.89). The use of an inertial measurement unit orientation algorithm as a pre-processing step improved accuracy; however, the addition of person-specific variables increased error with average RMSE 4.99° (SD = 1.83°). CONCLUSIONS: Hip and knee joint angle can be measured with a good degree of accuracy from a single inertial measurement unit using machine learning. This offers the ability to monitor and record dynamic joint angle with a single sensor outside of the clinic.