A Robust approach to froth flotation images classification using edge-aware progressive refinement attention and reward mechanisms
摘要
Effective flotation beneficiation depends on correctly identifying flotation conditions via froth image analysis. Low visual differences in froth features, such as bubble size, texture, and colour, reduce discriminability, thereby impairing the accurate recognition of the flotation stage. To overcome these challenges, deep learning methods must be applied to improve classification accuracy and flotation effectiveness. We propose a novel Edge-Aware Progressive Refined Attention (EAPRA) mechanism within a deep learning framework to address challenges of similar-class misclassification. This mechanism enhances discriminative features by leveraging edge information, progressively refining attention from global to local regions, suppressing background noise, and effectively extracting texture and shape variation features. In the EAPRA, we used multi-level feature extraction with EfficientNet and an attention mechanism for improved feature extraction. Additionally, we introduce a reward layer that adjusts class logits based on classification confidence, preventing overfitting and improving model robustness. Evaluations on two distinct industrial flotation datasets confirm the model’s superior performance and robustness, achieving accuracy of 98.4 ± 0.03 and 96.7 ± 0.03, respectively, outperforming existing methods.