Entropy-aware Masking for Masked Language Modeling

Gokul Srinivasagan, Kai Hartung, Munir Georges


Abstract
Masked language modeling has become a standard pretraining objective for training encoder-based language models. In this approach, certain tokens in the input are masked, and the model learns to predict them using the surrounding context. This process enables the model to capture both syntactic and semantic properties of language. Conventionally, the tokens selected for masking are chosen at random, which may not always yield the most effective learning signals. In this work, we examine a token masking strategy based on entropy distribution. We use the model’s entropy over token predictions to identify which tokens should be masked. This method aims to target tokens that are more informative and uncertain to improve the training efficacy. We also propose a novel self-masking approach that enhances training efficiency without relying on an external reference model. Experimental results demonstrate that our method achieves an average performance improvement of 5% in GLUE scores compared to the baseline. Further, we experiment with combining knowledge distillation with entropy masking, resulting in the best overall results.
Anthology ID:
2026.starsem-conference.27
Volume:
Proceedings of the 15th Joint Conference on Lexical and Computational Semantics (*SEM 2026)
Month:
July
Year:
2026
Address:
San Diego, California, United States
Editors:
Saif M. Mohammad, Nedjma Ousidhoum
Venues:
*SEM | WS
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
395–400
Language:
URL:
https://preview.aclanthology.org/ingest-acl-workshops/2026.starsem-conference.27/
DOI:
Bibkey:
Cite (ACL):
Gokul Srinivasagan, Kai Hartung, and Munir Georges. 2026. Entropy-aware Masking for Masked Language Modeling. In Proceedings of the 15th Joint Conference on Lexical and Computational Semantics (*SEM 2026), pages 395–400, San Diego, California, United States. Association for Computational Linguistics.
Cite (Informal):
Entropy-aware Masking for Masked Language Modeling (Srinivasagan et al., *SEM 2026)
Copy Citation:
PDF:
https://preview.aclanthology.org/ingest-acl-workshops/2026.starsem-conference.27.pdf