전력선 인식을 위한 의미론적 세분화 모델과 학습 전략 (Semantic Segmentation Model and Training Strategy for Powerline Recognition)

서지정보

· 발행기관 : 한국차세대컴퓨팅학회
· 수록지 정보 : 한국차세대컴퓨팅학회 논문지 / 19권 / 1호 / 78 ~ 88페이지
· 저자명 : 김대한, 노현철, 김민우, 안병태, 최동걸

초록

최근 의미론적 세분화 작업에서 컨볼루션과 트랜스포머 기반의 다양한 딥러닝 모델이 제안되고 있다. 이러한 모델들은 공개 벤치마크의 다양한 객체에 대해 주목할만한 성능을 보여준다. 그러나 우리는 전력선과 같이 전경과 배경의 비율이 과도하게 불균형하며 얇고 긴 객체에 대해 최신 모델조차도 분할에 실패하는 경향이 있음을 관찰한다. 본 논문에서 우리는 전력선 인식을 위해 전력선 데이터 세트를 수집하고, 해당 데이터 세트에서 전력선 검출 실패에 대한 다음의 세 가지 원인이 있음을 발견한다. 1) 레이블링 작업의 어려움, 2) 대규모 훈련 데이터 세트의 부족, 3) 극도로 불균형한 클래스 분포. 더 나아가, 우리는 해당 문제를 해결하기 위해 전력선 인식에 대해 높은 성능을 달성하는 모델 선택에 대한 기준과 훈련 전략을 제시한다. 실험 결과 제시된 학습 전략은 인식 작업에서 기준선 모델과 비교해 20 mIoU 이상의 성능 향상을 달성한다.

영어초록

Recently, various deep learning models based on convolution and transformer have been proposed for semantic segmentation tasks. These models show remarkable performance on a variety of objects on public benchmarks. However, we observe that even state-of-the-art models tend to fail in segmentation for thin and long objects with an excessively imbalanced foreground-background ratio, such as powerlines. In this paper, we collect a powerline dataset for powerline segmentation and identify three main causes for powerline segmentation failure in the dataset. 1) difficulty in labeling tasks, 2) lack of large training datasets, 3) extremely unbalanced class distribution. Furthermore, we present criteria for model selection and training strategies that achieve high performance for powerline segmentation. Our experimental results shows that the presented training strategy achevied an improvement of over 20 mIoU in the powerline segmentation task compared to the baseline model.

참고자료

· 없음