论文标题
用α-钉詹森 - 香农脱落的焊接接头的细粒度分类
Fine-grained Classification of Solder Joints with α-skew Jensen-Shannon Divergence
论文作者
论文摘要
焊接联合检查(SJI)是生产印刷电路板(PCB)的关键过程。 SJI期间检测焊料错误非常具有挑战性,因为焊接接头的尺寸很小,并且可以采取各种形状。在这项研究中,我们首先表明焊料的特征多样性低,并且可以作为一项精细的图像分类任务执行的SJI,该任务侧重于难以固定的对象类。为了提高细粒度的分类精度,发现通过最大化熵来惩罚自信模型预测,在文献中很有用。与此信息内联,我们建议使用α-铲詹森 - 香农发散(α-JS)来惩罚对模型预测的信心。我们将基于注意机制,分割技术,变压器模型和特定损失函数的现有基于熵指定的方法和基于现有的基于熵指定的方法和方法进行比较。我们表明,在细化的焊料联合分类任务中,所提出的方法可以达到不同模型的F1得分和竞争精度。最后,我们可视化激活图,并表明,通过熵进行熵,更精确的类歧视区域是局部的,这也更适合噪声。接受代码将在此处提供。
Solder joint inspection (SJI) is a critical process in the production of printed circuit boards (PCB). Detection of solder errors during SJI is quite challenging as the solder joints have very small sizes and can take various shapes. In this study, we first show that solders have low feature diversity, and that the SJI can be carried out as a fine-grained image classification task which focuses on hard-to-distinguish object classes. To improve the fine-grained classification accuracy, penalizing confident model predictions by maximizing entropy was found useful in the literature. Inline with this information, we propose using the α-skew Jensen-Shannon divergence (α-JS) for penalizing the confidence in model predictions. We compare the α-JS regularization with both existing entropyregularization based methods and the methods based on attention mechanism, segmentation techniques, transformer models, and specific loss functions for fine-grained image classification tasks. We show that the proposed approach achieves the highest F1-score and competitive accuracy for different models in the finegrained solder joint classification task. Finally, we visualize the activation maps and show that with entropy-regularization, more precise class-discriminative regions are localized, which are also more resilient to noise. Code will be made available here upon acceptance.