Layer-Adaptive State Pruning for Deep State Space Models

Minseon Gwak, Seongrok Moon, Joohwan Ko +1

Due to the lack of state dimension optimization methods, deep state space models (SSMs) have sacrificed model capacity, training search space, or stability to alleviate computation…