위키득계정 ↗
기술

DRAEM

DRAEM기술
연구 공개
2021년
발표
ICCV 2021
대상
영상의 표면 이상
학습 구성
재구성과 판별의 결합
DRAEM은 표면 영상에서 이상 부분을 찾기 위해 재구성과 판별을 함께 학습하는 모델이다. 이상이 있는 입력과 정상 모습으로 재구성한 결과를 함께 표현해 이상 위치를 추정한다.

재구성에서 판별로

정상 부분은 잘 복원하고 이상 부분은 잘 복원하지 못하게 하는 접근에서는 결과를 위치 지도로 바꾸는 후처리가 필요할 수 있다. DRAEM은 이 문제를 정상과 이상을 나누는 판별 문제로 함께 다룬다.

입력과 복원 결과의 결합

이상이 포함된 영상과 이상이 없는 형태로 재구성한 영상의 공동 표현을 학습한다. 동시에 정상 예시와 이상 예시 사이의 결정 경계를 학습해 영상 안에서 이상이 나타난 위치를 직접 추정한다.

모의 이상을 이용한 학습

실제 결함만을 모으는 대신 일반적인 이상 모의 생성으로 학습할 수 있도록 설계했다. 원 연구는 MVTec 이상 탐지 자료와 DAGM 표면 결함 자료에서 탐지와 위치 추정 성능을 평가했다.

관련 문서

이상 탐지 / 컴퓨터 비전 / 오토인코더 / MemAE / 영상 분할 / 표현 학습