Since the introduction of Masked Autoencoders, various improvements to masking techniques have been explored. In this paper, we rethink masking strategies for audio representation learning using masked prediction-based self-supervised learning (SSL) on general audio spectrograms. While recent informed masking techniques have attracted attention, we observe that they incur substantial computational overhead. Motivated by this observation, we propose dispersion-weighted masking (DWM), a lightweight masking strategy that leverages the spectral sparsity inherent in the frequency structure of audio content. Our experiments show that inverse block masking, commonly used in recent SSL frameworks, improves audio event understanding performance while introducing a trade-off in generalization. The proposed DWM alleviates these limitations and computational complexity, leading to consistent performance improvements. This work provides practical guidance on masking strategy design for masked prediction-based audio representation learning.
Figures & tables
Fig. 1: Masking examples on an audio spectrogram (hint ratio rh=0.0 ). Object-centric masking (SGIM) almost entirely covers audio events (objects), whereas the proposed DWM applies dispersion-weighted random masking based on patch-wise spectral variability, yielding masking patterns that loosely align with object-centric behavior.
Env. sound tasks
Speech tasks
Music tasks
Masking strategy
ESC-50
US8K
SPCV2
VC1
VF
CRM-D
GTZAN
NSynth
Surge
Avg.
(MSM-MAE variants)
Random
89.2 ± 0.4
87.2 ± 0.2
96.1 ± 0.1
73.4 ± 0.4
97.9 ± 0.1
71.3 ± 0.2
80.3 ± 1.0
74.6 ± 0.4
43.2 ± 0.2
79.2
IBM [ 6 ]
90.1 ± 0.3
88.2 ± 0.4
95.4 ± 0.1
65.1 ± 0.5
97.7 ± 0.1
70.9 ± 0.6
80.5 ± 2.3
74.2 ± 0.6
42.8 ± 0.1
78.3
vs. random [pp]
+0.9
+1.0
-0.7
-8.3
-0.2
-0.4
+0.2
-0.4
-0.4
-0.9
DWM (proposed)
89.9 ± 0.7
87.4 ± 0.1
95.9 ± 0.1
72.5 ± 0.7
97.8 ± 0.1
70.9 ± 0.3
81.8 ± 1.1
75.5 ± 0.1
43.3 ± 0.3
79.5
TABLE I: Linear evaluation results (%) with 95% CI comparing different masking strategies.
Masking
AS2M
AS20K
ESC-50
SPCV2
VC1
strategy
mAP
mAP
acc(%)
acc(%)
acc(%)
(MAE variants)
Random
47.4 ± 0.1
37.9 ± 0.0
95.4 ± 0.1
98.4 ± 0.0
96.6 ± 0.2
IBM [ 6 ]
47.4 ± 0.1
38.6 ± 0.1
95.4 ± 0.3
98.3 ± 0.0
95.4 ± 0.5
diff.[pp]
0.0
+0.7
0.0
-0.1
-1.2
DWM
47.5 ± 0.1
38.3 ± 0.1
95.5 ± 0.2
98.4 ± 0.1
96.4 ± 0.2
TABLE II: Fine-tuning results with 95% CI comparing masking strategies.
Env. sound tasks
Speech tasks
Music tasks
Scheduling
ESC-50
US8K
SPCV2
VC1
VF
CRM-D
GTZAN
NSynth
Surge
Avg.
Baseline MSM-MAE
89.2 ± 0.4
87.2 ± 0.2
96.1 ± 0.1
73.4 ± 0.4
97.9 ± 0.1
71.3 ± 0.2
80.3 ± 1.0
74.6 ± 0.4
43.2 ± 0.2
79.2
Hint ratio rh =0.0
89.9 ± 1.1
87.3 ± 0.7
95.5 ± 0.1
69.1 ± 0.1
97.7 ± 0.0
70.4 ± 0.6
83.4 ± 4.5
75.4 ± 0.2
42.5 ± 0.9
79.0
Hint ratio rh =0.5
87.4 ± 0.9
86.3 ± 0.6
96.0 ± 0.2
71.8 ± 0.5
97.6 ± 0.3
70.5 ± 0.9
82.1 ± 3.1
74.3 ± 0.4
42.5 ± 0.5
78.7
Schedule γ =1
88.1 ± 0.2
87.3 ± 0.3
96.2 ± 0.2
70.0 ± 0.5
97.7 ± 0.2
71.4 ± 0.9
82.0 ± 6.4
76.0 ± 0.2
42.6 ± 0.7
79.0
Schedule γ =2
89.9 ± 0.7
87.4 ± 0.1
95.9 ± 0.1
72.5 ± 0.7
97.8 ± 0.1
70.9 ± 0.3
81.8 ± 1.1
75.5 ± 0.1
43.3 ± 0.3
79.5
TABLE III: Ablation of hint ratio scheduling in DWM: Linear evaluation results (%) with 95% CI.
Fig. 2: Hint ratio rh schedules over training epochs for ablation settings.
Audio self-supervised learning (SSL) aims to learn general-purpose representations from large-scale unlabeled audio data. While recent advances have been driven mainly by generative reconstruction objectives, contrastive approaches remain less explored, partly due to the difficulty of designing effective audio augmentations and the large batch sizes required for contrastive pre-training. We introduce AudioMosaic, a contrastive learning-based audio encoder for general audio understanding. During pre-training, AudioMosaic constructs positive pairs by applying structured time-frequency masking to spectrogram patches, which reduces memory usage and enables efficient large-batch training. Compared with generative approaches, the AudioMosaic encoder learns more discriminative utterance-level representations that demonstrate strong transferability across datasets, domains, and acoustic conditions. Extensive experiments show that AudioMosaic achieves state-of-the-art performance on several standard audio benchmarks under both linear probing and fine-tuning. We further show that integrating the pretrained AudioMosaic encoder into audio-language models improves performance on audio-language tasks. The code is publicly available in our GitHub repository.
Hanxun Huang, Qizhou Wang, Xingjun Ma +3
School of Computing and Information Systems, The University of Melbourne, Australia · Institute of Trustworthy Embodied AI, Fudan University, China · Baskin School of Engineering, University of California, Santa Cruz, USA
Self-supervised learning (SSL) has driven substantial progress in audio representation learning, though existing methods have increasingly relied on elaborate pre-training recipes to reach competitive performance. A markedly different pre-training philosophy underpins the most influential progress in language modeling and, more recently, in visual representation learning: rather than train encoders as static feature extractors, models are trained to predict the next element, a discrete token or a continuous embedding, from the preceding context. Autoregressive prediction thereby provides a unified pre-training interface that transfers across modalities, compelling the model to learn the underlying data distribution. We ask whether such a simple causal paradigm can yield strong audio learners, given that audio's temporal structure makes autoregressive prediction of patch embeddings a natural fit. We introduce NAPE (Next-Audio-Patch-Embedding prediction), a self-supervised framework in which a causal Transformer predicts each next patch embedding of a log-mel spectrogram from the previous ones, using causal masking and stop-gradient as its sole training signal. The design is intentionally minimalist, avoiding reconstruction decoders, acoustic tokenizers, student-teacher setups, and auxiliary regularization losses. Across six audio and speech benchmarks, NAPE achieves state-of-the-art fine-tuning performance on several tasks, scales consistently across encoder sizes, and yields strong linear-probing results. NAPE also produces structured attention patterns without explicit supervision.
Masked modeling has emerged as a robust self-supervised learning framework. However, most methods rely on random masking, which disregards the structural properties of different data modalities. To align with the spatiotemporal and spectral characteristics of video and audio data, we introduce a structured noise-based masking approach. By filtering white noise into different color noise distributions, we generate structured masks that capture modality-specific patterns without requiring handcrafted heuristics or access to the data. Our approach enhances masked video and audio modeling frameworks without any additional computational cost. Experiments show that structured noise masking consistently outperforms random masking, underscoring the value of modality-aware masking strategies for representation learning.
Aritra Bhowmik, Carlos Hinojosa, Fida Mohammad Thoker +2
University of Amsterdam · King Abdullah University of Science and Technology