用文本增强提升图表理解,小模型也能超大模型
AskChart: Universal Chart Understanding through Textual Enhancement
- 融合图文信息的专家混合模型,显式利用图表中的文字标签
- 4.6B参数模型在多项任务上超越13B模型,最高提升68.3%
- 适合需要轻量高效图表理解的应用场景
图表理解任务如ChartQA和Chart-to-Text旨在自动提取并解析图表中的关键信息,使用户能通过提问或转换将视觉数据变为结构化内容。现有方法主要依赖图表图像的视觉线索,未能显式利用图表中嵌入的丰富文本信息(如数据标签、坐标轴标签),而这些文本对人类直观理解至关重要。此外,现有模型通常庞大且计算成本高,限制了实际应用。本文提出AskChart,一种通用模型,通过混合专家(MoE)架构显式整合图表的文本与视觉线索,学习增强的图文联合表示,有效处理多种图表理解任务,同时保持较小模型规模。为捕捉图文模态间的协同效应,我们构建了一个包含约750万样本的大规模数据集ChartBank,用于对齐图文信息并促进视觉实体与文本的提取。为有效训练AskChart,设计了三阶段训练策略,以对齐视觉与文本模态,优化MoE层的学习。在五个数据集上的大量实验表明,AskChart在四项图表理解任务中均取得显著性能提升。值得注意的是,4.6B参数的AskChart在开放式ChartQA任务中表现优于13B参数的先进模型68.3%,在Chart-to-Text任务中提升49.2%,在ChartQA和Chart-to-Table任务中也达到相当水平。
原文摘要 · Abstract (English)
Chart understanding tasks such as ChartQA and Chart-to-Text involve automatically extracting and interpreting key information from charts, enabling users to query or convert visual data into structured formats. State-of-the-art approaches primarily focus on visual cues from chart images, failing to explicitly incorporate rich textual information (e.g., data labels and axis labels) embedded within the charts. This textual information is vital for intuitive human comprehension and interpretation of charts. Moreover, existing models are often large and computationally intensive, limiting their practical applicability. In this paper, we introduce AskChart, a universal model that explicitly integrates both textual and visual cues from charts using a Mixture of Experts (MoE) architecture. AskChart facilitates the learning of enhanced visual-textual representations of charts for effectively handling multiple chart understanding tasks, while maintaining a smaller model size. To capture the synergy between visual and textual modalities, we curate a large-scale dataset named ChartBank with about 7.5M data samples, which helps align textual and visual information and facilitates the extraction of visual entities and text. To effectively train AskChart, we design a three-stage training strategy to align visual and textual modalities for learning robust visual-textual representations and optimizing the learning of the MoE layer. Extensive experiments across five datasets demonstrate the significant performance gains of AskChart in four chart understanding tasks. Remarkably, AskChart with 4.6B parameters outperforms state-of-the-art models with 13B parameters by 68.3% in Open-ended ChartQA and 49.2% in Chart-to-Text tasks, while achieving comparable performance in ChartQA and Chart-to-Table tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。