Adversarial training with GatedTabTransformer and the GAN to predict re-employment factors of Korean male workers after an industrial accident
Year: 2025
Type: article
Abstract: Background: The re-employment of male workers after an industrial accident is a critical problem with substantial socioeconomic implications. This study proposes an advanced predictive model integrating the GatedTabTransformer with the generative adversarial network using adversarial training to improve the accuracy and robustness of re-employment predictions. Methods: We compared the performance of the proposed model against traditional machine learning techniques, including logistic regression, k-nearest neighbors, support vector machine, linear discriminant analysis, random forest, bagging, adaptive boosting and extreme gradient boosting on a dataset of 1383 male workers after an industrial accident. Results: The proposed model outperforms these traditional methods across the performance metrics, achieving an accuracy of 89.2%and an area under the receiver operating characteristic curve of 0.924. Furthermore, the analysis identified previous employment duration, age, injury severity, education level, and industry type as the most significant factors influencing re-employment. Conclusions: These findings underscore the potential of advanced machine learning techniques in addressing complex real-world problems and provide actionable insight for policymakers and practitioners focused on improving re-employment outcomes for male workers after an industrial accident.
Source: Journal of Men s Health
Author: Haewon Byeon
Institution: Korea University of Technology and Education
Language: English
Subfield: Artificial Intelligence
Field: Computer Science
Domain: Physical Sciences
Open Access status: gold
Estimated APC paid: $1,500
OpenAlex ID: W4412763379