Agricultural injuries remain a significant occupational hazard, causing substantial human and economic losses worldwide. This study investigates the prediction of agricultural injury severity using both linear and ensemble machine learning (ML) models and applies explainable AI (XAI) techniques to understand the contribution of input features. Data from AgInjuryNews (from 2015 to 2024) was preprocessed to extract relevant attributes such as location, time, age, and safety measures. The dataset comprised 2,421 incidents categorized as fatal or non-fatal. Various ML models, including Naive Bayes (NB), Decision Tree (DT), Support Vector Machine (SVM), Random Forest (RF), and Gradient Boosting (GB), were trained and evaluated using standard performance metrics. Ensemble models demonstrated superior accuracy and recall compared to linear models, with XGBoost achieving a recall of 100% for fatal injuries. However, all models faced challenges in predicting nonfatal injuries due to class imbalance. SHAP analysis provided insights into feature importance, with age, gender, location, and time emerging as the most influential predictors across models. This research highlights the effectiveness of ensemble ML models in injury prediction while emphasizing the need for balanced datasets and XAI techniques for actionable insights. The findings have practical implications for enhancing agricultural safety and guiding policy interventions.
Competing Interest StatementThe authors have declared no competing interest.
Funding StatementThis study did not receive any funding
Author DeclarationsI confirm all relevant ethical guidelines have been followed, and any necessary IRB and/or ethics committee approvals have been obtained.
Yes
I confirm that all necessary patient/participant consent has been obtained and the appropriate institutional forms have been archived, and that any patient/participant/sample identifiers included were not known to anyone (e.g., hospital staff, patients or participants themselves) outside the research group so cannot be used to identify individuals.
Yes
I understand that all clinical trials and any other prospective interventional studies must be registered with an ICMJE-approved registry, such as ClinicalTrials.gov. I confirm that any such study reported in the manuscript has been registered and the trial registration ID is provided (note: if posting a prospective study registered retrospectively, please provide a statement in the trial ID field explaining why the study was not registered in advance).
Yes
I have followed all appropriate research reporting guidelines, such as any relevant EQUATOR Network research reporting checklist(s) and other pertinent material, if applicable.
Yes
Data AvailabilityAll data produced in the present study are available upon reasonable request to the authors
留言 (0)