TAMER: Tree-Aware Transformer for Handwritten Mathematical Expression Recognition

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Zhu, Jianhua, Zhao, Wenqi, Li, Yu, Hu, Xingjian, Gao, Liangcai
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866913606933676032
author Zhu, Jianhua
Zhao, Wenqi
Li, Yu
Hu, Xingjian
Gao, Liangcai
author_facet Zhu, Jianhua
Zhao, Wenqi
Li, Yu
Hu, Xingjian
Gao, Liangcai
contents Handwritten Mathematical Expression Recognition (HMER) has extensive applications in automated grading and office automation. However, existing sequence-based decoding methods, which directly predict $\LaTeX$ sequences, struggle to understand and model the inherent tree structure of $\LaTeX$ and often fail to ensure syntactic correctness in the decoded results. To address these challenges, we propose a novel model named TAMER (Tree-Aware Transformer) for handwritten mathematical expression recognition. TAMER introduces an innovative Tree-aware Module while maintaining the flexibility and efficient training of Transformer. TAMER combines the advantages of both sequence decoding and tree decoding models by jointly optimizing sequence prediction and tree structure prediction tasks, which enhances the model's understanding and generalization of complex mathematical expression structures. During inference, TAMER employs a Tree Structure Prediction Scoring Mechanism to improve the structural validity of the generated $\LaTeX$ sequences. Experimental results on CROHME datasets demonstrate that TAMER outperforms traditional sequence decoding and tree decoding models, especially in handling complex mathematical structures, achieving state-of-the-art (SOTA) performance.
format Preprint
id arxiv_https___arxiv_org_abs_2408_08578
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle TAMER: Tree-Aware Transformer for Handwritten Mathematical Expression Recognition
Zhu, Jianhua
Zhao, Wenqi
Li, Yu
Hu, Xingjian
Gao, Liangcai
Computer Vision and Pattern Recognition
Handwritten Mathematical Expression Recognition (HMER) has extensive applications in automated grading and office automation. However, existing sequence-based decoding methods, which directly predict $\LaTeX$ sequences, struggle to understand and model the inherent tree structure of $\LaTeX$ and often fail to ensure syntactic correctness in the decoded results. To address these challenges, we propose a novel model named TAMER (Tree-Aware Transformer) for handwritten mathematical expression recognition. TAMER introduces an innovative Tree-aware Module while maintaining the flexibility and efficient training of Transformer. TAMER combines the advantages of both sequence decoding and tree decoding models by jointly optimizing sequence prediction and tree structure prediction tasks, which enhances the model's understanding and generalization of complex mathematical expression structures. During inference, TAMER employs a Tree Structure Prediction Scoring Mechanism to improve the structural validity of the generated $\LaTeX$ sequences. Experimental results on CROHME datasets demonstrate that TAMER outperforms traditional sequence decoding and tree decoding models, especially in handling complex mathematical structures, achieving state-of-the-art (SOTA) performance.
title TAMER: Tree-Aware Transformer for Handwritten Mathematical Expression Recognition
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2408.08578