AOI-SSL: Self-Supervised Framework for Efficient Segmentation of Wire-bonded Semiconductors In Optical Inspection

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Figueira, Joaquín, Van Gastel, Rob, D'Amicantonio, Giacomo, Liu, Zhuoran, Bucur, Ioan Gabriel, Boughorbel, Faysal, Bondarev, Egor
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866917486975254528
author Figueira, Joaquín
Van Gastel, Rob
D'Amicantonio, Giacomo
Liu, Zhuoran
Bucur, Ioan Gabriel
Boughorbel, Faysal
Bondarev, Egor
author_facet Figueira, Joaquín
Van Gastel, Rob
D'Amicantonio, Giacomo
Liu, Zhuoran
Bucur, Ioan Gabriel
Boughorbel, Faysal
Bondarev, Egor
contents Segmentation models in automated optical inspection of wire-bonded semiconductors are typically device-specific and must be re-trained when new devices or distribution shifts appear. We introduce AOI-SSL, a training-efficient framework for semantic segmentation of wire-bonded semiconductors by combining small-domain self-supervised pre-training of vision transformers with in-context inference that minimizes the need of labeled examples. We pre-train SOTA self-supervised algorithms in a small industrial inspection dataset and find that Masked Autoencoders are the most effective in this small-data setting, improving downstream segmentation while reducing the labeled fine-tuning effort. We further introduce in-context, patch-level retrieval methods that predict masks directly from dense encoder embeddings with negligible additional training. We show that, in this setting, simple similarity-based retrieval performs on par with more complex attention-based aggregation used currently in the literature. Furthermore, our experiments demonstrate that self-supervised pre-training significantly improves segmentation quality compared to training from scratch and to ImageNet pre-trained backbones under a fixed fine-tuning computational budget. Finally, the results reveal that retrieval based segmentation outperforms fine-tuning when targeting single device images, allowing for near-instant adaptation to difficult samples.
format Preprint
id arxiv_https___arxiv_org_abs_2605_12430
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle AOI-SSL: Self-Supervised Framework for Efficient Segmentation of Wire-bonded Semiconductors In Optical Inspection
Figueira, Joaquín
Van Gastel, Rob
D'Amicantonio, Giacomo
Liu, Zhuoran
Bucur, Ioan Gabriel
Boughorbel, Faysal
Bondarev, Egor
Computer Vision and Pattern Recognition
I.5.4; I.4.9
Segmentation models in automated optical inspection of wire-bonded semiconductors are typically device-specific and must be re-trained when new devices or distribution shifts appear. We introduce AOI-SSL, a training-efficient framework for semantic segmentation of wire-bonded semiconductors by combining small-domain self-supervised pre-training of vision transformers with in-context inference that minimizes the need of labeled examples. We pre-train SOTA self-supervised algorithms in a small industrial inspection dataset and find that Masked Autoencoders are the most effective in this small-data setting, improving downstream segmentation while reducing the labeled fine-tuning effort. We further introduce in-context, patch-level retrieval methods that predict masks directly from dense encoder embeddings with negligible additional training. We show that, in this setting, simple similarity-based retrieval performs on par with more complex attention-based aggregation used currently in the literature. Furthermore, our experiments demonstrate that self-supervised pre-training significantly improves segmentation quality compared to training from scratch and to ImageNet pre-trained backbones under a fixed fine-tuning computational budget. Finally, the results reveal that retrieval based segmentation outperforms fine-tuning when targeting single device images, allowing for near-instant adaptation to difficult samples.
title AOI-SSL: Self-Supervised Framework for Efficient Segmentation of Wire-bonded Semiconductors In Optical Inspection
topic Computer Vision and Pattern Recognition
I.5.4; I.4.9
url https://arxiv.org/abs/2605.12430