arXiv · 2104.04692
Not All Attention Is All You Need
Abstract
Beyond the success story of pre-trained language models (PrLMs) in recent natural language processing, they are susceptible to over-fitting due to unusual large model size. To this end, dropout serves as a therapy. However, existing methods like random-based, knowledge-based and search-based dropout are more general but less effective onto self-attention based models, which are broadly chosen as the fundamental architecture of PrLMs. In this paper, we propose a novel dropout method named AttendOut to let self-attention empowered PrLMs capable of more robust task-specific tuning. We demonstrate that state-of-the-art models with elaborate training design may achieve much stronger results. We verify the universality of our approach on extensive natural language processing tasks.
Explore related subjects
Keep this discovery
Hongqiu Wu, Hai Zhao, Min Zhang. 2021-04-10. Not All Attention Is All You Need. https://arxiv.org/abs/2104.04692
Cite the original work for its findings. Save a collection to share your selection of sources.