Unknown

Dataset Information

0

Self-Supervised Learning Improves Accuracy and Data Efficiency for IMU-Based Ground Reaction Force Estimation.


ABSTRACT:

Objective

Recent deep learning techniques hold promise to enable IMU-driven kinetic assessment; however, they require large extents of ground reaction force (GRF) data to serve as labels for supervised model training. We thus propose using existing self-supervised learning (SSL) techniques to leverage large IMU datasets to pre-train deep learning models, which can improve the accuracy and data efficiency of IMU-based GRF estimation.

Methods

We performed SSL by masking a random portion of the input IMU data and training a transformer model to reconstruct the masked portion. We systematically compared a series of masking ratios across three pre-training datasets that included real IMU data, synthetic IMU data, or a combination of the two. Finally, we built models that used pre-training and labeled data to estimate GRF during three prediction tasks: overground walking, treadmill walking, and drop landing.

Results

When using the same amount of labeled data, SSL pre-training significantly improved the accuracy of 3-axis GRF estimation during walking compared to baseline models trained by conventional supervised learning. Fine-tuning SSL model with 1-10% of walking data yielded comparable accuracy to training baseline model with 100% of walking data. The optimal masking ratio for SSL is 6.25-12.5%.

Conclusion

SSL leveraged large real and synthetic IMU datasets to increase the accuracy and data efficiency of deep-learning-based GRF estimation, reducing the need for labeled data.

Significance

This work, with its open-source code and models, may unlock broader use cases of IMU-driven kinetic assessment by mitigating the scarcity of GRF measurements in practical applications.

SUBMITTER: Tan T 

PROVIDER: S-EPMC10849467 | biostudies-literature | 2024 Jan

REPOSITORIES: biostudies-literature

altmetric image

Publications

Self-Supervised Learning Improves Accuracy and Data Efficiency for IMU-Based Ground Reaction Force Estimation.

Tan Tian T   Shull Peter B PB   Hicks Jenifer L JL   Uhlrich Scott D SD   Chaudhari Akshay S AS  

bioRxiv : the preprint server for biology 20240125


<h4>Objective</h4>Recent deep learning techniques hold promise to enable IMU-driven kinetic assessment; however, they require large extents of ground reaction force (GRF) data to serve as labels for supervised model training. We thus propose using existing self-supervised learning (SSL) techniques to leverage large IMU datasets to pre-train deep learning models, which can improve the accuracy and data efficiency of IMU-based GRF estimation.<h4>Methods</h4>We performed SSL by masking a random por  ...[more]

Similar Datasets

| S-EPMC7038404 | biostudies-literature
| S-EPMC10773877 | biostudies-literature
| S-EPMC10099259 | biostudies-literature
| S-EPMC10044327 | biostudies-literature
| S-EPMC5371282 | biostudies-literature
| S-EPMC10466106 | biostudies-literature
| S-EPMC11493597 | biostudies-literature
| S-EPMC9792450 | biostudies-literature
| S-EPMC6335151 | biostudies-literature
| S-EPMC10774911 | biostudies-literature