Preprint
Machine Learning

BROS

August 1, 2021

0

Citations

0

Influential Citations

Venue

2021

Year

Abstract

Built upon BERT, encodes relative positions of texts in 2D space and learns from unlabeled documents with area masking strategy.

Analysis

Why This Paper Matters

Document understanding is a critical yet challenging area in AI, as real-world documents (forms, invoices, receipts) combine textual content with complex spatial layouts. Traditional NLP models like BERT treat text as a flat sequence, ignoring the 2D arrangement of words on a page. BROS addresses this gap by encoding relative positions of text tokens in 2D space, enabling the model to capture layout cues such as alignment, proximity, and reading order. This is particularly important for tasks where layout determines meaning, such as distinguishing a field label from its value.

The paper also introduces a novel area masking strategy for self-supervised pre-training, which masks contiguous regions of the document rather than random tokens. This forces the model to learn both textual and spatial context, making it more robust for downstream tasks. By building on BERT, BROS leverages existing pre-training knowledge while adapting it to the document domain.

Technical Contributions

  • 2D Relative Position Encoding: Instead of absolute positions, BROS uses relative offsets between token pairs in both x and y dimensions, allowing the model to generalize across different document layouts and sizes.
  • Area Masking Strategy: During pre-training, contiguous rectangular regions of the document are masked, requiring the model to predict missing tokens based on surrounding text and layout. This is more natural for documents than random token masking.
  • BERT-based Architecture: The model extends BERT with additional position embeddings for 2D coordinates, maintaining compatibility with existing transformer infrastructure.

Results

The abstract does not include specific metrics or comparisons. Typically, such models are evaluated on benchmarks like FUNSD (form understanding), SROIE (receipt OCR), or RVL-CDIP (document classification). Expected improvements would be in F1 scores for entity extraction and classification accuracy, especially on layout-sensitive tasks.

Significance

BROS contributes to the growing field of document AI by providing a principled way to incorporate spatial information into transformer models. This has practical implications for automating document processing in industries like finance, healthcare, and legal. The area masking strategy also offers a more effective pre-training objective for documents, potentially inspiring future work on self-supervised learning for structured data.