arXiv:2412.09936cs.CV2024-12被引 9

用图文对训练模型,单图估算卡路里并支持对话交互。

CaLoRAify: Calorie Estimation with Visual-Text Pairing and LoRA-Driven Visual Language Models

  • 基于图文对训练视觉语言模型,实现食材识别与卡路里估算。
  • 在33万图文对数据集上,显著提升卡路里估算准确率。
  • 支持单图输入和对话式交互,适合健康饮食应用。

肥胖是全球主要可预防慢性病的首要原因。传统卡路里估算工具依赖特定数据格式或复杂流程,难以在真实场景中应用。近年来,视觉语言模型(VLMs)在理解现实语境和实现对话交互方面表现优异,适用于食材分析等下游任务。然而,将VLM应用于卡路里估算需领域特定数据与对齐策略。为此,我们构建了包含33万张图像-文本对的CalData数据集,结合大规模食谱数据与详细营养说明,以支持鲁棒的视觉语言训练。在此基础上,提出CaLoRAify框架,通过视觉-文本对训练实现食材识别与卡路里估算的联合优化。推理时仅需单张食物图像即可估算卡路里,并保留基于智能体的对话灵活性。结合低秩适配(LoRA)与检索增强生成(RAG)技术,显著提升基础VLM在卡路里估算垂直领域的性能。代码与数据已开源:https://github.com/KennyYao2001/16824-CaLORAify。

原文摘要 · Abstract (English)

The obesity phenomenon, known as the heavy issue, is a leading cause of preventable chronic diseases worldwide. Traditional calorie estimation tools often rely on specific data formats or complex pipelines, limiting their practicality in real-world scenarios. Recently, vision-language models (VLMs) have excelled in understanding real-world contexts and enabling conversational interactions, making them ideal for downstream tasks such as ingredient analysis. However, applying VLMs to calorie estimation requires domain-specific data and alignment strategies. To this end, we curated CalData, a 330K image-text pair dataset tailored for ingredient recognition and calorie estimation, combining a large-scale recipe dataset with detailed nutritional instructions for robust vision-language training. Built upon this dataset, we present CaLoRAify, a novel VLM framework aligning ingredient recognition and calorie estimation via training with visual-text pairs. During inference, users only need a single monocular food image to estimate calories while retaining the flexibility of agent-based conversational interaction. With Low-rank Adaptation (LoRA) and Retrieve-augmented Generation (RAG) techniques, our system enhances the performance of foundational VLMs in the vertical domain of calorie estimation. Our code and data are fully open-sourced at https://github.com/KennyYao2001/16824-CaLORAify.

卡路里估算视觉语言模型LoRA

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。