mlsanityData to 14 Oct 2026

HIRA: A Human-in-the-Loop Retrieval-Augmented Cascade for Document Classification in Regulated Industries

arXiv 2608.21792 ↗First posted 22 Aug 2026cs.AI cs.CV cs.IRcited by 0

On the map: Retrieval-augmented generation → Generation and context integration

Abstract (arXiv)

Document classification in regulated industries is constrained by data residency, limited cold-start labels, scarce review capacity, and costly model-governance procedures. We present HIRA, a training-free, on-premises retrieval-augmented cascade for document classification in regulated deployments that combines BM25 over OCR text, dense text embeddings, and image-level representations through validation-calibrated weighted reciprocal-rank fusion. Confident documents are classified directly by retrieval; uncertain or visually confusable documents are passed to a locally hosted LLM verifier, which receives the OCR text, retrieved exemplars, label descriptions, and confusion-specific terms. When the verifier remains uncertain, the document is sent to human review. Each correction is stored as a margin-weighted retrieval exemplar and updates a Dirichlet-smoothed confusion graph, letting the system improve without updating model weights. On a private 80-class trade-finance corpus, HIRA processes the full 30,233-document production stream while requesting human correction for only 1,945 documents (6.4%), improving Macro-F1 from 0.6218 to 0.8548. On the corrected Tobacco-3482 benchmark, HIRA reaches 0.9423 Macro-F1 with a locally hosted DeepSeek-R1-Distill-Qwen-32B verifier, 17.4 percentage points above the zero-shot LLM baseline, while invoking the verifier for only about 40% of documents and reducing LLM calls by approximately 60%. With 518 human corrections (24.8% of the pool), HIRA matches the fully labelled pool oracle, in which all 2,086 pool documents are indexed with their ground-truth labels. These results show that selective human feedback and retrieval-memory adaptation can be a practical alternative to repeated model retraining for long-tail document classification in regulated deployments.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Generation and context integration, one of 15 directions in Retrieval-augmented generation. Builds on 13 of its 22 references that are in the map. First posted on arXiv 22 Aug 2026.

Builds on

13 of 22 references.

References · 13

  1. 2024The Llama 3 Herd of Modelscited by 14,715
  2. 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,944
  3. 2020Dense Passage Retrieval for Open-Domain Question Answeringcited by 3,824
  4. 2020MiniLM: Deep Self-Attention Distillation for Task-Agnostic Compression of Pre-Trained Transformerscited by 1,463
  5. 2023FrugalGPT: How to Use Large Language Models While Reducing Cost and Improving Performancecited by 853
  6. 2019Generalization through Memorization: Nearest Neighbor Language Modelscited by 800
  7. 2022LayoutLMv3: Pre-training for Document AI with Unified Text and Image Maskingcited by 485
  8. 2020LayoutLMv2: Multi-modal Pre-training for Visually-Rich Document Understandingcited by 408
  9. 2022DiT: Self-supervised Pre-training for Document Image Transformercited by 152
  10. 2022Unifying Vision, Text, and Layout for Universal Document Processingcited by 140
  11. 2022PP-OCRv3: More Attempts for the Improvement of Ultra Lightweight OCR Systemcited by 106
  12. 2023kNN Prompting: Beyond-Context Learning with Calibration-Free Nearest Neighbor Inferencecited by 47
1 more
  1. 2025Exploring Selective Retrieval-Augmentation for Long-Tail Legal Text Classificationcited by 1

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026A Classifier That Teaches Itself: Self-Improving, Frozen-gate Training (SIFT) for Dynamic Document Classification
  2. 2023Retrieval-augmented Multi-label Text Classification
  3. 2026RealDocBench: A Benchmark for Field-Level QA and Layout Understanding on Real-World Regulated Documents
  4. 2026Retrieval Augmented Classification for Confidential Documents
  5. 2026Docs2Synth: A Synthetic Data Trained Retriever Framework for Scanned Visually Rich Documents Understanding
  6. 2026Constraint-Aware Hierarchical Search for Regulation-Driven Fine-Grained Classification
  7. 2026RAPT: Retrieval-Augmented Post-hoc Thresholding for Multi-Label Classification
  8. 2026DocHRL: A Hierarchical Reinforcement Learning Framework for Cost-Optimised Document Classification