단 하나의 정상 이미지로 다중 이상 감지를 혁신하다: OneNIP의 등장
Gao Bin-Bin 박사 연구팀이 개발한 OneNIP는 단 하나의 정상 이미지만으로 다중 클래스 이상을 감지하는 혁신적인 기술입니다. 기존 방법의 한계를 극복하고, 산업용 벤치마크에서 우수한 성능을 입증하여 다양한 분야에서의 응용 가능성을 높였습니다.

최근 딥러닝 기반의 비정상 감지 기술이 급속도로 발전하고 있습니다. 특히, 자기 주의력 트랜스포머를 활용한 비지도 학습 기반 재구성 네트워크는 다중 클래스 이상 감지 분야에서 뛰어난 성능을 보여주고 있습니다. 하지만, 이러한 모델들은 목표 특징에 주로 의존하여 정상 및 비정상 특징 모두를 완벽하게 재구성하는 경우가 발생하며, 이로 인해 이상 감지에 실패할 수 있다는 한계가 존재했습니다. 또한, 낮은 공간 해상도의 잠재 공간에서 재구성을 수행하기 때문에 정확한 이상 분할이 어려웠습니다.
Gao Bin-Bin 박사가 이끄는 연구팀은 이러한 문제를 해결하기 위해 단 하나의 정상 이미지 프롬프트(OneNIP) 를 활용한 새로운 방법을 제안했습니다. OneNIP는 기존의 방법들과는 달리, 단 하나의 정상 이미지만을 사용하여 비정상 특징을 복원하고 재구성합니다. 이는 다중 클래스 이상 감지 성능을 크게 향상시키는 혁신적인 접근 방식입니다. 이는 마치 한 장의 사진으로 전체 그림을 이해하고, 그 안의 이상 요소를 정확하게 찾아내는 것과 같습니다. 연구팀은 여기에 더하여, 실제 정상 이미지와 합성된 비정상 이미지를 사용하여 재구성 오차를 회귀하는 지도 학습 기반 개선기를 제안하여 픽셀 단위의 이상 분할 정확도를 더욱 높였습니다.
OneNIP는 MVTec, BTAD, VisA 등 세 가지 산업용 이상 감지 벤치마크에서 기존 방법들을 능가하는 성능을 달성했습니다. 이는 OneNIP의 실용성과 우수성을 명확하게 보여주는 결과입니다. GitHub(https://github.com/gaobb/OneNIP)에서 코드와 사전 훈련된 모델을 공개하여, 다른 연구자들이 OneNIP를 활용한 연구를 더욱 확장할 수 있도록 지원하고 있습니다.
OneNIP의 등장은 단일 이미지만으로도 다양한 유형의 이상을 효율적이고 정확하게 감지할 수 있는 가능성을 열었습니다. 이는 의료 영상 분석, 제조 공정 모니터링, 자율 주행 등 다양한 분야에서 혁신적인 응용 가능성을 제시하며, 앞으로 더욱 발전된 이상 감지 기술 개발에 중요한 이정표가 될 것으로 예상됩니다. 하지만, 실제 산업 현장 적용을 위해서는 더욱 다양한 데이터셋에 대한 성능 검증과, 실시간 처리 성능 향상 등 추가적인 연구가 필요할 것으로 보입니다.
Reference
[arxiv] Learning to Detect Multi-class Anomalies with Just One Normal Image Prompt
Published: (Updated: )
Author: Bin-Bin Gao
http://arxiv.org/abs/2505.09264v1