根据文档复杂度动态选择分类策略,降低计算成本。
DocHRL: A Hierarchical Reinforcement Learning Framework for Cost-Optimised Document Classification

- 分层强化学习决定每份文档用哪个模型或人工审核
- 在RVL-CDIP上达0.973宏F1,单文档成本降至2.74单位
- 适合需要降本增效的文档处理系统开发者
现实中的文档分类流程对所有文档使用相同的模型序列,导致资源浪费:简单文档被过度处理,复杂文档可能得不到充分分析。本文提出DocHRL,一种分层强化学习框架,可针对每份文档自适应选择最经济的分类策略。该框架将分类任务建模为两层策略的序列决策问题:顶层策略在视觉分类器、大语言模型、OCR和人工审核之间选择,子策略则确定具体调用的模型或工具。奖励信号为负的总预期成本,包含推理成本、误分类成本和人工标注成本。在RVL-CDIP基准上,使用近端策略优化训练后,DocHRL在16类文档上实现0.973的宏F1,平均单文档成本降至2.74个归一化单位,显著低于固定独立分类器的成本。结果表明,成本感知的强化学习可同时提升分类性能与运行效率。
原文摘要 · Abstract (English)
Real-world document classification pipelines typically apply the same sequence of models to every incoming document, regardless of its complexity or type. This leads to inefficient use of compute and human resources: simple documents are over-processed while difficult ones may not receive enough scrutiny. We introduce DocHRL, a hierarchical reinforcement learning framework that learns to adaptively and dynamically select the most cost-effective classification policy on a per-document basis. DocHRL formulates document classification as a sequential decision problem with a two-level policy hierarchy: a top-level policy selects among broad options (vision classifiers, LLMs, OCR, and human-in-the-loop review), while option-specific sub-policies choose the concrete model or tool to invoke. The reward signal is the negative total expected cost, which captures inference cost, cost of misclassification, and cost of human labelling. Trained with Proximal Policy Optimisation on the RVL-CDIP benchmark, DocHRL achieves a macro F1 of 0.973 across 16 document classes while reducing average per-document cost to 2.74 normalised units compared to substantially higher costs incurred by fixed standalone classifiers. Our results demonstrate that cost-aware reinforcement learning can simultaneously improve classification performance and operational efficiency in document understanding systems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。