Welcome to the official repository for our solution to the Industrial and Trustworthy AI Challenge: Welding Quality Detection, hosted by Confiance.ai, Renault Group, and IRT SystemX.
Develop a trustworthy AI component that automatically classifies welding seams from automotive production line images into:
- β
OK
: Weld is normal. - β
KO
: Weld has defects. - β
UNKNOWN
: AI is uncertain.
The model assists human operators by reducing manual inspections and improving safety.
- Dataset: 22,753 weld images (C20, C33, C102)
- Imbalance: ~98% OK vs ~2% KO
- Variability: Lighting, blur, angle, position
- Outputs:
- Class:
OK
,KO
,UNKNOWN
- Probability per class (optional)
- OOD score for distribution awareness
- Class:
This is an evolving solution. Core directions include:
- π CNN-based classifier (ResNet/EfficientNet)
- π Transfer learning for limited KO data
- π¨ Augmentation: rotation, blur, contrast, motion
- π― Uncertainty estimation (e.g., MC-Dropout)
β οΈ OOD detection using softmax entropy + VAE- π Feature extraction from welds (length, width, area)
- π§ Explainability using GradCAM