XAI 기반 반려견 품종 분류

초록

최근 반려견에 대한 관심이 날이 갈수록 뜨거워지고 있다. 우리나라는 반려견을 기르는 인구가 급증하면서 '펫코노미’ 시대가 도래했다. 2020년 현재 반려견은 약 512만 가구에서 기르는 것으로 나타났다. 사람들은 반려견을 입양할 때 외모와 성격에 따라 견종을 선택한다. 하지만 펫샾에서 더 비싼 품종으로 속이는 문제가 발생해오고 있다. 이에 딥러닝을 이용한 품종 구별에 관한 연구가 활발히 진행되어왔다. 하지만 딥러닝을 이용한 품종 구별 방법을 구매자가 이해하는 것은 어렵다. 이러한 문제를 해결하기 위해 본 논문에서 우리는 반려견의 품종을 구별하는 CNN 기반의 딥러닝 모델을 구축한 후 이를 XAI 기법 중 하나인 LIME을 이용하여 분류 근거를 이미지로 보여준다. 데이터 셋을 위해 우리나라에서 가장 많이 키우는 견종을 선정한 후, 서로 구분이 어려운 품종들을 추가하였다. 그런 다음 실험들을 통해 최적의 하이퍼파라미터들을 찾아 분류의 근거를 가장 잘 보여주는 이미지를 추출했다. 이를 근거로 펫을 입양하는 사람은 자신이 입양하려는 펫의 품종을 믿을 수 있고, 펫샵 역시 품종의 근거를 제시함으로써 신뢰를 구축할 수 있다.

키워드