TRivia is a novel self-supervised fine-tuning framework of vision-language models for table recognition. This repository contains the TRivia-3B, an advanced table recognition VLMs trained from Qwen2.5-VL-3B using TRivia, and demo code. TRivia-3B has demonstrated superior performance on multiple real-world table recognition benchmarks.
Key Features:
- ⭐ Powerful table recognition capabilities, generalizing across digital tables, scanned tables, and photographed tables.
- 📃 Reproducible training framework that pushes the boundaries of table recognition capabilities using unlabeled table images.
Benchmark Performance
We compare the performance of TRivia-3B with other table recognition solution on three benchmarks: OmnidocBench v1.5, CC-OCR and OCRBench v2
Installation
TRivia-3B is trained based on Qwen2.5-VL-3B so that you can follow the Qwen2.5-VL-3B installation guide.
We highly recommend installing vLLM >= 0.7.2 to improve inference speed.
Usage
TRivia-3B supports table parsing with table images as input and outputting OTSL tags as results.
TRivia-3B is an experimental model, and it currently does not support parsing formulas in tables or tables with images.
Using vLLM for offline inference
Make sure you have installed vllm >= 0.7.2. Papre your table images in a folder and run the following command:
python run_vllm_offline_inf.py --ckpt_root opendatalab/TRivia-3B --image_root /path/to/images --output_path ./vllm_offline_output.json
# Examples
python run_vllm_offline_inf.py --ckpt_root opendatalab/TRivia-3B --image_root ./examples --output_path ./examples_output.json
The output is a JSON file (example) which is formatted as folows:
[
{
"path": "...", // Image path
"otsl": "...", // Unprocessed OTSL tags output by the model
"html": "...", // Converted HTML tags
}
]
Using vLLM for online deployment
You can start either a vLLM or SGLang server to serve LLMs efficiently, and then access it using an OpenAI-style API.
- Start vLLM Server
vllm serve opendatalab/TRivia --port 10000 --gpu_memory_utilization 0.8
- Table Image Request
import base64
from openai import OpenAI
from otsl_utils import otsl_to_html
client = OpenAI(
api_key="EMPTY",
base_url="http://127.0.0.1:10000/v1",
timeout=3600
)
image_path = "./examples/docstructbench_llm-raw-scihub-o.O-ijc.22994.pdf_3_5.png"
with open(path, "rb") as image_file:
base64_image = base64.b64encode(image_file.read()).decode('utf-8')
messages = [
{
"role": "user",
"content": [
{
"type": "text",
"text": "You are an AI specialized in recognizing and extracting table from images. Your mission is to analyze the table image and generate the result in OTSL format using specified tags. Output only the results without any other words and explanation." # Make sure to use this prompt for optimal performance.
},
{
"type": "image_url",
"image_url": {"url": f"data:image/jpeg;base64,{base64_image}"}
}
]
}
]
response = client.chat.completions.create(
model="opendatalab/TRivia",
messages=messages,
temperature=0.0,
max_tokens=8192
)
otsl_content = response.choices[0].message.content
html_content = otsl_to_html(otsl_content)
print(f"Generated otsl tags: {otsl_content}")
print(f"HTML table: {html_content}")
Citation
@misc{zhang2025triviaselfsupervisedfinetuningvisionlanguage,
title={TRivia: Self-supervised Fine-tuning of Vision-Language Models for Table Recognition},
author={Junyuan Zhang and Bin Wang and Qintong Zhang and Fan Wu and Zichen Wen and Jialin Lu and Junjie Shan and Ziqi Zhao and Shuya Yang and Ziling Wang and Ziyang Miao and Huaping Zhong and Yuhang Zang and Xiaoyi Dong and Ka-Ho Chow and Conghui He},
year={2025},
eprint={2512.01248},
archivePrefix={arXiv},
primaryClass={cs.CV},
url={https://arxiv.org/abs/2512.01248},
}