本平台为互联网非涉密平台,严禁处理、传输国家秘密或工作秘密

基于开集学习与特征自适应的烟丝杂物检测方法

Open-set learning and feature adaptation for tobacco foreign object detection

  • 摘要: 卷烟制丝过程中混入的塑料、纸片等杂物严重影响产品质量。为解决传统深度学习检测模型在工业现场难以应对杂物类别动态变化的“开集”问题,提出了一种基于开集学习与特征自适应的烟丝杂物检测框架(OS-FOD)。该框架在Faster R-CNN基础上,融合了类对比学习网络(CCL)与不确定性感知网络(UAN),通过结构化特征空间与动态伪未知样本挖掘,使模型具备识别未知杂物的能力。同时,针对烟丝图像背景复杂、目标细小的特点,设计了烟丝特征自适应增强模块(FSFA),通过注意力机制抑制背景噪声、增强杂物特征。结果表明:①烟丝特征自适应增强模块(FSFA)通过通道与空间双重注意力机制,能够有效抑制背景噪声、增强细小杂物特征,使模型在具备开集识别能力的同时保持较高的闭集检测精度;②类对比学习网络(CCL)通过拉近类内样本、推开类间样本,构造紧凑的已知类特征空间与明确的边界区域,降低了未知样本被误判为已知类别的风险;③不确定性感知网络(UAN)动态挖掘高质量伪未知样本,使模型显式学习“未知”概念,未知类召回率(URecall)达到82.42%;④模型对开集杂物的平均识别率为83.20%,表明模型对“未知”杂物具有较好的检测识别能力。该研究为动态工业环境下的杂物检测提供了一种具备开集识别能力的有效方案。

     

    Abstract: Foreign objects such as plastics and paper scraps mixed into cut tobacco during the cigarette manufacturing process seriously affect product safety. Traditional deep learning-based detection models perform well on closed-set datasets but struggle with the "open-set" problem, where impurity categories dynamically change in industrial scenarios. This paper proposes an Open-Set Foreign Object Detection framework (OS-FOD) for tobacco impurities. Built upon Faster R-CNN, OS-FOD integrates a Class Contrastive Learning (CCL) network and an Uncertainty-Aware Labeling Network (UAN). By structuring the feature space and dynamically mining pseudo-unknown samples, the model learns to identify unknown impurities. Furthermore, to address the complex background and small, low-contrast target characteristics, a Feature Self-Adaptive Enhancement module (FSFA) is designed to suppress background noise and enhance impurity features via an attention mechanism. The main conclusions are as follows: 1) the Feature Self-Adaptive Enhancement (FSFA) module suppresses background noise and enhances fine-grained impurity features through dual channel and spatial attention mechanisms, enabling the model to maintain high closed-set detection accuracy while acquiring open-set recognition capability; 2)the Class Contrastive Learning (CCL) network pulls intra-class samples together and pushes inter-class samples apart, constructing a compact known-class feature space with clear boundary regions and reducing the risk of unknown samples being misclassified as known categories; 3)the Uncertainty-Aware Labeling Network (UAN) dynamically mines high-quality pseudo-unknown samples, enabling the model to explicitly learn the "unknown" concept and achieving an unknown-class recall of 82.42%; 4)the open-set decision fusion module calibrates the classification confidence with multi-source uncertainty, improving the reliability of open-set discrimination. This provides an effective solution for reliable impurity detection in dynamic industrial environments. This provides an effective solution for reliable impurity detection in dynamic industrial environments.

     

/

返回文章
返回