A Comparative Study of Construction Site Accident Risk Screening Models Using Public Construction Project Data - Focusing on Tabular Machine Learning and Attribute-Based Heterogeneous Graph Models -
This study compares tabular machine learning models and attribute-based heterogeneous graph models for construction site accident risk screening using 214,293 public construction project records with a 0.91% accident rate. Six models were compared under five repeated stratified splits and a common hyperparameter search, with PR-AUC as the primary metric. XGBoost achieved the highest PR-AUC of 0.268 ± 0.022, ranked first in all five splits, and captured 35.3% of accidents within the top 1% of sites. The graph-based models exceeded the accident-rate baseline but did not outperform the tabular models, indicating limited benefits without direct relational identifiers. Under missing-signal control, the estimated dependence on missingness varied from −2.1% to −23.2% across splits, indicating that single-split estimates are unreliable.
Authors
- Sangil Yoon
- Seokhyeon Moon
- JooSung Lee
- SangHwan Kim
- YongHan Ahn
Publication Details
- Journal
- Korean Journal of Computational Design and Engineering
- Published
- 2026-09-10
- DOI
- https://doi.org/10.7315/cde.2026.197
- Primary Topic
- Occupational Health and Safety Research
- Type
- article
- Field-Weighted Citation Impact
- 0.00