Amputee Fall Risk Classification Using Machine Learning and Smartphone Sensor Data from 2-Minute and 6-Minute Walk Tests

Abstract

The 6-min walk test (6MWT) is commonly used to assess a person’s physical mobility and aerobic capacity. However, richer knowledge can be extracted from movement assessments using artificial intelligence (AI) models, such as fall risk status. The 2-min walk test (2MWT) is an alternate assessment for people with reduced mobility who cannot complete the full 6MWT, including some people with lower limb amputations; therefore, this research investigated automated foot strike (FS) detection and fall risk classification using data from a 2MWT. A long short-term memory (LSTM) model was used for automated foot strike detection using retrospective data (n = 80) collected with the Ottawa Hospital Rehabilitation Centre (TOHRC) Walk Test app during a 6-min walk test (6MWT). To identify FS, an LSTM was trained on the entire six minutes of data, then re-trained on the first two minutes of data. The validation set for both models was ground truth FS labels from the first two minutes of data. FS identification with the 6-min model had 99.2% accuracy, 91.7% sensitivity, 99.4% specificity, and 82.7% precision. The 2-min model achieved 98.0% accuracy, 65.0% sensitivity, 99.1% specificity, and 68.6% precision. To classify fall risk, a random forest model was trained on step-based features calculated using manually labeled FS and automated FS identified from the first two minutes of data. Automated FS from the first two minutes of data correctly classified fall risk for 61 of 80 (76.3%) participants; however, <50% of participants who fell within the past six months were correctly classified. This research evaluated a novel method for automated foot strike identification in lower limb amputee populations that can be applied to both 6MWT and 2MWT data to calculate stride parameters. Features calculated using automated FS from two minutes of data could not sufficiently classify fall risk in lower limb amputees.

Keywords: 2MWT; 6MWT; LSTM; amputee; artificial intelligence; fall risk classification; foot strike detection; random forest; smartphone.

Figure 1

Experimental set-up: smartphone on posterior pelvis.

Figure 2

Filtered smartphone signals over time. Medio-lateral acceleration (yellow curve), vertical acceleration (red curve), and anterior–posterior (AP) acceleration (green curve) were used to identify ground truth foot strikes. Typically, foot strikes correspond with an AP acceleration peak followed by a vertical acceleration peak. Video recording of the trial was used to confirm the timestamp of foot strikes. Vertical blue lines indicate frames manually identified as ground truth labels.