딥러닝 기반의 철강 표면의 결함 검출기 (Deep Learning based Surface Defect Detector on Metal Surface)

서지정보

· 발행기관 : 한국지식정보기술학회
· 수록지 정보 : 한국지식정보기술학회 논문지 / 16권 / 5호 / 897 ~ 904페이지
· 저자명 : 서상민

초록

철강 산업에서는 철강의 표면 결함을 감지하고 수리해야 한다. 따라서 철강 표면 결함의 자동 감지는 철강 제조 공정에서 중요한 역할을 한다. 이러한 결함 검출을 위해 HAAR 기능 기반 캐스케이드 분류기 및 서포트 벡터 머신(SVM)과 같은 머신 러닝 기반 분류 방법이 널리 사용되었다. 딥 러닝 방법이 대중화되면서 최근에는 신경망 기반 표면 결함 검출이 도입되고 있다. 일반적으로 많은 연구자들이 최근 ILSVRC (ImageNet Large Scale Visual Recognition Challenge)에서 우승한 성능이 검증된 신경망을 채택한 후, 가중치와 마지막 레이어를 수정하여 표면 결함 감지에 사용하는데, 이를 전이 학습이라고 한다. 이전 연구에서는 ResNet152를 사용했으며, 서로 다른 연구에서 성능은 각각 F1=0.975 및 F1=0.912였다. 그러나 그들의 연구에 사용된 신경망은 매우 넓고 깊어서, 훈련된 가중치를 저장하기 위해서는, 큰 메모리와 많은 승산기-누적기 (multiplier -accumulator, MAC)가 필요하다. 그리고 이것은 강철 표면의 표면 결함을 예측하기 위해 고가의 하드웨어 시스템이 필수적임을 의미합니다. 이 논문은 표면 결함 탐지 전용의 작은 신경망을 제안한다. 제안된 네트워크는 3개의 합성곱 레이어와 2개의 완전 연결 레이어만 있습니다. 실험 결과로부터 F1=0.931, 최소 AUC (area under the curve)=0.995를 얻었다.

영어초록

Steel surface defect should be detected and repaired in steel industry. Therefore, automatic detection of the steel surface defects plays a vital role in the steel manufacturing process. For the defect detection, machine learning based classification methods have been widely used such as HAAR feature-based cascade classifiers and support vector machines (SVM). As deep learning methods have been popular, the neural network based surface defect detection has been recently introduced. As for the methods, many researchers, in general, adopt a trained neural network, which is mainly winner in the recent ILSVRC (ImageNet Large Scale Visual Recognition Challenge). Then, the weights and last layers are modified to be used for surface defect detection (SDD), which is called transfer learning. In the previous researches, ResNet152 (winner in ILSVRC 2015) was used and the resulting performances were F1=0.975 and F1=0.912 in two different studies, respectively. However, the neural network used in their research has very wide and deep. Therefore, huge memories to save the trained weights and many multiplier–accumulators (MAC) are necessary, which means expensive hardware systems are essential to predict surface defect on the steel surface. This paper suggests a small neural network dedicated to surface defect detection. The proposed network has only three convolution layers and two fully connected layers. From the experimental results, we obtained F1=0.931 and minimum AUC (area under the curve)=0.995.

참고자료

· 없음