Multi-LLM Collaborative Caption Generation in Scientific Documents
Jaeyoung Kim, Jongho Lee, Hong-Jun Choi, Ting-Yao Hsu, Chieh-Yang Huang, Sungchul Kim, Ryan Rossi, Tong Yu, Clyde Lee Giles, Ting-Hao 'Kenneth' Huang, Sungchul Choi
TL;DR
MLBCAP tackles the challenge of scientific figure captioning by integrating textual and visual cues through a three-stage pipeline: quality assessment, diverse caption generation, and judgment. It cleans training data from arXiv-derived captions, generates diverse candidate captions with four specialized LLMs, and selects/refines the best caption using a prominent LLM, producing both long and short versions. Human evaluations show MLBCAP captions are preferred to author-written captions and outperform baselines, though traditional metrics like BLEU/ROUGE do not reliably reflect quality. The approach demonstrates the value of multi-LLM collaboration for high-quality, multimodal figure captions in scientific documents and provides a practical framework with code available at the project repository.
Abstract
Scientific figure captioning is a complex task that requires generating contextually appropriate descriptions of visual content. However, existing methods often fall short by utilizing incomplete information, treating the task solely as either an image-to-text or text summarization problem. This limitation hinders the generation of high-quality captions that fully capture the necessary details. Moreover, existing data sourced from arXiv papers contain low-quality captions, posing significant challenges for training large language models (LLMs). In this paper, we introduce a framework called Multi-LLM Collaborative Figure Caption Generation (MLBCAP) to address these challenges by leveraging specialized LLMs for distinct sub-tasks. Our approach unfolds in three key modules: (Quality Assessment) We utilize multimodal LLMs to assess the quality of training data, enabling the filtration of low-quality captions. (Diverse Caption Generation) We then employ a strategy of fine-tuning/prompting multiple LLMs on the captioning task to generate candidate captions. (Judgment) Lastly, we prompt a prominent LLM to select the highest quality caption from the candidates, followed by refining any remaining inaccuracies. Human evaluations demonstrate that informative captions produced by our approach rank better than human-written captions, highlighting its effectiveness. Our code is available at https://github.com/teamreboott/MLBCAP
