서울시립대학교

- 컴퓨터 비전 및 인공지능 분야 최고 권위 학회 ‘CVPR 2024’ 에서 논문 최종 채택

- 다양한 구동 환경(GPU, CPU, Mobile)에서 범용적으로 적용되는 이미지 분류 인공지능 모델 개발

서울시립대학교(총장 원용걸) 인공지능학과 기계지능연구실 노영민 교수 연구팀(윤석주 석사과정)이 최근 컴퓨터 비전 분야 CVPR 2024에 제출한 논문이 최종 채택되었다고 밝혔다.

IEEE/CVF 주관하는 CVPR(Computer Vision and Pattern Recognition Conference)은 컴퓨터 비전 및 인공지능-패턴인식 분야에서 최고 권위 학회로 인정받고 있으며, 이번 CVPR 2024는 오는 6월 19일부터 21일까지 미국 시애틀 컨벤션 센터에서 개최될 예정이다.

▲  서울시립대 인공지능학과 기계지능연구실(MIL) 연구팀(좌측부터) 윤석주 석사과정, 노영민 교수

해당 논문의 제목은 ‘SHViT: Single-Head Vision Transformer with Memory Efficient Macro Design’으로, 컴퓨터 비전 분야에서 새로운 모델을 소개하여 속도와 성능을 모두 개선하는 것에 초점을 맞추고 있다. 연구팀은 다양한 인공지능 모델의 구동 환경(GPU, CPU, Mobile)에서 이미지 분류, 객체 탐지 및 영상 분할에 대해 세계최고의 성능과 속도를 보여주는 컴퓨터 비전 backbone 모델을 제안했다. 이 논문에서는 기존의 컴퓨터 비전 모델들이 갖고 있는 계산 중복도 문제를 지적하고, 이를 해결하기 위해 새로운 설계 방법을 제시했다.

▲  제안하는 SHViT 모델의 구조

이 모델은 입력 이미지를 큰 조각으로 처리하고, 단일 헤드만을 사용하여 중복된 계산을 줄이는 방식을 채택하여 모델의 성능은 유지하면서도 속도를 크게 향상했다.

이번 연구는 컴퓨터 비전 분야에서의 중요한 발전을 이끌어냄으로써, 미래에 더욱 빠르고 효율적인 인공지능 기술을 기대할 수 있게 한다. 서울시립대학교의 노영민 교수 연구팀은 앞으로도 컴퓨터 비전과 인공지능 분야에 더 많은 혁신을 이끌어낼 것으로 기대된다.

본 연구는 한국연구재단 개인기초연구사업과 서울시립대학교 데이터사이언스 융합인재양성사업단의 지원으로 수행되었다.