Multi-Level Feature Fusion for Continual Learning in Visual Quality Inspection
By: Johannes C. Bauer , Paul Geng , Stephan Trattnig and more
Potential Business Impact:
Teaches machines to fix broken things better.
Deep neural networks show great potential for automating various visual quality inspection tasks in manufacturing. However, their applicability is limited in more volatile scenarios, such as remanufacturing, where the inspected products and defect patterns often change. In such settings, deployed models require frequent adaptation to novel conditions, effectively posing a continual learning problem. To enable quick adaptation, the necessary training processes must be computationally efficient while still avoiding effects like catastrophic forgetting. This work presents a multi-level feature fusion (MLFF) approach that aims to improve both aspects simultaneously by utilizing representations from different depths of a pretrained network. We show that our approach is able to match the performance of end-to-end training for different quality inspection problems while using significantly less trainable parameters. Furthermore, it reduces catastrophic forgetting and improves generalization robustness to new product types or defects.
Similar Papers
A Dataset and Baseline for Deep Learning-Based Visual Quality Inspection in Remanufacturing
CV and Pattern Recognition
Helps machines spot broken car parts automatically.
Generalizable and Adaptive Continual Learning Framework for AI-generated Image Detection
CV and Pattern Recognition
Detects fake pictures even from new AI.
A Comparative Study of Custom CNNs, Pre-trained Models, and Transfer Learning Across Multiple Visual Datasets
CV and Pattern Recognition
Transfer learning finds best image matches.