Segment Any Anomaly without Training via Hybrid Prompt Regularization

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Cao, Yunkang, Xu, Xiaohao, Sun, Chen, Cheng, Yuqi, Du, Zongwei, Gao, Liang, Shen, Weiming
Natura: Preprint
Pubblicazione: 2023
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866909492279508992
author Cao, Yunkang
Xu, Xiaohao
Sun, Chen
Cheng, Yuqi
Du, Zongwei
Gao, Liang
Shen, Weiming
author_facet Cao, Yunkang
Xu, Xiaohao
Sun, Chen
Cheng, Yuqi
Du, Zongwei
Gao, Liang
Shen, Weiming
contents We present a novel framework, i.e., Segment Any Anomaly + (SAA+), for zero-shot anomaly segmentation with hybrid prompt regularization to improve the adaptability of modern foundation models. Existing anomaly segmentation models typically rely on domain-specific fine-tuning, limiting their generalization across countless anomaly patterns. In this work, inspired by the great zero-shot generalization ability of foundation models like Segment Anything, we first explore their assembly to leverage diverse multi-modal prior knowledge for anomaly localization. For non-parameter foundation model adaptation to anomaly segmentation, we further introduce hybrid prompts derived from domain expert knowledge and target image context as regularization. Our proposed SAA+ model achieves state-of-the-art performance on several anomaly segmentation benchmarks, including VisA, MVTec-AD, MTD, and KSDD2, in the zero-shot setting. We will release the code at \href{https://github.com/caoyunkang/Segment-Any-Anomaly}{https://github.com/caoyunkang/Segment-Any-Anomaly}.
format Preprint
id arxiv_https___arxiv_org_abs_2305_10724
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Segment Any Anomaly without Training via Hybrid Prompt Regularization
Cao, Yunkang
Xu, Xiaohao
Sun, Chen
Cheng, Yuqi
Du, Zongwei
Gao, Liang
Shen, Weiming
Computer Vision and Pattern Recognition
Artificial Intelligence
We present a novel framework, i.e., Segment Any Anomaly + (SAA+), for zero-shot anomaly segmentation with hybrid prompt regularization to improve the adaptability of modern foundation models. Existing anomaly segmentation models typically rely on domain-specific fine-tuning, limiting their generalization across countless anomaly patterns. In this work, inspired by the great zero-shot generalization ability of foundation models like Segment Anything, we first explore their assembly to leverage diverse multi-modal prior knowledge for anomaly localization. For non-parameter foundation model adaptation to anomaly segmentation, we further introduce hybrid prompts derived from domain expert knowledge and target image context as regularization. Our proposed SAA+ model achieves state-of-the-art performance on several anomaly segmentation benchmarks, including VisA, MVTec-AD, MTD, and KSDD2, in the zero-shot setting. We will release the code at \href{https://github.com/caoyunkang/Segment-Any-Anomaly}{https://github.com/caoyunkang/Segment-Any-Anomaly}.
title Segment Any Anomaly without Training via Hybrid Prompt Regularization
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2305.10724