Daily Arxiv

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

TransXSSM: A Hybrid Transformer State Space Model with Unified Rotary Position Embedding

Created by
  • Haebom

저자

Bingheng Wu, Jingze Shi, Yifan Wu, Nan Tang, Yuyu Luo

개요

본 논문은 Transformer의 장기 의존성 포착 능력과 State Space Model (SSM)의 선형 시간 시퀀스 모델링 능력을 결합하는 새로운 하이브리드 모델 TransXSSM을 제안합니다. Transformer와 SSM의 위치 인코딩 방식 차이(Transformer의 RoPE와 SSM의 암시적 위치 표현)로 인한 성능 저하 문제를 해결하기 위해, 통합된 회전 위치 임베딩(Unified RoPE) 기법을 제시합니다. Unified RoPE를 사용하여 Transformer와 SSM 계층을 일관되게 통합함으로써, 4K 길이의 시퀀스에서 기존 Transformer 모델보다 학습 및 추론 속도가 각각 42.3%, 29.5% 향상되고, 언어 모델링 벤치마크에서 4% 이상 높은 정확도를 달성합니다. 또한, 모델 크기 증가에 따른 성능 향상 효율도 더 높음을 보여줍니다.

시사점, 한계점

시사점:
Transformer와 SSM의 장점을 통합한 효율적이고 정확한 장문맥 모델링 가능성 제시.
Unified RoPE를 통한 위치 인코딩 불일치 문제 해결.
기존 Transformer 모델 대비 향상된 학습 및 추론 속도, 정확도 달성.
모델 크기 증가에 따른 성능 향상의 효율성 증대.
한계점:
제안된 방법의 일반화 성능에 대한 추가적인 실험 필요.
다른 유형의 시퀀스 데이터나 작업에 대한 적용 가능성 검증 필요.
Unified RoPE의 설계 및 최적화에 대한 추가적인 연구 필요.
👍