Machine Learning Based Model for Predicting Head Injury Criterion (HIC)
Published March 31, 2020 by The Stapp Association in United States
This content contains downloadable datasetsAnnotation ability available
The objective of this study is to develop a machine learning based predictive model from the available crash test data and use it for predicting injury metrics. In this study, a model was developed for predicting the head injury criterion, HIC15, using pre-test features (vehicle, test, occupant and restraint related). This problem was solved as a classification task, in which HIC15 with a threshold of 700 was divided into three classes i.e. low, medium and high. Crash test data was collected from the NHTSA database and was split into training and test datasets. Predictive models were developed from the training dataset using cross-validation while the test dataset was only used at the final step to evaluate the chosen predictive model. A logistic regression based predictive model was chosen as it demonstrated minimal overfitting and gave the highest F1 score (0.81) on the validation dataset. This chosen model gave a F1 score of 0.82 on the test (new/unseen) dataset.
CitationHasija, V. and Takhounts, E., "Machine Learning Based Model for Predicting Head Injury Criterion (HIC)," SAE Technical Paper 2019-22-0016, 2020, https://doi.org/10.4271/2019-22-0016.
Data Sets - Support Documents
|[Unnamed Dataset 1]|
|[Unnamed Dataset 2]|
|[Unnamed Dataset 3]|
|[Unnamed Dataset 4]|
- Albon, C. (2018a) Machine Learning with Python Cookbook, pp. 81. O’Reilly Media, Inc.
- Albon, C. (2018b) Machine Learning with Python Cookbook, pp. 184. O’Reilly Media, Inc.
- Breiman L, Friedman J, Olshen R, and Stone C. (1984) Classification and Regression Trees, Wadsworth, Belmont, CA.
- Breiman L. (2001) Random Forests, Machine Learning, 45(1), 5-32.
- Brownlee, J. (2018a) Machine Learning Mastery with Python. pp. 53. Machine Learing Mastery.
- Brownlee J. (2018b) Master Machine Learning Algorithms, Pages 52-61, Machine Learning Mastery.
- Brownlee J. (2018c) Master Machine Learning Algorithms, Pages 116-125, Machine Learning Mastery.
- Chen T. , Guestrin, C. (2016) XGBoost: A Scalable Tree Boosting System. https://arxiv.org/abs/1603.02754
- Pedregosa et al. (2011) Scikit-learn: Machine Learning in Python, pp. 2825-2830, JMLR 12.
- Scikit-learn User Guide (2019).