Back to Feed
Multimodal / Reinforcement Learning

Improving Japanese Structured Document Parsing

Original: Stockmark-Nemotron-3-Nano-Omni-JapanDocReader: Structured Document Parsing via Capability Injection and Forgetting Control

Listen to the summary

Uses a voice available on your device

Audio options
On this page 4 sections
Related concepts 5 concepts

Key Takeaways

  • Used a mixed supervised fine-tuning strategy to inject parsing skills while preserving visual question answering performance.
  • Achieved an overall parsing score of 87.67 on OmniDocBench-JASyn using reinforcement learning.
  • Maintained a solid visual question answering score of 0.844 after the mixed training strategy.
  • Used a validity gate and task-aligned similarity metrics to improve model reliability during reinforcement learning.

Summary & Methodology Analysis

The researchers addressed the performance degradation that typically occurs when adding new task capabilities to a multimodal model. By using a data engine with two synthetic streams, they combined a Japanese Document VQA Stream with a programmatic structured parsing stream. This mixed strategy allowed them to perform supervised fine-tuning, which is the process of further training a pre-trained model on a specific, labeled dataset to refine its output, while mitigating the loss of original capabilities. They initially observed that parsing-centric fine-tuning increased the DocParse-Overall score from 70.43 to 86.56 but saw a drop in VQA-Overall from 0.863 to 0.826. By balancing these datasets, they successfully maintained parsing performance at 86.60 while recovering the VQA-Overall score to 0.844.

To further improve the parsing output, the team implemented reinforcement learning (RL), a training approach where a model learns by receiving rewards for desired behaviors, using the DAPO algorithm on the fine-tuned checkpoint. They utilized a representation-aware reward that incorporates a strict validity gate and task-aligned similarity metrics. Furthermore, they employed variance-based prompt filtering, a method to select training samples that highlight meaningful performance differences, to avoid training on saturated or unlearnable examples. This final step pushed the DocParse-Overall score to 87.67, demonstrating the effectiveness of combining mixed fine-tuning with targeted reinforcement learning.

Despite these improvements, the paper notes significant limitations in the current approach. The reinforcement learning process creates a trade-off where gains in parsing accuracy often lead to drift, or degradation, in the model's visual question answering performance. Additionally, the researchers identified that their current comparison between thinking-mode and instruct-mode does not fully isolate reasoning traces from other prompt effects. The paper does not specify the exact hardware requirements or training time involved in these processes.

Interactive System Flowchart

Click diagram to expand and zoom

Cross-Examination & FAQs

A deeper dive clarifying mechanics, constraints, and baseline evaluations.

Q1. What is the main goal of this research?

The goal is to inject structured document parsing capabilities into a reasoning-based multimodal model without losing its ability to perform document visual question answering.

Q2. What model was produced by this study?

The study produced the Stockmark-Nemotron-3-Nano-Omni-JapanDocReader model.

Q3. How successful was the final model?

The model achieved an overall DocParse-Overall score of 87.67 on the OmniDocBench-JASyn benchmark.

Q4. How did the team balance two different capabilities during fine-tuning?

They constructed a data engine with two synthetic streams and performed supervised fine-tuning using a mixed strategy that combined both structured document parsing data and visual question answering data.

Q5. What role did reinforcement learning play in the methodology?

The team applied DAPO-based parsing-centric reinforcement learning on the mixed fine-tuning checkpoint, which used a representation-aware reward system to further improve parsing performance.

Q6. What is variance-based prompt filtering?

It is a technique used to select RL prompts that expose meaningful quality differences without being already saturated or impossible for the model to learn.

Q7. Did the visual question answering capability recover after mixed training?

Yes, after the drop caused by parsing-centric fine-tuning, the mixed strategy recovered the VQA-Overall score to 0.844.

Q8. What were the primary limitations identified by the authors?

The authors identified a trade-off where parsing gains lead to VQA drift and noted that their current testing does not fully isolate reasoning traces from other prompt effects.

Q9. What hardware was used to train these models?

The paper does not specify the hardware used for training.