Uni-MuMER: Unified Multi-Task Fine-Tuning of Vision-Language Model for Handwritten Mathematical Expression Recognition

Yu Li (International Digital Economy Academy) · Yuxuan Zhou (Southern University of Science and Technology) · Jin Jiang (Peking University) · Jianhua Zhu (Peking University) · Shuai Peng (Peking University) · Baole (Zhongguancun Institute of Artificial Intelligence) · Liangcai Gao (Peking University)
crohme datasetcross-task generalizationerror-driven learninghandwritten mathematical expression recognitionhme100k datasetlightweight specialized modeloptical character recognitionpretrained vision-language modelsrecognition consistencystructured spatial reasoningsuper state-of-the-art performancesymbol countingtree-aware chain-of-thoughtunified solutionsvisually similar characterszero-shot setting

Handwritten Mathematical Expression Recognition (HMER) remains a persistent challenge in Optical Character Recognition (OCR) due to the inherent freedom of symbol layouts and variability in handwriting styles. Prior methods have faced performance bottlenecks by proposing isolated architectural modifications, making them difficult to integrate coherently into a unified framework. Meanwhile, recent advances in pretrained vision-language models (VLMs) have demonstrated strong cross-task generalization, offering a promising foundation for developing unified solutions. In this paper, we introduce Uni-MuMER, which fully fine-tunes a VLM for the HMER task without modifying its architecture, effectively injecting domain-specific knowledge into a generalist framework. Our method integrates three data-driven tasks: Tree-Aware Chain-of-Thought (Tree-CoT) for structured spatial reasoning, Error-Driven Learning (EDL) for reducing confusion among visually similar characters, and Symbol Counting (SC) for improving recognition consistency in long expressions. Experiments on the CROHME and HME100K datasets show that Uni-MuMER achieves super state-of-the-art performance, outperforming the best lightweight specialized model SSAN by 16.31\% and the top-performing VLM Gemini2.5-flash by 24.42\% under zero-shot setting. Our datasets, models, and code are open-sourced at: https://github.com/BFlameSwift/Uni-MuMER