HTS-AT: A Hierarchical Token-Semantic Audio Transformer for Sound Classification and Detection
This work addresses scalability issues in audio transformers for researchers and practitioners by improving efficiency and performance in sound classification and detection tasks.
The authors tackled the problem of audio classification and detection by introducing HTS-AT, a hierarchical token-semantic audio transformer that reduces model parameters by 65% and training time by 85% compared to previous audio transformers, achieving new state-of-the-art results on AudioSet and ESC-50 datasets.
Audio classification is an important task of mapping audio samples into their corresponding labels. Recently, the transformer model with self-attention mechanisms has been adopted in this field. However, existing audio transformers require large GPU memories and long training time, meanwhile relying on pretrained vision models to achieve high performance, which limits the model's scalability in audio tasks. To combat these problems, we introduce HTS-AT: an audio transformer with a hierarchical structure to reduce the model size and training time. It is further combined with a token-semantic module to map final outputs into class featuremaps, thus enabling the model for the audio event detection (i.e. localization in time). We evaluate HTS-AT on three datasets of audio classification where it achieves new state-of-the-art (SOTA) results on AudioSet and ESC-50, and equals the SOTA on Speech Command V2. It also achieves better performance in event localization than the previous CNN-based models. Moreover, HTS-AT requires only 35% model parameters and 15% training time of the previous audio transformer. These results demonstrate the high performance and high efficiency of HTS-AT.