English

COMO: Closed-Loop Optical Molecule Recognition with Minimum Risk Training

Computer Vision and Pattern Recognition 2026-04-28 v1 Artificial Intelligence Machine Learning

Abstract

Optical chemical structure recognition (OCSR) translates molecular images into machine-readable representations like SMILES strings or molecular graphs, but remains challenging in real-world documents due to inexhaustible variations in chemical structures, shorthand conventions, and visual noise. Most existing deep-learning-based approaches rely on teacher forcing with token-level Maximum Likelihood Estimation (MLE). This training paradigm suffers from exposure bias, as models are trained under ground-truth prefixes but must condition on their own previous predictions during inference. Moreover, token-level MLE objectives hinder the optimization towards molecular-level evaluation criteria such as chemical validity and structural similarity. Here we introduce Minimum Risk Training (MRT) to OCSR and propose COMO (Closed-loop Optical Molecule recOgnition), a closed-loop framework that mitigates exposure bias by directly optimizing over molecule-level, non-differentiable objectives, by iteratively sampling and evaluating the model's own predictions. Experiments on ten benchmarks including synthetic and real-world chemical diagrams from patent and scientific literature demonstrate that COMO substantially outperforms existing rule-based and learning-based methods with less training data. Ablation studies further show that MRT is architecture-agnostic, demonstrating its potential for broad application to end-to-end OCSR systems.

Keywords

Cite

@article{arxiv.2604.23546,
  title  = {COMO: Closed-Loop Optical Molecule Recognition with Minimum Risk Training},
  author = {Zhuoqi Lyu and Qing Ke},
  journal= {arXiv preprint arXiv:2604.23546},
  year   = {2026}
}
R2 v1 2026-07-01T12:35:31.500Z