Image-to-LaTeX Converter for Mathematical Formulas and Text

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Gurgurov, Daniil, Morshnev, Aleksey
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929452360925184
author Gurgurov, Daniil
Morshnev, Aleksey
author_facet Gurgurov, Daniil
Morshnev, Aleksey
contents In this project, we train a vision encoder-decoder model to generate LaTeX code from images of mathematical formulas and text. Utilizing a diverse collection of image-to-LaTeX data, we build two models: a base model with a Swin Transformer encoder and a GPT-2 decoder, trained on machine-generated images, and a fine-tuned version enhanced with Low-Rank Adaptation (LoRA) trained on handwritten formulas. We then compare the BLEU performance of our specialized model on a handwritten test set with other similar models, such as Pix2Text, TexTeller, and Sumen. Through this project, we contribute open-source models for converting images to LaTeX and provide from-scratch code for building these models with distributed training and GPU optimizations.
format Preprint
id arxiv_https___arxiv_org_abs_2408_04015
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Image-to-LaTeX Converter for Mathematical Formulas and Text
Gurgurov, Daniil
Morshnev, Aleksey
Computation and Language
Computer Vision and Pattern Recognition
In this project, we train a vision encoder-decoder model to generate LaTeX code from images of mathematical formulas and text. Utilizing a diverse collection of image-to-LaTeX data, we build two models: a base model with a Swin Transformer encoder and a GPT-2 decoder, trained on machine-generated images, and a fine-tuned version enhanced with Low-Rank Adaptation (LoRA) trained on handwritten formulas. We then compare the BLEU performance of our specialized model on a handwritten test set with other similar models, such as Pix2Text, TexTeller, and Sumen. Through this project, we contribute open-source models for converting images to LaTeX and provide from-scratch code for building these models with distributed training and GPU optimizations.
title Image-to-LaTeX Converter for Mathematical Formulas and Text
topic Computation and Language
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2408.04015