이 논문은 초대형 커널 크기를 가진 컨볼루션 신경망 모델을 제안한다. 기존 연구에서는 커널 크기를 늘리면 매개변수 수가 급격히 증가하여 최적화 문제가 발생하는 문제가 있었다. 이 논문에서는 인간의 주변 시각 메커니즘에서 영감을 얻어 주변 컨볼루션(Peripheral Convolution)이라는 새로운 컨볼루션 형태를 제안하였다.
주변 컨볼루션은 다음과 같은 특징을 가진다:
이를 바탕으로 제안한 Parameter-efficient Large Kernel Network (PeLK) 모델은 다양한 컴퓨터 비전 과제에서 최신 성능을 달성하였다. 특히 ImageNet 분류, ADE20K 의미 분할, COCO 객체 탐지 등에서 우수한 성능을 보였다. 또한 101x101 크기의 초대형 커널을 성공적으로 적용할 수 있었다.
분석 결과, PeLK의 강력한 성능은 주변 컨볼루션을 통해 매우 큰 유효 수용 영역(ERF)을 확보할 수 있었기 때문인 것으로 나타났다. 또한 주변 컨볼루션은 O(K^2)에서 O(log K)로 파라미터 복잡도를 줄일 수 있어, 초대형 커널 크기를 효율적으로 적용할 수 있었다.
이 연구는 생물학적으로 영감을 얻은 메커니즘이 현대 신경망 설계에 유용할 수 있음을 시사한다.
A otro idioma
del contenido fuente
arxiv.org
Ideas clave extraídas de
by Honghao Chen... a las arxiv.org 03-13-2024
https://arxiv.org/pdf/2403.07589.pdfConsultas más profundas