3.1. Data and Preprocessing
This study conducts experiments using a publicly available high-frequency multivariate sensor dataset [
28]. The original dataset contains 62,631 sensor records with a sampling frequency of 1 Hz, and each record is associated with a UTC timestamp. The dataset contains 13 sensor variables: Temperature, Humidity, TVOC, eCO2, Raw H2, Raw Ethanol, Pressure, PM1.0, NC0.5, NC1.0, NC2.5, PM2.5, and Fire Alarm. These variables collectively characterize the dynamic features of environmental states, gas variations, and particulate matter concentrations in high-frequency particulate matter monitoring scenarios. It should be clarified that the dataset used in this study is a publicly available smoke-sensing dataset, rather than data collected from an actual powder and granular material processing workshop. Therefore, this dataset is used as a proxy benchmark for evaluating the proposed model in high-frequency multivariate PM2.5 forecasting. The results mainly demonstrate the methodological effectiveness of the proposed forecasting framework on public sensor data, while its direct generalization to real industrial dust monitoring scenarios still requires further validation using field data.
It should be noted that, in terms of temporal structure, this dataset is not a single continuous sequence, but is composed of five mutually separated continuous sampling segments. The continuous segments were identified according to timestamp continuity, and a temporal gap larger than 1 s was regarded as a discontinuity. There are obvious temporal gaps among different segments, and the PM2.5 concentration levels and fluctuation ranges vary considerably across segments. Considering that multi-step-ahead time series forecasting imposes relatively high requirements on temporal continuity and distribution consistency, this study does not directly merge all segments for modeling. Instead, two continuous segments with the longest lengths were selected as the research objects to ensure sufficient samples for sliding-window forecasting, and 49,989 valid samples are ultimately retained. For the interval region between the two segments, the corresponding samples are explicitly skipped during the sample generation stage to avoid constructing sliding windows across segment boundaries. This ensures that both the input sequences and the forecasting targets are derived from the same continuous observation segment, thereby improving temporal consistency and modeling reliability.
On this basis, the retained records are divided into the training set, validation set, and test set according to a ratio of 7:1:2 in strict chronological order, and no random shuffling is performed before data splitting. To avoid information leakage caused by overlapping sliding windows near subset boundaries, a sample is assigned to a subset only when both its historical input window and future prediction window are completely located within that subset. Samples crossing the boundaries between the training, validation, and test sets are discarded. The training set is used for model parameter learning, the validation set is used for hyperparameter tuning and early stopping control, and the test set is used for final performance evaluation. In this study, the forecasting task is formulated as a multivariate time series forecasting problem. Although the model jointly predicts all 12 sensor variables, PM2.5 is selected as the primary evaluation target because it is directly related to particulate matter concentration and dust-risk warning. Considering that the Fire Alarm variable in the original dataset is mainly used for classification, it is removed from the input features. The remaining 12 variables, namely Temperature, Humidity, TVOC, eCO2, Raw H2, Raw Ethanol, Pressure, PM1.0, NC0.5, NC1.0, NC2.5, and PM2.5, are retained as model inputs, with the historical observations of PM2.5 also included in the input sequence. The normalization parameters are fitted only on the training set and then applied to the validation and test sets, so that information from the validation and test periods is not used during data scaling. The experimental setting uses an input length of 300 steps and a forecasting horizon of 60 steps, that is, the observed sequence over the past 300 s is used to forecast the PM2.5 concentration variation over the next 60 s. It should also be noted that the experimental results in this study mainly reflect the forecasting performance of the model under continuous high-frequency monitoring scenarios with relatively consistent distributions. The generalization capability under extreme high-concentration segments and more complex cross-distribution conditions still requires further investigation.
3.2. Experimental Settings
All experiments were implemented using the PyTorch deep learning framework (version 2.0.0, CUDA 11.8) and conducted on a platform equipped with an Intel Xeon Silver 4114 CPU and an NVIDIA GeForce RTX 3090 GPU. During training, the Adam optimizer was adopted for parameter updating, and the configurations of the remaining hyperparameters are listed in
Table 3. To ensure a fair comparison, all compared models were evaluated using the same chronological data split, preprocessing procedure, input length, forecasting horizon, and evaluation metrics.
Based on the search ranges listed in
Table 3, the hyperparameter configuration of each model was selected according to the validation set performance. For models involving different output strategies, both Type 1 and Type 2 were evaluated on the validation set, and the better performing strategy was fixed for the final test evaluation. The selected learning rate, model dimension, dropout rate, and output strategy of each compared model are summarized in
Table 4.
After the selected hyperparameters and output strategy were fixed, each model was independently trained and tested using three random seeds, namely 2021, 2022, and 2023. The final results are reported as mean ± standard deviation over the three runs.
3.3. Evaluation Metrics
To comprehensively evaluate the forecasting performance of the proposed model, two groups of metrics are adopted in this study. The first group is used to measure the overall prediction accuracy of PM2.5 concentration, including MSE, MAE, RMSE, MAPE and R
2. MSE and RMSE reflect the magnitude of forecasting errors and are sensitive to large deviations, MAE represents the average absolute deviation between the predicted values and the true values, offering good interpretability. MAPE characterizes the relative forecasting error, and R
2 is used to measure the degree to which the model fits the variation trend of the target variable, and a value closer to 1 indicates better fitting performance. These metrics are defined as follows:
where MSE denotes Mean Squared Error, MAE denotes Mean Absolute Error, RMSE denotes Root Mean Squared Error, MAPE denotes Mean Absolute Percentage Error and R
2 denotes the coefficient of determination.
denotes the true value,
denotes the predicted value,
denotes the number of samples, and
denotes the mean of the true values.
is a small constant used to avoid division by zero in the calculation of MAPE.
In addition to the overall prediction metrics, Warning-oriented metrics are introduced to evaluate model performance during high-concentration periods. Previous studies on extreme value forecasting have shown that average regression errors alone are insufficient to characterize the prediction performance for extreme high or low values [
29]. The warning threshold θ is defined as the 90th percentile of PM2.5 values in the training set, so that the threshold is determined without using validation or test information. Five metrics are used for warning evaluation, namely High MAE, Peak Shift Error, Peak Underestimation Rate, False Alarm Rate, and Missed Alarm Rate. For clarity, they are denoted as High MAE, PSE, PUR, FAR, and MAR, respectively. Their definitions are summarized in
Table 5. Since the sampling frequency is 1 Hz, one time step in PSE corresponds to one second.
3.4. Model Performance Evaluation
To verify the effectiveness of the proposed model in the high-frequency dust concentration forecasting task, this study selects DLinear, Transformer, Informer, Reformer, Flowformer, iTransformer, iInformer, iReformer, and iFlowformer as baseline models. All models are trained and tested under identical conditions in terms of the data splitting strategy, input length, forecasting horizon, and evaluation metrics to ensure the fairness and reliability of the experimental results. Taking the PM2.5 forecasting task with an input length of 300 steps and a forecasting horizon of 60 steps as an example, the forecasting results of different models are analyzed. The average results of different models on the test set are presented in
Table 6.
As shown in
Table 6, the proposed adaptive fusion model achieves the best overall forecasting performance among all compared models. It obtains the lowest MSE of (1.8292 ± 0.1353) × 10
−3, MAE of 0.0334 ± 0.0010, RMSE of 0.0428 ± 0.0016, and MAPE of 0.0177 ± 0.0005, as well as the highest R
2 of 0.9744 ± 0.0019. These results indicate that the proposed model achieves superior overall prediction accuracy and fitting capability under the setting of an input length of 300 and a prediction horizon of 60 steps. The relatively small standard deviations over three random seeds further suggest that the proposed model maintains stable performance under different random initializations.
Among the baseline models, the inverted structure models generally achieve better forecasting performance than the conventional Transformer-based models. In particular, iReformer shows the strongest performance among the baselines, with an MSE of (2.5258 ± 0.0499) × 10−3, an MAE of 0.0392 ± 0.0005, an RMSE of 0.0503 ± 0.0005, an MAPE of 0.0207 ± 0.0003, and an R2 of 0.9647 ± 0.0007. Based on the mean values, the proposed adaptive fusion model reduces MSE, MAE, RMSE, and MAPE by 27.58%, 14.80%, 14.91%, and 14.49%, respectively, compared with iReformer. Compared with the original iTransformer backbone, the corresponding reductions are 31.08%, 15.87%, 16.89%, and 16.11%, respectively, while R2 increases from 0.9629 ± 0.0030 to 0.9744 ± 0.0019.
The standalone DLinear model obtains an MSE of (55.5129 ± 0.0679) × 10−3, an MAE of 0.1813 ± 0.0003, and an R2 of 0.2233 ± 0.0009, indicating that a purely linear decomposition model is insufficient for this high-frequency PM2.5 forecasting task. However, when DLinear is used as an auxiliary branch in the proposed hybrid framework, it provides complementary local trend information to the nonlinear iTransformer backbone. Therefore, the improvement of the proposed model can be attributed to the coordinated modeling of multivariate dependencies, dual-stage nonlinear feature refinement, and auxiliary local trend compensation.
As shown in
Table 7, the proposed adaptive fusion model exhibits favorable warning-related performance during high-concentration periods. It achieves the lowest High MAE of 0.0457 ± 0.0040 and the lowest PSE of 2.7002 ± 0.2373 among all compared models. This indicates that the proposed model not only reduces the prediction error in high-concentration regions, but also improves the temporal localization of peak events. Compared with the original iTransformer backbone, the High MAE decreases from 0.0542 ± 0.0048 to 0.0457 ± 0.0040, corresponding to a reduction of 15.68%. Meanwhile, the PSE decreases from 2.9236 ± 0.1402 to 2.7002 ± 0.2373. These results suggest that the adaptive fusion strategy enhances the model sensitivity to high-concentration variations while maintaining stable peak timing estimation.
The FAR of the proposed model is 29.2778 ± 3.7516 × 10−4, which is lower than those of iTransformer, iInformer, iReformer, and iFlowformer. This suggests that the improvement in high-concentration prediction is not obtained by simply increasing the number of warning predictions. In contrast, although the conventional Transformer model obtains an extremely low FAR, its MAR reaches 0.7644 ± 0.0292, indicating that most true high-concentration points are not detected. Therefore, a low FAR alone does not necessarily indicate reliable warning performance. Informer, Reformer, Flowformer, and DLinear also show substantially larger High MAE, PSE, and MAR values, reflecting their limited ability to capture abrupt high-concentration changes. It should be noted that the PUR of the proposed model is not the lowest among all models, suggesting that peak underestimation still exists in some cases. Nevertheless, considering its lowest High MAE and PSE together with relatively low FAR and MAR, the proposed adaptive fusion model achieves a more balanced performance in high-concentration warning evaluation.
To further examine the above warning-related behavior visually, a multi-case forecasting comparison is provided in
Figure 2, including low concentration, local peak, abrupt transition, and failure cases. For clarity, only the proposed model, iTransformer, and DLinear are shown. The proposed model generally follows the ground truth more closely than the compared models under stable, local peak, and abrupt transition conditions. In the stable period, the proposed model maintains a smooth prediction and avoids the obvious overestimation observed in DLinear. In the local peak and abrupt transition cases, it better captures the main rising and falling trends, indicating improved adaptability to high-concentration variations. However, the failure case shows that the proposed model still underestimates the amplitude of some sharp peaks. This observation is consistent with the warning-related results and indicates that peak amplitude preservation remains a limitation of the current model.
3.5. Performance Under Different Forecasting Horizons
To further evaluate the performance of the proposed model under different forecasting horizons, this study fixes the input sequence length at 300 steps and sets the forecasting horizons to 30, 60, 96, 180, and 300 steps, respectively, thereby constructing five forecasting tasks: 300-30, 300-60, 300-96, 300-180, and 300-300. Except for the forecasting horizon, the data partitioning strategy, model architecture, and major hyperparameter settings remain identical across all experimental groups to ensure the comparability of different forecasting tasks. The experimental results are presented in
Table 8 and
Table 9.
As shown in
Table 8 and
Table 9, the forecasting and warning performance of the proposed model changes regularly with the increase in the forecasting horizon. For the forecasting metrics, MSE increases from 1.4467 × 10
−3 to 5.7816 × 10
−3, MAE increases from 0.0294 to 0.0593, and R
2 decreases from 0.9797 to 0.9197 as the forecasting horizon increases from 30 to 300 steps. This indicates that longer horizon forecasting introduces greater uncertainty and increases the difficulty of accurately modeling future PM2.5 variations. A similar trend can also be observed in the warning-related metrics. High MAE increases from 0.0372 to 0.0942, PSE increases from 1.6023 to 10.1461, and MAR increases from 0.1134 to 0.3428, indicating that peak value estimation, peak timing localization, and high-concentration detection become more challenging under longer forecasting horizons. Nevertheless, the model maintains relatively low errors and high R
2 values under short and medium forecasting horizons, suggesting that the proposed method has stable forecasting and warning capability within a moderate prediction range.
3.6. Ablation and Interpretability Analysis
To verify the contribution of each proposed improvement module to the enhancement of model performance, this study conducts ablation experiments based on the iTransformer backbone framework. Specifically, six models are constructed, including the original iTransformer, the iTransformer with the dual-stage feed-forward network, the iTransformer with the DLinear branch and fixed weight fusion, the iTransformer with the DLinear branch and adaptive gated fusion, the model simultaneously introducing the dual-stage feed-forward network and the DLinear branch with fixed-weight fusion, and the complete model with the adaptive gated fusion mechanism. All models are trained and tested under identical conditions regarding the data partitioning strategy, input–output settings, and evaluation metrics. The experimental results are presented in
Table 10.
As shown in
Table 10, the proposed model achieves the best overall forecasting performance among all ablation variants. Compared with the original iTransformer, the proposed model reduces the MSE from 2.6539 × 10
−3 to 1.8292 × 10
−3, while the MAE, RMSE, and MAPE are also reduced from 0.0397, 0.0515, and 0.0211 to 0.0334, 0.0428, and 0.0177, respectively. Meanwhile, R
2 increases from 0.9629 to 0.9744. These results indicate that the proposed hybrid structure effectively improves the overall PM2.5 forecasting accuracy.
The contribution of each module can be further observed from the intermediate variants. After replacing the original feed-forward network with the DualStageFFN, the MSE decreases from 2.6539 × 10−3 to 2.0363 × 10−3, indicating that the dual-stage nonlinear feature refinement enhances the representation capability of the iTransformer backbone. When only the DLinear branch is introduced, the MSE decreases to 2.5250 × 10−3 under fixed fusion and to 2.3775 × 10−3 under adaptive fusion. This suggests that the DLinear branch can provide complementary trend information, but its independent contribution is relatively limited compared with the DualStageFFN. When both DualStageFFN and DLinear are used, the fixed fusion variant obtains an MSE of 1.9033 × 10−3, and the complete adaptive fusion model further reduces it to 1.8292 × 10−3. Therefore, the final improvement is not caused by a single module, but by the combined effect of nonlinear feature refinement, auxiliary linear trend modeling, and adaptive branch weighting.
Table 11 further compares the ablation variants from the perspective of warning-oriented forecasting performance. The results show that different modules have different effects on high-concentration prediction and threshold-based warning reliability. Compared with the original iTransformer, the DualStageFFN variant reduces High MAE from 0.0542 to 0.0444 and decreases MAR from 0.2014 to 0.1430, indicating that the enhanced nonlinear backbone improves the fitting ability for high-concentration periods and helps reduce missed alarms. However, its FAR increases from 36.5785 × 10
−4 to 41.0063 × 10
−4, suggesting that stronger sensitivity to high-concentration points may also introduce more false alarms.
The variants with only the DLinear branch do not consistently improve the warning-oriented metrics. In particular, the fixed fusion variant without DualStageFFN increases High MAE and FAR, while the adaptive fusion variant shows higher PUR and MAR than the original iTransformer. This indicates that linear trend compensation alone is insufficient for reliable high-concentration warning. After combining DualStageFFN and DLinear, the fixed fusion variant achieves the lowest High MAE, whereas the proposed adaptive fusion model achieves the lowest FAR. These results suggest that fixed fusion is more conservative in fitting high-concentration points, while adaptive fusion provides better control of false alarms. Overall, the proposed model achieves the best overall forecasting accuracy and a more balanced warning performance, although peak underestimation and missed alarms remain limitations that require further improvement.
To further explain the internal mechanism behind the ablation results, feature representation visualization and adaptive gate weight analysis are conducted in the following part. The feature map comparison is used to examine how the DualStageFFN changes the representation of particulate-related variables, while the gate weight distribution is used to analyze how the model balances the iTransformer branch and the DLinear branch under different PM2.5 regimes.
Figure 3 presents the feature representations extracted by the conventional FFN and the two stages of the proposed DualStageFFN. The complete feature map includes all sensor variables, while the analysis focuses on particulate related variables because PM2.5 is the primary forecasting target in this study. Compared with the conventional FFN, Stage 1 produces broader activation responses across feature dimensions, indicating that the expanded hidden representation enhances the feature extraction capacity of the backbone network. Stage 2 further refines these responses and makes the representation more concentrated on particulate-related features. The peak to stable activation ratio increases from 1.17 after Stage 1 to 1.64 after Stage 2, suggesting that the second stage further strengthens the representation of peak-related features rather than simply repeating the first transformation. This explains why the DualStageFFN improves the overall forecasting performance and the high-concentration prediction performance in the ablation results.
Figure 4 shows the distribution of adaptive gate weights under different PM2.5 conditions. In the proposed fusion formulation, a larger gate value indicates a higher contribution from the iTransformer branch, whereas a smaller gate value indicates a higher contribution from the DLinear branch. The PM2.5 gate has a mean value of 0.4619, a standard deviation of 0.0233, and a range of 0.3458 to 0.6315, indicating that the fusion weight does not collapse to a fixed coefficient. The mean gate decreases from 0.4694 in stable windows to 0.4533 in transition windows, and from 0.4650 in normal concentration windows to 0.4416 in local peak windows. This indicates that the model assigns relatively more weight to the DLinear auxiliary branch during high-concentration and high-variation periods. Therefore, the adaptive gate works as a context-dependent branch balancing mechanism, which helps the model combine nonlinear dependency modeling and local trend compensation.