Tool wear prediction based on multi-domain feature fusion and TabPFN-SimAM integration
摘要
In the process of machining, the condition of tool wear directly determines both the quality of the workpiece and the production efficiency. To address the limitations of existing prediction methods, this study proposes a novel collaborative optimization approach integrating feature fusion and predictive modeling, namely a tool wear prediction method based on multi-domain feature fusion and the TabPFN–SimAM attention mechanism. Partial Least Squares (PLS) is employed to perform supervised fusion of multi-sensor features, thereby constructing a highly discriminative feature set. Subsequently, the SimAM attention mechanism is introduced to adaptively reweight the fused features, enhancing the representation of critical information. The Tabular Prior-data Fitted Network (TabPFN) serves as the core predictive model, and an Enhanced Zebra Optimization Algorithm (EZOA) is employed to perform combinational search for optimal solutions of the key parameters. Experimental results based on the PHM2010 dataset demonstrate that the proposed method achieves high prediction accuracy in tool wear estimation, with an average R² exceeding 0.99. Its performance significantly outperforms single-domain feature methods as well as several mainstream benchmark models, including SVR, XGBoost, BiLSTM, Transformer, and TabPFN. These results verify the comprehensive superiority of the proposed approach in terms of prediction accuracy and robustness.