The Devil is in the Details: Evaluating Limitations of Transformer-based Methods for Granular Tasks
This work addresses a limitation in transformer models for tasks requiring fine-grained text representation, which is incremental as it builds on known methods.
The paper tackles the problem of transformer-based models underperforming on granular textual similarity tasks, showing they are consistently outperformed by simple baselines like TF-IDF, and proposes a method to incorporate TF-IDF that achieves up to 36% relative improvement.
Contextual embeddings derived from transformer-based neural language models have shown state-of-the-art performance for various tasks such as question answering, sentiment analysis, and textual similarity in recent years. Extensive work shows how accurately such models can represent abstract, semantic information present in text. In this expository work, we explore a tangent direction and analyze such models' performance on tasks that require a more granular level of representation. We focus on the problem of textual similarity from two perspectives: matching documents on a granular level (requiring embeddings to capture fine-grained attributes in the text), and an abstract level (requiring embeddings to capture overall textual semantics). We empirically demonstrate, across two datasets from different domains, that despite high performance in abstract document matching as expected, contextual embeddings are consistently (and at times, vastly) outperformed by simple baselines like TF-IDF for more granular tasks. We then propose a simple but effective method to incorporate TF-IDF into models that use contextual embeddings, achieving relative improvements of up to 36% on granular tasks.