Command Palette

Search for a command to run...

4달 전

다중 레이블 이미지 분류를 위한 적응형 그래프 컨볼루션 네트워크(ML-AGCN)

{Djamila Aouada Oyebade Oyedotun Enjie Ghorbel Inder Pal Singh}

초록

본 논문에서는 다중 레이블 이미지 분류를 위한 새로운 그래프 기반 접근법인 다중 레이블 적응형 그래프 컨볼루션 네트워크(Multi-Label Adaptive Graph Convolutional Network, ML-AGCN)를 제안한다. 그래프 기반 방법은 다중 레이블 분류 분야에서 큰 잠재력을 보여왔다. 그러나 기존의 이러한 접근법들은 레이블 간 의존성을 모델링하기 위해 그래프 구조를 수동적으로 고정하는 방식을 사용하는데, 이는 최적의 구조가 아닐 수 있다. 이를 해결하기 위해, 우리는 그래프 구조를 엔드 투 엔드(end-to-end) 방식으로 학습하는 방법을 제안한다. 구체적으로, 그래프 노드 간의 쌍별 중요도를 추정하기 위해 어텐션 기반 메커니즘을 도입하고, 서로 다른 노드 간의 특징 유사성을 유지하기 위해 유사도 기반 메커니즘을 활용한다. 이를 통해 그래프를 더 유연하게 적응형으로 모델링할 수 있는 새로운 방식을 제공한다. 제안한 ML-AGCN의 성능은 MS-COCO와 VG-500이라는 두 가지 유명한 데이터셋에서 평가되었으며, 실험 결과 ML-AGCN이 최신 기술 대비 뛰어난 성능을 보이며 동시에 모델 파라미터 수를 감소시킴을 확인할 수 있었다.

벤치마크

벤치마크방법론지표
multi-label-image-classification-on-mscocoML-AGCN
mean average precision: 86.9

AI로 AI 구축

아이디어에서 출시까지 — 무료 AI 공동 코딩, 즉시 사용 가능한 환경, 최적 가격 GPU로 AI 개발을 가속화하세요.

AI 공동 코딩
즉시 사용 가능한 GPU
최적 가격
시작하기

Hyper Newsletters

최신 정보 구독하기
한국 시간 매주 월요일 오전 9시 에 이번 주의 최신 업데이트를 메일로 발송합니다
이메일 서비스 제공: MailChimp
다중 레이블 이미지 분류를 위한 적응형 그래프 컨볼루션 네트워크(ML-AGCN) | 연구 논문 | HyperAI초신경