Daily Arxiv

전 세계에서 발간되는 인공지능 관련 논문을 정리하는 페이지 입니다.
본 페이지는 Google Gemini를 활용해 요약 정리하며, 비영리로 운영 됩니다.
논문에 대한 저작권은 저자 및 해당 기관에 있으며, 공유 시 출처만 명기하면 됩니다.

MambaNeXt-YOLO: A Hybrid State Space Model for Real-time Object Detection

Created by
  • Haebom

저자

Xiaochun Lei, Siqi Wu, Weilin Wu, Zetao Jiang

개요

본 논문은 실시간 객체 탐지에서 제한된 계산 자원 하에서 속도와 정확도의 균형을 맞추기 위해 MambaNeXt-YOLO라는 새로운 객체 탐지 프레임워크를 제안합니다. 기존 Transformer 기반 아키텍처의 높은 계산 복잡도 문제를 해결하기 위해 선형 상태 공간 모델인 Mamba를 활용하여 효율적인 시퀀스 모델링을 구현합니다. 이는 CNN과 Mamba를 통합한 MambaNeXt Block, 다중 스케일 객체 탐지를 개선하는 MAFPN, 그리고 경량화를 통한 에지 디바이스 배포 지원이라는 세 가지 주요 기여를 통해 이루어집니다. PASCAL VOC 데이터셋에서 전이 학습 없이 31.9 FPS로 66.6% mAP를 달성하였으며, NVIDIA Jetson Xavier NX 및 Orin NX와 같은 에지 디바이스에서의 배포를 지원합니다.

시사점, 한계점

시사점:
Mamba 기반의 효율적인 시퀀스 모델링을 통해 실시간 객체 탐지의 속도와 정확도를 향상시켰습니다.
에지 디바이스에서의 실시간 객체 탐지 가능성을 높였습니다.
전이 학습 없이도 높은 성능을 달성했습니다.
MambaNeXt Block과 MAFPN은 향후 다른 객체 탐지 모델에도 적용될 수 있는 잠재력을 가지고 있습니다.
한계점:
PASCAL VOC 데이터셋만 사용하여 평가되었으므로 다른 데이터셋에서의 성능은 추가적인 검증이 필요합니다.
MambaNeXt-YOLO의 일반화 성능에 대한 추가적인 분석이 필요합니다.
다른 최첨단 실시간 객체 탐지 모델과의 더욱 포괄적인 비교 분석이 필요합니다.
👍