A Study on a Random Forest-Based Algorithm for Predicting Yield Rates in Process-Oriented Production Lines
DOI:
https://doi.org/10.54097/hwwtzy41Keywords:
Yield Rate Prediction, Random Forest, Process-based Cascaded Architecture, Highly Imbalanced Data, Hybrid ResamplingAbstract
Industrial production line data poses three intertwined difficulties: the features are high-dimensional and anonymized, the process is strongly sequential, and the class distribution is extremely skewed (171:1). This paper presents a Process-based Cascaded Random Forest (P-RF) algorithm tailored to these conditions. Instead of flattening the feature space, P-RF maps the physical order of workstations onto a hierarchy of forest modules and lets a suspicion probability flow forward along the line, so that cumulative error effects are captured rather than ignored. To cope with the severe imbalance, SMOTE resampling, cost-sensitive learning, and grid search are combined into a single optimization scheme. Experiments show that the final model reaches an overall accuracy of 91.21% with a macro-average F1 score of 0.4839, while cutting false alarms to 29,361—a trade-off that favors low disruption on high-volume lines. A global feature-importance traceback further confirms that the transmission probabilities from preceding workstations dominate model decisions, revealing the “error snowball effect” at work. This not only supports the soundness of the P-RF architecture, but also gives production managers a concrete handle for quality control and precise traceability.
Downloads
References
[1] Chang, Y. Q., Sun, X. T., Zhong, L. S., & et al. (2021). Evaluation of industrial process operating conditions based on an improved random forest algorithm. Acta Automatica Sinica, 47(9), 2214 2225. https://doi.org/10.16383/j.aas.c200341.
[2] Adetunji, A. B., Akande, O. N., Ajala, F. A., & et al. (2022). House price prediction using random forest machine learning technique. Procedia Computer Science, 199, 806 813. https:// doi. org/ 10.1016/j.procs.2022.01.101.
[3] Salman, H. A., Kalakeh, A., & Steiti, A. (2024). Random forest algorithm overview. Babylonian Journal of Machine Learning, 2024, 69 79.
[4] Zhao, X. H., Liu, L., Pu, J. P., & et al. (2026). Prediction of vibration in the lower housing of a coal mill based on modal analysis and PCA WOA RF. Journal of Shandong University (Engineering Edition), 56(1), 150 157.
Downloads
Published
Issue
Section
License
Copyright (c) 2026 Frontiers in Computing and Intelligent Systems

This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.

