用强化学习自动选最优模型和特征,提升3D打印翘曲预测精度。
Automated Data Engineering and Feature Selection for the Case Study of Warpage Detection in Fused Deposition Modeling

- 基于SHAP的可解释性分析筛选特征子集,结合强化学习动态优化模型选择。
- 测试集AUC从0.9248提升至0.9731,平均奖励增长超50%。
- 适合需要自动化数据处理与特征优化的工业制造场景。
本研究提出一种自动化数据处理(ADP)框架,用于评估并优化熔融沉积成型(FDM)过程中机器学习模型与特征组合的性能。该框架采用受强化学习启发的策略更新机制,在217个数据集上对完整特征集和通过SHAP可解释人工智能(SHAP XAI)选出的特征子集训练多个模型。每轮迭代中,框架评估各模型-特征组合的预测准确率与F1分数,计算标量奖励并更新$Q$值以指导后续选择。利用SHAP XAI生成降维但信息丰富的特征子集,使框架能探索不同维度下的性能表现。实验显示策略随多轮迭代逐步演化,奖励分布反映性能稳定性。结果表明,借助该框架结合XAI算法,可有效收敛至最优模型-特征配置,显著提升预测准确率与稳定性:测试集AUC由0.9248提升至0.9731,平均奖励值较基线全特征配置提升超过50%。
原文摘要 · Abstract (English)
This study contributes toward development of an Automated Data Processing (ADP) framework designed to evaluate and reinforce optimal machine learning model-feature combinations for predictive tasks in fused deposition modeling (FDM) process datasets. The methodology is centered around a reinforcement learning-inspired policy updating mechanism, where multiple machine learning models are trained on both full feature sets and feature subsets selected through Shapley-based Explainable AI (SHAP XAI) across 217 datasets. At each episode, the framework assesses the predictive accuracy and F1-scores of each model-feature pair, computes a scalar reward, and updates $Q$ values to guide future model selection. SHAP XAI feature importance was employed to generate reduced yet informative feature subsets to enable the framework to explore performance with dimensionality. The policy was shown to evolve over multiple episodes, with reward distributions used to visualize performance stability. Overall, results indicate that leveraging the ADP framework through XAI algorithms successfully converges toward optimal model-feature configurations with improved accuracy and stability. Specifically, the proposed framework improves the test-set AUC from 0.9248 to 0.9731 and increases the mean reward value by more than fifty percent compared with the baseline full-feature configuration.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。