Daily Arxiv

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

Parameter Efficient Mamba Tuning via Projector-targeted Diagonal-centric Linear Transformation

Created by
  • Haebom
Category
Empty

저자

Seokil Ham, Hee-Seon Kim, Sangmin Woo, Changick Kim

개요

본 논문은 Transformer 아키텍처를 대체할 가능성이 있는 Mamba 아키텍처에 대한 Parameter-Efficient Fine-Tuning (PEFT) 방법을 제시합니다. 기존의 Mamba 아키텍처에서 State-Space Models (SSMs)의 중요성이 강조되었지만, 본 연구는 전이 학습에 있어서는 Projectors가 SSMs보다 훨씬 중요한 역할을 한다는 것을 발견했습니다. 이러한 발견을 바탕으로, pretrained Projectors만을 Diagonal-centric Linear Transformation을 통해 최적화하는 새로운 PEFT 방법인 ProDiaL을 제안합니다. ProDiaL은 전체 파라미터의 1% 미만만을 사용하면서도 비전 및 언어 Mamba 모델에서 모두 강력한 성능을 보여줍니다.

시사점, 한계점

시사점:
Mamba 아키텍처에서 전이 학습에 있어 Projectors의 중요성을 규명했습니다.
Mamba 아키텍처에 특화된 효율적인 PEFT 방법인 ProDiaL을 제시했습니다.
ProDiaL은 적은 파라미터로 비전 및 언어 모델 모두에서 우수한 성능을 달성했습니다.
한계점:
ProDiaL의 성능이 다른 PEFT 방법들과 비교 분석되지 않았습니다.
ProDiaL의 일반화 성능에 대한 추가적인 실험이 필요합니다.
다양한 크기의 Mamba 모델에 대한 실험 결과가 제시되지 않았습니다.
👍