Daily Arxiv

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

MambAttention: Mamba with Multi-Head Attention for Generalizable Single-Channel Speech Enhancement

Created by
  • Haebom

저자

Nikolai Lund Kuhne, Jesper Jensen, Jan {\O}stergaard, Zheng-Hua Tan

개요

본 논문은 Mamba와 xLSTM과 같은 새로운 시퀀스 모델이 단일 채널 음성 향상, 자동 음성 인식, 자기 지도 학습 기반 오디오 표현 학습에서 최첨단 모델과 비슷하거나 더 나은 성능을 보인다는 기존 연구를 바탕으로 합니다. 하지만 LSTM 및 Mamba와 같은 시퀀스 모델은 과적합 경향이 있다는 점을 지적하며, 이 문제를 해결하기 위해 Mamba와 시간-주파수 멀티 헤드 어텐션 모듈을 결합한 새로운 하이브리드 아키텍처인 MambAttention을 제안합니다. 더욱 도전적인 노이즈 유형과 낮은 신호 대 잡음비를 갖는 VoiceBank+Demand Extended (VB-DemandEx) 데이터셋을 사용하여 모델을 학습시켰으며, DNS 2020 및 EARS-WHAM_v2와 같은 도메인 외 데이터셋에서 기존 최첨단 LSTM, xLSTM, Mamba 및 Conformer 기반 시스템보다 성능이 월등히 뛰어나다는 것을 실험적으로 보여줍니다. 또한, 시간-주파수 멀티 헤드 어텐션 모듈의 가중치 공유가 일반화 성능에 중요한 역할을 한다는 점을 에이블레이션 스터디를 통해 밝힙니다. 마지막으로, LSTM 및 xLSTM에 공유 시간-주파수 멀티 헤드 어텐션 모듈을 통합하는 실험을 수행하여 도메인 외 데이터셋에서 성능 향상을 확인했지만, MambAttention 모델이 모든 평가 지표에서 더 우수한 성능을 유지함을 확인합니다.

시사점, 한계점

시사점:
Mamba와 시간-주파수 멀티 헤드 어텐션 모듈을 결합한 MambAttention 모델이 단일 채널 음성 향상에서 우수한 일반화 성능을 보임.
시간-주파수 멀티 헤드 어텐션 모듈의 가중치 공유가 일반화 성능 향상에 중요한 역할을 함.
LSTM 및 xLSTM에 시간-주파수 멀티 헤드 어텐션 모듈을 통합하여 성능 향상을 얻을 수 있음.
도전적인 노이즈 환경을 포함하는 VB-DemandEx 데이터셋이 제시됨.
한계점:
제안된 MambAttention 모델의 성능 향상이 특정 데이터셋에 국한될 가능성.
다른 유형의 음성 향상 문제나 더 복잡한 시나리오에 대한 일반화 성능은 추가 연구가 필요.
VB-DemandEx 데이터셋의 구성 및 범용성에 대한 자세한 설명 부족.
👍